Search papers, labs, and topics across Lattice.
This paper addresses the computational bottleneck of Nonlinear Model Predictive Control (NMPC) by approximating the optimal control move with a state-dependent quadratic program (QP) tailored for input-affine nonlinear systems. The authors introduce a single-network residual-corrector architecture that combines an analytic baseline for initial QP parameters with a neural network that learns necessary corrections, ensuring efficient constraint satisfaction through a differentiable interior-point layer. Validation on a three-link planar robotic arm shows that this method achieves significant speed improvements over traditional NLP solvers while preserving tracking performance.
Achieving orders-of-magnitude speedup in nonlinear control without sacrificing performance could revolutionize real-time applications in robotics and automation.
Nonlinear Model Predictive Control requires solving a constrained nonlinear program (NLP) in real-time at every sampling instant, a computational bottleneck that limits deployment on resource-constrained hardware or at high sampling rates. We address this challenge for the broad class of input-affine nonlinear systems to show that the optimal control move can be approximated by a state-dependent quadratic program (QP) whose cost parameters depend on the current state and reference. We propose a single-network residual-corrector architecture: a state-dependent analytic baseline provides initial QP parameters, and the network learns only the corrections needed to match the full NLP solution; the QP is solved by a differentiable interior-point layer, guaranteeing constraint satisfaction for the first control action. The network is trained offline on data generated by an NLP solver using a hybrid loss that combines supervised imitation and KKT-residual penalties. We validate the approach on a three-link planar robotic arm with Cartesian end-effector tracking, demonstrating orders-of-magnitude speedup over the NLP solver while maintaining comparable tracking performance.