Learning-enhanced Nonlinear Model Predictive Control using Knowledge-based Neural Ordinary Differential Equations and Deep Ensembles

by   Kong Yao Chee, et al.

Nonlinear model predictive control (MPC) is a flexible and increasingly popular framework used to synthesize feedback control strategies that can satisfy both state and control input constraints. In this framework, an optimization problem, subjected to a set of dynamics constraints characterized by a nonlinear dynamics model, is solved at each time step. Despite its versatility, the performance of nonlinear MPC often depends on the accuracy of the dynamics model. In this work, we leverage deep learning tools, namely knowledge-based neural ordinary differential equations (KNODE) and deep ensembles, to improve the prediction accuracy of this model. In particular, we learn an ensemble of KNODE models, which we refer to as the KNODE ensemble, to obtain an accurate prediction of the true system dynamics. This learned model is then integrated into a novel learning-enhanced nonlinear MPC framework. We provide sufficient conditions that guarantees asymptotic stability of the closed-loop system and show that these conditions can be implemented in practice. We show that the KNODE ensemble provides more accurate predictions and illustrate the efficacy and closed-loop performance of the proposed nonlinear MPC framework using two case studies.


page 1

page 2

page 3

page 4


KNODE-MPC: A Knowledge-based Data-driven Predictive Control Framework for Aerial Robots

In this work, we consider the problem of deriving and incorporating accu...

LEARNEST: LEARNing Enhanced Model-based State ESTimation for Robots using Knowledge-based Neural Ordinary Differential Equations

State estimation is an important aspect in many robotics applications. I...

Autonomous Drifting with 3 Minutes of Data via Learned Tire Models

Near the limits of adhesion, the forces generated by a tire are nonlinea...

Online Dynamics Learning for Predictive Control with an Application to Aerial Robots

In this work, we consider the task of improving the accuracy of dynamic ...

Deep Learning Tubes for Tube MPC

Learning-based control aims to construct models of a system to use for p...

Constrained Neural Ordinary Differential Equations with Stability Guarantees

Differential equations are frequently used in engineering domains, such ...

Enhancing Sample Efficiency and Uncertainty Compensation in Learning-based Model Predictive Control for Aerial Robots

The recent increase in data availability and reliability has led to a su...

Please sign up or login with your details

Forgot password? Click here to reset