How Well Do Latent World Models Understand Partially Observable Safety Constraints?
On the map: World models → Latent dynamics models and model-based RL
Abstract (arXiv)
Latent world models are a promising approach for learning state representations and dynamics directly from high-dimensional observations, enabling robot control in hard-to-model settings. However, control performance ultimately depends on the latent representation encoding the required information for the task. In this work, we study latent-space safe control problems and show how partial observability can induce control failures when safety-relevant information is not preserved in the latent state. Specifically, we identify two world model failure modes: estimation gaps, where current observations do not reveal safety-critical quantities (e.g., temperature in a cooking task), and prediction gaps, where failures are observable once they occur but cannot be reliably anticipated from available observations. We introduce two diagnostics for these gaps: a mutual-information-based measure of safety observability and a rollout-based measure of future safety predictability. Finally, we present mitigation strategies for each failure mode: privileged multimodal supervision for estimation gaps and conformal risk calibration for prediction gaps. Across two hardware case studies -- using unimodal RGB world models and multimodal RGB+Tactile and RGB+Thermal variants -- we show that these mitigation strategies improve the safety of a Franka Research 3 manipulator on challenging cooking tasks under partial observability, albeit with increased conservativeness. More broadly, our work raises the question of when world model state representations are sufficient for reliable robot control
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Latent dynamics models and model-based RL, one of 11 directions in World models. Builds on 19 of its 38 references that are in the map. First posted on arXiv 7 Oct 2025.
Builds on
19 of 38 references.
Closest ancestors · 5
Same direction, newest first
- 2025Uncertainty-aware Latent Safety Filters for Avoiding Out-of-Distribution Failures1 May
- 2025Generalizing Safety Beyond Collision-Avoidance via Latent-Space Reachability Analysis2 Feb
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Control25 Oct
- 2019Dream to Control: Learning Behaviors by Latent Imagination3 Dec
- 2018World Models27 Mar
Other references · 14
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2015Continuous control with deep reinforcement learningcited by 3,407
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
- 2016Understanding intermediate layers using linear classifier probescited by 1,353
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,264
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planningcited by 386
- 2017Hamilton-Jacobi Reachability: A Brief Overview and Recent Advancescited by 201
- 2017A General Safety Framework for Learning-Based Control in Uncertain Robotic Systemscited by 163
- 2019Robust Model Predictive Shielding for Safe Reinforcement Learning with Stochastic Dynamicscited by 47
- 2022Guaranteed Discovery of Control-Endogenous Latent States with Multi-Step Inverse Modelscited by 45
- 2021Action-Sufficient State Representation Learning for Control with Structural Constraintscited by 31
- 2020Multimodal Sensor Fusion with Differentiable Filterscited by 29
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026When World Models Lie: Adaptive Safety Analysis Under Wrong Imaginations
- 2024Zero-shot Safety Prediction for Autonomous Robots with Foundation World Models
- 2025Safety Certification in the Latent space using Control Barrier Functions and World Models
- 2026Modeling Latent Disturbances for Robust Decision-Making in World Models
- 2025AnySafe: Adapting Latent Safety Filters at Runtime via Safety Constraint Parameterization in the Latent Space
- 2025How Safe Will I Be Given What I Saw? Calibrated Prediction of Safety Chances for Image-Controlled Autonomy
- 2026Learning from World Feedback: Why Model Uncertainty Fails as a Risk Signal in Model-Based RL
- 2023How Safe Am I Given What I See? Calibrated Prediction of Safety Chances for Image-Controlled Autonomy