mlsanityData to 14 Oct 2026

Back to Parsimonious Latents: Learning Task-Centric World Models from Visual Foundations

arXiv 2605.25620 ↗First posted 25 May 2026cs.AIcited by 0

On the map: World models → Latent predictive world models

Abstract (arXiv)

World models enable agents to predict future dynamics conditioned on actions, making the choice of latent representation central to planning and control. Such representations are often either learned directly from pixels with limited semantic structure or inherited from frozen visual foundation models with excessive task-irrelevant detail, yielding state spaces that are poorly matched to downstream planning and control. This is especially challenging in reward-free offline settings, where the model must learn from fixed trajectories without reward supervision or online interaction. To address this, we propose TC-WM, a framework for turning foundation-model embeddings into compact, task-sufficient world representations. The key design is to treat the pretrained embedding space as a semantic scaffold rather than as the final state space: TC-WM linearly projects high-dimensional visual embeddings into a compact latent as the dynamic space, aligns a subspace with the agent's physical state via contrastive learning, and reconstructs embeddings to preserve useful visual structure. This combines the generality of foundation features with the controllability of task-centric dynamics. Theoretically, we show that TC-WM suffices to identify the underlying task-centric latent factors up to a simple transformation. Empirically, TC-WM enables test-time planning across diverse environments (e.g., Robomimic and D4RL), achieving better world-modeling quality and more precise control than state-of-the-art approaches.

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 57 of its 71 references that are in the map. First posted on arXiv 25 May 2026.

Builds on

57 of 71 references.

Closest ancestors · 5

Same direction, newest first

  1. 2026DreamSAC: Learning Hamiltonian World Models via Symmetry Exploration8 Mar
  2. 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planning7 Nov
  3. 2024Revisiting Feature Prediction for Learning Visual Representations from Video15 Feb
  4. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture19 Jan
  5. 2022Temporal Difference Learning for Model Predictive Control9 Mar

Other references · 52

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  4. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  5. 2023Visual Instruction Tuningcited by 7,869
  6. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  7. 2022Scalable Diffusion Models with Transformerscited by 6,054
  8. 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
  9. 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
  10. 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
  11. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
  12. 2018World Modelscited by 1,513
38 more
  1. 2023Align your Latents: High-Resolution Video Synthesis with Latent Diffusion Modelscited by 1,476
  2. 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,427
  3. 2025DINOv3cited by 1,414
  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. 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,194
  7. 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
  8. 2020Mastering Atari with Discrete World Modelscited by 1,007
  9. 2018DeepMind Control Suitecited by 956
  10. 2020Object-Centric Learning with Slot Attentioncited by 876
  11. 2025Cosmos World Foundation Model Platform for Physical AIcited by 864
  12. 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 809
  13. 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 725
  14. 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
  15. 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
  16. 2020robosuite: A Modular Simulation Framework and Benchmark for Robot Learningcited by 511
  17. 2019Variational Autoencoders and Nonlinear ICA: A Unifying Frameworkcited by 503
  18. 2020Planning to Explore via Self-Supervised World Modelscited by 368
  19. 2019Multi-Object Representation Learning with Iterative Variational Inferencecited by 360
  20. 2024Diffusion for World Modeling: Visual Details Matter in Ataricited by 300
  21. 2025Diffusion Transformers with Representation Autoencoderscited by 295
  22. 2016Deep Visual Foresight for Planning Robot Motioncited by 285
  23. 2015Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Imagescited by 274
  24. 2021Contrastive Learning Inverts the Data Generating Processcited by 221
  25. 2018RoboTurk: A Crowdsourcing Platform for Robotic Skill Learning through Imitationcited by 204
  26. 2018State Representation Learning for Control: An Overviewcited by 127
  27. 2018Zero-Shot Visual Imitationcited by 117
  28. 2020On Linear Identifiability of Learned Representationscited by 105
  29. 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 103
  30. 2017Information Theoretic Model Predictive Control: Theory and Applications to Autonomous Drivingcited by 90
  31. 2025Back to the Features: DINO as a Foundation for Video World Modelscited by 58
  32. 2017Linking connectivity, dynamics and computations in low-rank recurrent neural networkscited by 51
  33. 2021People construct simplified mental representations to plancited by 47
  34. 2025World-in-World: World Models in a Closed-Loop Worldcited by 47
  35. 2025Latent Diffusion Planning for Imitation Learningcited by 26
  36. 2025Do What You Say: Steering Vision-Language-Action Models via Runtime Reasoning-Action Alignment Verificationcited by 26
  37. 2023Neural Foundations of Mental Simulation: Future Prediction of Latent Representations on Dynamic Scenescited by 13
  38. 2025Learning Massively Multitask World Models for Continuous Controlcited by 12

50 most cited of 52 All 52 (JSON)

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Learning Task-Sufficient World Models by Synergizing Agentic Exploration and Structured Modeling
  2. 2026Sensorimotor World Models: Perception for Action via Inverse Dynamics
  3. 2026Beyond a single latent space: a dual-latent world model for long-horizon planning
  4. 2026Hierarchical Planning with Latent World Models
  5. 2020Planning from Pixels using Inverse Dynamics Models
  6. 2026Representation World Model: Learning States, Transition and Executable Plans in Representation
  7. 2026Contrastive World Models
  8. 2026AcrossWAM1.0:A Modular Latent World-Action Stack for Compact Robot Policies