mlsanityData to 14 Oct 2026

ATM: Action-Consistency Transfer Matrix for Diagnosing and Improving Latent World Models

arXiv 2606.09028 ↗First posted 8 Jun 2026cs.CV cs.AI cs.ROcited by 3

On the map: World models → Latent predictive world models

Abstract (arXiv)

Latent world models can achieve accurate latent prediction yet still differ substantially in downstream planning performance. We argue that a key source of this discrepancy lies in the structure of action-induced latent transitions. We formalize action-identifiability through Bayes inverse risk, characterizing how much uncertainty about an action remains after observing the transition it induces. Model-predicted transitions can become highly self-decodable while encoding a domain-specific action relationship that fails to transfer to real environment transitions. We characterize this mismatch through cross-domain inverse transfer, instantiated as the Action-Consistency Transfer Matrix (ATM), a 2×2 inverse-risk matrix over real and predicted transition domains. Across TwoRoom, PushT, and OGBench-Cube, true-transition action-identifiability tracks downstream planning substantially better than standard prediction loss, while the full ATM reveals highly self-decodable yet cross-domain-inconsistent predicted transitions. Controlled interventions on the two domains further produce distinct transition structures and planning outcomes, supporting this decomposition. The same diagnostics also support lightweight model screening, reaching 98.81% pairwise ranking accuracy for candidates separated by more than 5% success.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 14 of its 26 references that are in the map. First posted on arXiv 8 Jun 2026.

Builds on

14 of 26 references.

Closest ancestors · 3

Same direction, newest first

  1. 2026World Model Control by Trajectory Reachability Metrics21 May
  2. 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels13 Mar
  3. 2018Learning Latent Dynamics for Planning from Pixels12 Nov

Other references · 11

  1. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
  2. 2018World Modelscited by 1,513
  3. 2023Mastering Diverse Domains through World Modelscited by 1,253
  4. 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
  5. 2025Learning from Reward-Free Offline Data: A Case for Planning with Latent Dynamics Modelscited by 89
  6. 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitationcited by 31
  7. 2026World Action Verifier: Self-Improving World Models via Forward-Inverse Asymmetrycited by 18
  8. 2026Is the Future Compatible? Diagnosing Dynamic Consistency in World Action Modelscited by 11
  9. 2026STARRY: Spatial-Temporal Action-Centric World Modeling for Robotic Manipulationcited by 10
  10. 2026Enhancing Policy Learning with World-Action Modelcited by 6
  11. 2026World Models as Group Actionscited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026ActSWM: Action-Sensitive World Models for Long-Horizon Planning in Open-World Games
  2. 2026Planning in 8 Tokens: A Compact Discrete Tokenizer for Latent World Model
  3. 2026Beyond a single latent space: a dual-latent world model for long-horizon planning
  4. 2026SAGE: Subgoal-Conditioned Action Generation for Latent World Model Planning
  5. 2025WorldPrediction: A Benchmark for High-level World Modeling and Long-horizon Procedural Planning
  6. 2026ATLAS: Aligned Transport of Latent Structure for Reliable World Model Planning
  7. 2026One-Step Prediction Fidelity Is a Poor Proxy for Planning Performance in JEPA World Models
  8. 2026Delta-JEPA: Learning Action-Sensitive World Models via Latent Difference Decoding