Efficient model predictive control for nonlinear systems modelled by deep neural networks
Jianglin Lan
TL;DR
This work addresses controlling discrete-time systems whose nonlinear dynamics are represented by deep neural networks under state and input constraints. It introduces a dual-mode MPC approach combining a steady-state pre-stabilizing feedback with a constraint-enforcing MPC, and develops three solving strategies for the nonlinear MPC: an exact MIP encoding and two relaxations (LR and enhanced LR, eLR) to balance accuracy and computation. The enhanced LR approach achieves tracking performance close to the exact MIP while maintaining substantially lower computation, demonstrated on an inverted pendulum model with NN-based dynamics. These methods enable real-time control of NN-driven nonlinear systems and lay the groundwork for extensions to robust/tube-MPC and other activation functions. The results highlight a practical trade-off between optimality and speed, with eLR offering a scalable path toward real-time NN-assisted MPC.
Abstract
This paper presents a model predictive control (MPC) for dynamic systems whose nonlinearity and uncertainty are modelled by deep neural networks (NNs), under input and state constraints. Since the NN output contains a high-order complex nonlinearity of the system state and control input, the MPC problem is nonlinear and challenging to solve for real-time control. This paper proposes two types of methods for solving the MPC problem: the mixed integer programming (MIP) method which produces an exact solution to the nonlinear MPC, and linear relaxation (LR) methods which generally give suboptimal solutions but are much computationally cheaper. Extensive numerical simulation for an inverted pendulum system modelled by ReLU NNs of various sizes is used to demonstrate and compare performance of the MIP and LR methods.
