$\texttt{immrax}$: A Parallelizable and Differentiable Toolbox for Interval Analysis and Mixed Monotone Reachability in JAX
Abstract: We present an implementation of interval analysis and mixed monotone interval reachability analysis as function transforms in Python, fully composable with the computational framework JAX. The resulting toolbox inherits several key features from JAX, including computational efficiency through Just-In-Time Compilation, GPU acceleration for quick parallelized computations, and Automatic Differentiability. We demonstrate the toolbox's performance on several case studies, including a reachability problem on a vehicle model controlled by a neural network, and a robust closed-loop optimal control problem for a swinging pendulum.
- Althoff, M. (2015). An introduction to CORA 2015. In Proc. of the 1st and 2nd Workshop on Applied Verification for Continuous and Hybrid Systems, 120–151. EasyChair. 10.29007/zbkv. URL https://easychair.org/publications/paper/xMm.
- JuliaReach: a toolbox for set-based reachability. In Proceedings of the 22nd International Conference on Hybrid Systems: Computation and Control (HSCC), 39–44.
- JAX: composable transformations of Python+NumPy programs. URL http://github.com/google/jax.
- Efficient finite abstraction of mixed monotone systems. In Proceedings of the 18th International Conference on Hybrid Systems: Computation and Control, 58–67.
- ReachNN*: A tool for reachability analysis of neural-network controlled systems. In International Symposium on Automated Technology for Verification and Analysis, 537–542. Springer.
- A toolbox for fast interval arithmetic in numpy with an application to formal verification of neural network controlled system. In 2nd ICML Workshop on Formal Verification of Machine Learning.
- Efficient interaction-aware interval analysis of neural network feedback loops. arXiv preprint arXiv:2307.14938.
- Applied Interval Analysis. Springer London.
- Kidger, P. (2021). On Neural Differential Equations. Ph.D. thesis, University of Oxford.
- Equinox: neural networks in JAX via callable PyTrees and filtered transformations. Differentiable Programming workshop at Neural Information Processing Systems 2021.
- Rapid and accurate reachability analysis for nonlinear dynamic systems by exploiting model redundancy. Computers & Chemical Engineering, 106, 596–608. 10.1016/j.compchemeng.2017.08.001. ESCAPE-26.
- Tsitouras, C. (2011). Runge–kutta pairs of order 5 (4) satisfying only the first column simplifying assumption. Computers & Mathematics with Applications, 62(2), 770–775.
- Polar-Express: Efficient and precise formal reachability analysis of neural-network controlled systems. IEEE Transactions on Computer-Aided Design of Integrated Circuits and Systems.
- Efficient neural network robustness certification with general activation functions. In Advances in Neural Information Processing Systems, volume 31, 4944–4953.
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.
