Predictive Objectives Discard Exogenous Control-Relevant Features: A Controlled Mechanistic Study
Not on the map. Why?
Abstract (arXiv)
Joint-embedding predictive (JEPA-style) objectives learn representations by predicting future latents. In doing so they can discard features that are exogenous (uncontrollable by the agent) yet control-relevant, even when those features are trivially encodable. This occurs because the objective optimizes temporal predictability rather than control-relevance. We isolate this failure mode in a controlled 2x2 experimental design that varies feature controllability and relevance independently, using a predictability knob that decouples a feature's temporal predictability from its control-relevance. Comparing six objectives: reconstruction, JEPA, action-conditioned JEPA, controllability-based JEPA, inverse dynamics under a random policy, and reward-grounded JEPA, we observe that all evaluated reward-free predictive objectives leave the exogenous control-relevant feature near chance accuracy, while a reward-grounded variant retains it selectively. The remedy is label-efficient and robust: as little as 2% of reward-labeled transitions recovers the feature, the effect holds across two environments with different surface forms, and it persists across latent dimensions from 16 to 1024. Comparing the learned latent geometry against bisimulation theory's prediction, the JEPA latent realizes only a small fraction of the class separation a supervised reference attains.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 12 of its 14 references that are in the map. First posted on arXiv 29 Jun 2026.
Builds on
12 of 14 references.
References · 12
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,039
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 809
- 2020Learning Invariant Representations for Reinforcement Learning without Reconstructioncited by 436
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planningcited by 386
- 2019DeepMDP: Learning Continuous Latent Space Models for Representation Learningcited by 207
- 2025Critique of World Modelcited by 25
- 2026Sensorimotor World Models: Perception for Action via Inverse Dynamicscited by 20
- 2026Learning Invariant Visual Representations for Planning with Joint-Embedding Predictive World Modelscited by 13
- 2024Learning Action-based Representations Using Invariancecited by 7
Built on by
Citing papers per year: 2026: 2.
2 papers.