On the Identifiability of Controlled World Models
On the map: World models → Latent predictive world models
Abstract (arXiv)
World model serves as a promising tool to infer environment dynamics under high-dimensional observations and candidate actions. Recently, LeCun's JEPA provides a compelling framework for learning such models in representation space. Its action-conditioned extension plays a central role in visual control and latent-space planning, but leaves a fundamental question: can it recover the controlled dynamics from nonlinear observations? This paper presents a joint identifiability condition for controlled world models with Gaussian latent states, which consists of two coupled components: (1) representation identifiability and (2) transition identifiability. The former depends on the spectral separation property while the latter is related to non-degenerate variation of conditional action. We prove that when this condition holds, minimizing the LeJEPA-style predictive objective can recover both latent states and controlled dynamics in the sense of orthogonal transformation. We further prove that the upper bound of transition prediction error is inversely proportional to the spectral separation margin. We also characterize an attainable amplification of counterfactual prediction error that scales inversely with the weakest conditional action-excitation margin. The theoretical predictions are empirically supported across four nonlinear observation settings.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 14 of its 23 references that are in the map. First posted on arXiv 24 Jul 2026.
Builds on
14 of 23 references.
Closest ancestors · 2
Same direction, newest first
- 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixels13 Mar
- 2018Learning Latent Dynamics for Planning from Pixels12 Nov
Other references · 12
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2018World Modelscited by 1,444
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,327
- 2018Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representationscited by 929
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 482
- 2018Nonlinear ICA Using Auxiliary Variables and Generalized Contrastive Learningcited by 278
- 2016Unsupervised Feature Extraction by Time-Contrastive Learning and Nonlinear ICAcited by 270
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 227
- 2023Nonparametric Identifiability of Causal Representations from Unknown Interventionscited by 105
- 2023Identifiability of latent-variable and structural-equation models: from linear to nonlinearcited by 58
- 2023General Identifiability and Achievability for Causal Representation Learningcited by 34
- 2024On the Stability of Datatic Control Systemscited by 2
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026What Can Latent World Models Know? Physical Parameter Identifiability in Multimodal Predictive Representationscited by 3
- 2026Finite-Sample Metric Non-Collapse for Geometrically Supervised Latent World Models in Controlcited by 0
- 2026FIRM-WM: State-factorized factual-interventional recurrent modeling for reward-free visual planningcited by 0
- 2026I Act Therefore I Am: When Is JEPA's Action-Conditioning Enough to Learn Causal Mechanisms?cited by 0
- 2026Adaptive Latent Capacity for World Modelscited by 0
- 2026LRC-JEPA: Disentangling Dynamics and Residual Context for Efficient World Modelscited by 0
- 2026Control-Geometry Straightening for Sampling-Based Latent Planningcited by 0
- 2026Control-Geometry Straightening for Sampling-Based Latent Planning28 Sep 2026
- 2026LRC-JEPA: Disentangling Dynamics and Residual Context for Efficient World Models28 Sep 2026
- 2026Adaptive Latent Capacity for World Models26 Sep 2026
- 2026I Act Therefore I Am: When Is JEPA's Action-Conditioning Enough to Learn Causal Mechanisms?25 Sep 2026
- 2026FIRM-WM: State-factorized factual-interventional recurrent modeling for reward-free visual planning19 Sep 2026
- 2026Finite-Sample Metric Non-Collapse for Geometrically Supervised Latent World Models in Control7 Aug 2026
- 2026What Can Latent World Models Know? Physical Parameter Identifiability in Multimodal Predictive Representations29 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026When Does LeJEPA Learn a World Model?
- 2026No Gaussian Required: Contrastive Inverse Dynamics for JEPA World Models
- 2026World Models with Predictable Long-Horizon Marginals
- 2026Bilinear World Models: Learning Representations with Structured Dynamics for Efficient Control
- 2026Identifiability Without Gaussianity: Symbolic World Models and Near-Infinite Temporal Consistency
- 2026Diagnosing JEPA World Models with Action-Conditioned Predictive Consistency
- 2026JEPA-x: Cross-Predictive Physics Grounding for Forecastable Latent Dynamics
- 2026Hamiltonian JEPA: Action-Conditioned World Models with an Inherited Control State