Reachability analysis for piecewise affine systems with neural network-based controllers
Dieter Teichrib, Moritz Schulze Darup
TL;DR
This work tackles the challenge of preserving stability and constraint satisfaction when neural networks approximate MPC laws for piecewise affine systems. It develops an MILP-based framework to compute over-approximations of k-step reachable sets, certify that the closed-loop system converges to a small positively invariant set containing the origin, and enforce state/output constraints without requiring a baseline controller. It further shows how to modify the NN controller to guarantee asymptotic stability via a dual-mode switching law and Lyapunov-based design, demonstrated on a four-region PWA example with a maxout NN. The approach integrates PWA dynamics, maxout NN activations, and MILP encodings to provide verifiable safety and stability guarantees for NN-controlled PWA systems, with potential extensions to faster conservatisms using alternative reachability methods.
Abstract
Neural networks (NN) have been successfully applied to approximate various types of complex control laws, resulting in low-complexity NN-based controllers that are fast to evaluate. However, when approximating control laws using NN, performance and stability guarantees of the original controller may not be preserved. Recently, it has been shown that it is possible to provide such guarantees for linear systems with NN-based controllers by analyzing the approximation error with respect to a stabilizing base-line controller or by computing reachable sets of the closed-loop system. The latter has the advantage of not requiring a base-line controller. In this paper, we show that similar ideas can be used to analyze the closed-loop behavior of piecewise affine (PWA) systems with an NN-based controller. Our approach builds on computing over-approximations of reachable sets using mixed-integer linear programming, which allows to certify that the closed-loop system converges to a small set containing the origin while satisfying input and state constraints. We also show how to modify a given NN-based controller to ensure asymptotic stability for the controlled PWA system.
