Replacing K-infinity Function with Leaky ReLU in Barrier Function Design: A Union of Invariant Sets Approach for ReLU-Based Dynamical Systems
Pouya Samanipour, Hasan Poonawala
TL;DR
This work tackles safety guarantees for dynamical systems identified by ReLU networks or their PWA equivalents by replacing the conventional nonlinear class K_infinity barrier function with a practical Leaky ReLU alpha and by introducing the Union of Invariant Sets (UIS) to fuse information from multiple barrier candidates. It develops a systematic PWA barrier-function design on a shared partition, and constructs a UIS by taking the maximum of several BF candidates obtained with different linear alphas, producing a potentially larger forward-invariant set. The key contributions are (i) a Leaky ReLU-based alpha that ensures barrier conditions for bounded barrier functions, (ii) a PWA BF formulation with tractable linear optimizations, and (iii) the UIS framework that combines multiple invariant sets without increasing computational cost. The method is validated on ReLU-based dynamics, including inverted pendulum scenarios, demonstrating that UIS can enlarge invariant sets while preserving safety guarantees, with open-source code provided for reproducibility and practical adoption.
Abstract
In this paper, a systematic framework is presented for determining piecewise affine PWA barrier functions and their corresponding invariant sets for dynamical systems identified via Rectified Linear Unit (ReLU) neural networks or their equivalent PWA representations. A common approach to determining the invariant set is to use Nagumo's condition, or to utilize the barrier function with a class K-infinity function. It may be challenging to find a suitable class K-infinity function in some cases. We propose leaky ReLU as an efficient substitute for the complex nonlinear K-infinity function in our formulation. Moreover, we propose the Union of Invariant Sets (UIS) method, which combines information from multiple invariant sets in order to compute the largest possible PWA invariant set. The proposed framework is validated through multiple examples, showcasing its potential to enhance the analysis of invariant sets in ReLU-based dynamical systems. Our code is available at: https://github.com/PouyaSamanipour/UIS.git.
