Search papers, labs, and topics across Lattice.
This paper introduces Residual Koopman Model Predictive Control (RK-MPC), a data-driven MPC framework for quadruped locomotion that combines a nominal template model with a learned linear residual predictor in lifted Koopman coordinates. By learning to correct for model mismatch due to contact variability and terrain disturbances, RK-MPC achieves improved prediction fidelity while maintaining real-time performance. Hardware experiments on a Unitree Go1 demonstrate reliable blind locomotion across diverse off-road terrains at 500 Hz, outperforming Koopman/EDMD baselines with alternative observable dictionaries.
Quadruped robots can now traverse challenging terrains like snow and ice without relying on explicit terrain mapping, thanks to a data-driven MPC that learns to correct model errors on the fly.
This paper presents Residual Koopman MPC (RK-MPC), a Koopman-based, data-driven model predictive control framework for quadruped locomotion that improves prediction fidelity while preserving real-time tractability. RK-MPC augments a nominal template model with a compact linear residual predictor learned from data in lifted coordinates, enabling systematic correction of model mismatch induced by contact variability and terrain disturbances with provable bounds on multi-step prediction error. The learned residual model is embedded within a convex quadratic-program MPC formulation, yielding a receding-horizon controller that runs onboard at 500 Hz and retains the structure and constraint-handling advantages of optimization-based control. We evaluate RK-MPC in both Gazebo simulation and Unitree Go1 hardware experiments, demonstrating reliable blind locomotion across contact disturbances, multiple gait schedules, and challenging off-road terrains including grass, gravel, snow, and ice. We further compare against Koopman/EDMD baselines using alternative observable dictionaries, including monomial and $SE(3)$-structured bases, and show that the residual correction improves multi-step prediction and closed-loop performance while reducing sensitivity to the choice of observables. Overall, RK-MPC provides a practical, hardware-validated pathway for data-driven predictive control of quadrupeds in unstructured environments. See https://sriram-2502.github.io/rk-mpc for implementation videos.