Pixels to Proofs: Probabilistically-Safe Latent World Model Control via Parallel Conformal Robust MPC
About
We present SLS^2, a framework for safe feedback motion planning from pixels using robust model predictive control (MPC) in learned latent world models. Our approach trains an action-conditioned joint-embedding world model with compact Markovian latent states, enabling efficient gradient-based trajectory optimization through learned latent dynamics. To enforce safety for the true system despite imperfect latent predictions, we inform a GPU-accelerated system level synthesis (SLS) robust MPC scheme with conformal prediction to obtain calibrated latent error bounds and robust latent-space constraint sets. We further learn and conformalize a latent constraint checker, allowing the SLS planner to impose probabilistic safety constraints during closed-loop execution. We evaluate our method on vision-based control tasks, where it improves both goal-reaching performance and safety over latent world-model and safe-planning baselines.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Robust task completion | reacher | Safety Rate100 | 6 | |
| Robust task completion | OGBench Cube | Safety94.29 | 6 | |
| Robust task completion | Rope | Safety100 | 6 | |
| Nominal Planning | Reacher held-out trajectories 57 (test) | Success Rate83.5 | 4 | |
| Nominal Planning | Bimanual Rope Manipulation held-out trajectories (test) | Success Rate93.75 | 4 | |
| Nominal Planning | OGBench Cube held-out trajectories 58 (test) | Success Rate91.5 | 4 | |
| Nominal Planning | Push-T held-out trajectories 10 (test) | Success Rate51.54 | 4 |