mlsanityData to 14 Oct 2026

Learning Counterfactual World Models for Embodied Reasoning under Partial Observability

arXiv 2609.05834 ↗First posted 5 Sep 2026cs.AIcited by 0

On the map: World models → Latent predictive world models

Abstract (arXiv)

World models promise a general route to embodied intelligence: learn predictive dynamics once, then reason, plan, and act with them. Increasingly, the representations beneath such models are pretrained on large-scale video, interaction, and multimodal corpora, which raises a question prediction quality alone cannot answer: when is a learned representation actually actionable? We identify a failure mode we call counterfactual collapse: a model predicts visually plausible futures while failing to distinguish interventions with different behavioral consequences. This arises whenever a representation is optimized for perceptual similarity rather than intervention structure, which is precisely the objective under which most large-scale pretrained encoders are learned. We introduce Counterfactual Latent World Models (CLWM), which combine a recurrent belief-state encoder, action-conditioned latent dynamics, and a contrastive counterfactual objective that separates futures induced by distinct interventions even when their observations look alike. Across occluded manipulation, aliased navigation, and long-horizon manipulation, CLWM improves planning success over the strongest baseline (65.1% → 74.6% on Occluded Push and 67.3% → 78.9% on Aliased Maze) and reduces exploitative planning failures (18.4% → 9.7% on Deferred Kitchen), with ablations attributing the gains to hard counterfactual negatives, especially perceptual-alias negatives. Finally, our counterfactual separability metric, which tracks planning success across the five baseline model classes (r ≥ 0.94), is representation-agnostic: given intervention-outcome labels, it can audit any encoder, pretrained or trained from scratch, before a planner trusts it. We do not yet measure it on large-scale pretrained encoders. Here we establish the metric and its relationship to planning success for world models trained from scratch.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 16 of its 19 references that are in the map. First posted on arXiv 5 Sep 2026.

Builds on

16 of 19 references.

Closest ancestors · 1

Same direction, newest first

  1. 2018Learning Latent Dynamics for Planning from Pixels12 Nov

Other references · 15

  1. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  2. 2020Supervised Contrastive Learningcited by 2,829
  3. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
  4. 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
  5. 2023Mastering Diverse Domains through World Modelscited by 1,253
  6. 2018Recurrent World Models Facilitate Policy Evolutioncited by 785
  7. 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 773
  8. 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 760
  9. 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
  10. 2020Learning Invariant Representations for Reinforcement Learning without Reconstructioncited by 436
  11. 2022DayDreamer: World Models for Physical Robot Learningcited by 400
  12. 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 207
3 more
  1. 2020The Value Equivalence Principle for Model-Based Reinforcement Learningcited by 94
  2. 2019Causal Reasoning from Meta-reinforcement Learningcited by 53
  3. 2020Causal World Models by Unsupervised Deconfounding of Physical Dynamicscited by 15

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Counterfactual World Models via Digital Twin-conditioned Video Diffusion
  2. 2026ω-EVA: Envision, Verify, and Act with Latent Interactive World Models
  3. 2026Towards Predictive, Aligned, and Scalable Robot Learning
  4. 2026World Models for Embodied Intelligence: From Plausible to Controllable to Actionable
  5. 2025ENACT: Evaluating Embodied Cognition with World Modeling of Egocentric Interaction
  6. 2025Embodied-Reasoner: Synergizing Visual Search, Reasoning, and Action for Embodied Interactive Tasks
  7. 2026Dreaming when Necessary: Advancing World Action Models with Adaptive Multi-Modal Reasoning
  8. 2026Sensorimotor World Models: Perception for Action via Inverse Dynamics