Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces GPU-SLS, a GPU-parallelized framework for safe and robust nonlinear MPC that can handle high-dimensional uncertain robotic systems and long planning horizons. The method jointly optimizes a dynamically-feasible nominal trajectory, a tracking controller, and a closed-loop reachable set under disturbance in real-time using a novel GPU-accelerated quadratic program (QP) solver within an ADMM framework. Results show a 97.7% reduction in nominal trajectory solve times compared to CPU solvers and a 237x acceleration in SLS-based control and reachability, achieving 100% empirical safety on complex systems like quadrupeds and humanoids.
Achieve safe, robust nonlinear MPC for high-dimensional robots in just 20 milliseconds by unleashing the power of GPU-accelerated system level synthesis.
We present GPU-SLS, a GPU-parallelized framework for safe, robust nonlinear model predictive control (MPC) that scales to high-dimensional uncertain robotic systems and long planning horizons. Our method jointly optimizes an inequality-constrained, dynamically-feasible nominal trajectory, a tracking controller, and a closed-loop reachable set under disturbance, all in real-time. To efficiently compute nominal trajectories, we develop a sequential quadratic programming procedure with a novel GPU-accelerated quadratic program (QP) solver that uses parallel associative scans and adaptive caching within an alternating direction method of multipliers (ADMM) framework. The same GPU QP backend is used to optimize robust tracking controllers and closed-loop reachable sets via system level synthesis (SLS), enabling reachability-constrained control in both fixed- and receding-horizon settings. We achieve substantial performance gains, reducing nominal trajectory solve times by 97.7% relative to state-of-the-art CPU solvers and 71.8% compared to GPU solvers, while accelerating SLS-based control and reachability by 237x. Despite large problem scales, our method achieves 100% empirical safety, unlike high-dimensional learning-based reachability baselines. We validate our approach on complex nonlinear systems, including whole-body quadrupeds (61D) and humanoids (75D), synthesizing robust control policies online on the GPU in 20 milliseconds on average and scaling to problems with 2 x 10^5 decision variables and 8 x 10^4 constraints. The implementation of our method is available at https://github.com/Jeff300fang/gpu_sls.