Time-Certified and Efficient NMPC via Koopman Operator
Liang Wu, Yunhong Che, Bo Yang, Kangyu Lin, Ján Drgoňa
TL;DR
This work addresses certifying worst-case execution time for nonlinear MPC under state–input constraints by marrying a data-driven Koopman predictor with a dynamics-relaxed BoxQP formulation. The approach converts NMPC into a structured BoxQP via a Koopman lift and a penalty that enforces model dynamics within the objective, then solves it with a time-certified, feasible path-following IPM that exploits problem structure to drastically reduce the Newton system dimension. Key contributions include a) learning a high-dimensional Koopman model, b) a dynamics-relaxed construction that yields a structured BoxQP, c) a worst-case iteration bound with a cost-free initialization, and d) efficient linear-system computations leveraging Schur complements, enabling real-time, certificate-backed NMPC for large-scale PDE control. Practically, the method achieves substantial speedups over generic QP solvers while providing explicit execution-time certificates, making it well-suited for PDE-like control applications where tight timing guarantees are essential.
Abstract
Certifying and accelerating execution times of nonlinear model predictive control (NMPC) implementations are two core requirements. Execution-time certificate guarantees that the NMPC controller returns a solution before the next sampling time, and achieving faster worst-case and average execution times further enables its use in a wider set of applications. However, NMPC produces a nonlinear program (NLP) for which it is challenging to derive its execution time certificates. Our previous works, \citep{wu2025direct,wu2025time} provide data-independent execution time certificates (certified number of iterations) for box-constrained quadratic programs (BoxQP). To apply the time-certified BoxQP algorithm \citep{wu2025time} for state-input constrained NMPC, this paper i) learns a linear model via Koopman operator; ii) proposes a dynamic-relaxation construction approach yields a structured BoxQP rather than a general QP; iii) exploits the structure of BoxQP, where the dimension of the linear system solved in each iteration is reduced from $5N(n_u+n_x)$ to $Nn_u$ (where $n_u, n_x, N$ denote the number of inputs, states, and length of prediction horizon), yielding substantial speedups (when $n_x \gg n_u$, as in PDE control).
