POLAR [1] is a reachability analysis framework for neural-network controlled systems (NNCSs) based on polynomial arithmetic. Compared with existing arithmetic approaches that use standard Taylor models, our framework uses a novel approach to iteratively overapproximate the neuron output ranges layer-by-layer with a combination of Bernstein polynomial interpolation for continuous activation functions and Taylor model arithmetic for the other operations. This approach can overcome the main drawback in the standard Taylor model arithmetic, i.e. its inability to handle functions that cannot be well approximated by Taylor polynomials, and significantly improve the accuracy and efficiency of reachable states computation for NNCSs. To further tighten the overapproximation, our method keeps the Taylor model remainders symbolic under the linear mappings when estimating the output range of a neural network.
Experiment results across a suite of benchmarks show that POLAR significantly outperforms the state-of-the-art techniques on both efficiency and tightness of reachable set estimation.
Ubuntu 18.04, MATLAB 2016a or later
- Install dependencies through apt-get install
sudo apt-get install m4 libgmp3-dev libmpfr-dev libmpfr-doc libgsl-dev gsl-bin bison flex gnuplot-x11 libglpk-dev gcc-8 g++-8 libopenmpi-dev
./compile.sh # under the root directory ./
./run_motivating.sh
./run_attitude_control.sh
./run.sh
All results will be stored in ./outputs/
For SYSTEM, the results include a txt file that show the verification result and the POLAR running time, and a M file (with .m extension) that is used to plot the reachable sets computed by POLAR. One can check the result of SYSTEM by following commands.
vim SYSTEM_0.txt # verification result
SYSTEM_0.m # plotted reachable sets. Run the command in MATLAB.
The following results show that our approach significantly outperforms the state-of-the-arts on both efficiency and tightness of reachable set estimation.
Figure 1: Reachability analysis results. We can see that except for ex2-sigmoid, POLAR produces the tightest reachable set estimation (dark green sets) and successfully proves or disproves the reachability property for all the examples. This is in comparison with other state-of-the-art tools including ReachNN* [2, 3] (light green sets), Sherlock [4] (blue sets), Verisig 2.0 [5] (grey sets), and NNV [6] (yellow sets).
Table 1: V: number of state variables,
Chao Huang, Jiameng Fan, Xin Chen, Wenchao Li, Qi Zhu
[1] C.Huang, J.Fan, W.Li, X.Chen, and Q.Zhu. POLAR: A Polynomial Arithmetic Framework for Verifying Neural-Network Controlled Systems
[2] C.Huang, J.Fan, W.Li, X.Chen, and Q.Zhu. ReachNN: Reachability Analysis of Neural-Network Controlled Systems. ACM Transactions on Embedded Computing Systems (TECS), 18:1–22, October 2019.
[3] J.Fan, C.Huang, X.Chen, W.Li, and Q.Zhu. ReachNN*: A Tool for Reachability Analysis ofNeural-Network Controlled Systems. The 18th International Symposium on Automated Technology for Verification and Analysis (ATVA), October 2020.
[4] S.Dutta, X.Chen, and S.Sankaranarayanan. Reachability analysis for neural feedback systems using regressive polynomial rule inference. The 22nd ACM International Conference on Hybrid Systems: Computation and Control, April 2019.
[5] R.Ivanov, T.J. Carpenter, J.Weimer, R.Alur, G.J. Pappas, and I.Lee. [Verisig 2.0: Verification of neural network controllers using taylor model preconditioning]. The 33rd International Conference on Computer-Aided Verification, July 2021 (To appear).
[6] H.Tran, X.Yang, D.M.Lopez, P.Musau, L.V.Nguyen, W.Xiang, S.Bak, and T.T.Johnson. NNV: the neural network verification tool for deep neural networks and learning-enabled cyber-physical systems. The 32nd International Conference on Computer-Aided Verification, July 2020.