ZoPE: A Fast Optimizer for ReLU Networks with Low-Dimensional Inputs
This work addresses the need for efficient verification and optimization in safety-critical systems, though it is incremental as it builds on existing mixed-integer programming approaches with a novel method.
The paper tackles the problem of solving optimization problems over the output of feedforward ReLU networks with low-dimensional inputs to improve safety and robustness in neural networks, achieving a 25x speedup on an ACAS Xu benchmark and an 85x speedup on linear optimization problems compared to existing methods.
Deep neural networks often lack the safety and robustness guarantees needed to be deployed in safety critical systems. Formal verification techniques can be used to prove input-output safety properties of networks, but when properties are difficult to specify, we rely on the solution to various optimization problems. In this work, we present an algorithm called ZoPE that solves optimization problems over the output of feedforward ReLU networks with low-dimensional inputs. The algorithm eagerly splits the input space, bounding the objective using zonotope propagation at each step, and improves computational efficiency compared to existing mixed-integer programming approaches. We demonstrate how to formulate and solve three types of optimization problems: (i) minimization of any convex function over the output space, (ii) minimization of a convex function over the output of two networks in series with an adversarial perturbation in the layer between them, and (iii) maximization of the difference in output between two networks. Using ZoPE, we observe a $25\times$ speedup on property $1$ of the ACAS Xu neural network verification benchmark compared to several state-of-the-art verifiers, and an $85\times$ speedup on a set of linear optimization problems compared to a mixed-integer programming baseline. We demonstrate the versatility of the optimizer in analyzing networks by projecting onto the range of a generative adversarial network and visualizing the differences between a compressed and uncompressed network.