Back to Parsimonious Latents: Learning Task-Centric World Models from Visual Foundations
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
- 2026DreamSAC: Learning Hamiltonian World Models via Symmetry Exploration8 Mar
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planning7 Nov
- 2024Revisiting Feature Prediction for Learning Visual Representations from Video15 Feb
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture19 Jan
- 2022Temporal Difference Learning for Model Predictive Control9 Mar
Other references · 52
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2023Visual Instruction Tuningcited by 7,869
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
- 2018World Modelscited by 1,513
38 more
- 2023Align your Latents: High-Resolution Video Synthesis with Latent Diffusion Modelscited by 1,476
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,427
- 2025DINOv3cited by 1,414
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,353
- 2023Mastering Diverse Domains through World Modelscited by 1,253
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,194
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2020Mastering Atari with Discrete World Modelscited by 1,007
- 2018DeepMind Control Suitecited by 956
- 2020Object-Centric Learning with Slot Attentioncited by 876
- 2025Cosmos World Foundation Model Platform for Physical AIcited by 864
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 809
- 2021What Matters in Learning from Offline Human Demonstrations for Robot Manipulationcited by 725
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 703
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
- 2020robosuite: A Modular Simulation Framework and Benchmark for Robot Learningcited by 511
- 2019Variational Autoencoders and Nonlinear ICA: A Unifying Frameworkcited by 503
- 2020Planning to Explore via Self-Supervised World Modelscited by 368
- 2019Multi-Object Representation Learning with Iterative Variational Inferencecited by 360
- 2024Diffusion for World Modeling: Visual Details Matter in Ataricited by 300
- 2025Diffusion Transformers with Representation Autoencoderscited by 295
- 2016Deep Visual Foresight for Planning Robot Motioncited by 285
- 2015Embed to Control: A Locally Linear Latent Dynamics Model for Control from Raw Imagescited by 274
- 2021Contrastive Learning Inverts the Data Generating Processcited by 221
- 2018RoboTurk: A Crowdsourcing Platform for Robotic Skill Learning through Imitationcited by 204
- 2018State Representation Learning for Control: An Overviewcited by 127
- 2018Zero-Shot Visual Imitationcited by 117
- 2020On Linear Identifiability of Learned Representationscited by 105
- 2023Toward General-Purpose Robots via Foundation Models: A Survey and Meta-Analysiscited by 103
- 2017Information Theoretic Model Predictive Control: Theory and Applications to Autonomous Drivingcited by 90
- 2025Back to the Features: DINO as a Foundation for Video World Modelscited by 58
- 2017Linking connectivity, dynamics and computations in low-rank recurrent neural networkscited by 51
- 2021People construct simplified mental representations to plancited by 47
- 2025World-in-World: World Models in a Closed-Loop Worldcited by 47
- 2025Latent Diffusion Planning for Imitation Learningcited by 26
- 2025Do What You Say: Steering Vision-Language-Action Models via Runtime Reasoning-Action Alignment Verificationcited by 26
- 2023Neural Foundations of Mental Simulation: Future Prediction of Latent Representations on Dynamic Scenescited by 13
- 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
- 2026Learning Task-Sufficient World Models by Synergizing Agentic Exploration and Structured Modeling
- 2026Sensorimotor World Models: Perception for Action via Inverse Dynamics
- 2026Beyond a single latent space: a dual-latent world model for long-horizon planning
- 2026Hierarchical Planning with Latent World Models
- 2020Planning from Pixels using Inverse Dynamics Models
- 2026Representation World Model: Learning States, Transition and Executable Plans in Representation
- 2026Contrastive World Models
- 2026AcrossWAM1.0:A Modular Latent World-Action Stack for Compact Robot Policies