Bayesian Optimization for Automatic Tuning of Torque-Level Nonlinear Model Predictive Control
Gabriele Fadini, Deepak Ingole, Tong Duy Son, Alisa Rupenyan
TL;DR
The paper tackles the challenge of tuning high-dimensional parameters in torque-based nonlinear MPC for robotic trajectory tracking. It introduces an automated framework that combines Sparse Axis-Aligned Subspace Bayesian Optimization (SAASBO) with a digital twin of the UR10e, enabling efficient exploration of 12-dimensional MPC parameter space and rapid convergence to high-performance configurations. Simulation results show up to 41.9% improvement in tracking error over baselines, with real-robot validation achieving a 25.8% improvement, demonstrating effective sim-to-real transfer. The work highlights the value of sparsity-aware Bayesian optimization for complex, real-time robotic controllers and points to future work on richer physical modeling to further close the sim-to-real gap.
Abstract
This paper presents an auto-tuning framework for torque-based Nonlinear Model Predictive Control (nMPC), where the MPC serves as a real-time controller for optimal joint torque commands. The MPC parameters, including cost function weights and low-level controller gains, are optimized using high-dimensional Bayesian Optimization (BO) techniques, specifically Sparse Axis-Aligned Subspace (SAASBO) with a digital twin (DT) to achieve precise end-effector trajectory real-time tracking on an UR10e robot arm. The simulation model allows efficient exploration of the high-dimensional parameter space, and it ensures safe transfer to hardware. Our simulation results demonstrate significant improvements in tracking performance (+41.9%) and reduction in solve times (-2.5%) compared to manually-tuned parameters. Moreover, experimental validation on the real robot follows the trend (with a +25.8% improvement), emphasizing the importance of digital twin-enabled automated parameter optimization for robotic operations.
