Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Devesh Nath, Anutam Srinivasan, Haoran Yin, Ruitong Jiang, Jeffrey Fang, Glen Chou• 2026

Related benchmarks

TaskDatasetResultRank
Robust task completionreacher
Safety Rate100
6
Robust task completionOGBench Cube
Safety94.29
6
Robust task completionRope
Safety100
6
Nominal PlanningReacher held-out trajectories 57 (test)
Success Rate83.5
4
Nominal PlanningBimanual Rope Manipulation held-out trajectories (test)
Success Rate93.75
4
Nominal PlanningOGBench Cube held-out trajectories 58 (test)
Success Rate91.5
4
Nominal PlanningPush-T held-out trajectories 10 (test)
Success Rate51.54
4
Showing 7 of 7 rows

Other info

Follow for update