On the Sample Complexity of Imitation Learning for Smoothed Model Predictive Control

Daniel Pfrommer,Swati Padmanabhan,Kwangjun Ahn,Jack Umenberger,Tobia Marcucci,Zakaria Mhammedi,Ali Jadbabaie
2024-09-04
Abstract:Recent work in imitation learning has shown that having an expert controller that is both suitably smooth and stable enables stronger guarantees on the performance of the learned controller. However, constructing such smoothed expert controllers for arbitrary systems remains challenging, especially in the presence of input and state constraints. As our primary contribution, we show how such a smoothed expert can be designed for a general class of systems using a log-barrier-based relaxation of a standard Model Predictive Control (MPC) optimization problem. At the crux of this theoretical guarantee on smoothness is a new lower bound we prove on the optimality gap of the analytic center associated with a convex Lipschitz function, which we hope could be of independent interest. We validate our theoretical findings via experiments, demonstrating the merits of our smoothing approach over randomized smoothing.
Systems and Control,Machine Learning
What problem does this paper attempt to address?