Learning Visual Feature-Based World Models via Residual Latent Action
On the map: World models → Embodied and robot world models
also in 1 more
Abstract (arXiv)
World models predict future transitions from observations and actions. Existing works predominantly focus on image generation only. Visual feature-based world models, on the other hand, predict future visual features instead of raw video pixels, offering a promising alternative that is more efficient and less prone to hallucination. However, current feature-based approaches rely on direct regression, which leads to blurry or collapsed predictions in complex interactions, while generative modeling in high-dimensional feature spaces still remains challenging. In this work, we discover that a new type of latent action representation, which we refer to as Residual Latent Action (RLA), can be easily learned from DINO residuals. We also show that RLA is predictive, generalizable, and encodes temporal progression. Building on RLA, we propose RLA World Model (RLA-WM), which predicts RLA values via flow matching. RLA-WM outperforms both state-of-the-art feature-based and video-diffusion world models on simulation and real-world datasets, while being orders of magnitude faster than video diffusion. Furthermore, we develop two robot learning techniques that use RLA-WM to improve policy learning. The first one is a minimalist world action model with RLA that learns from actionless videos, and improves VLA on LIBERO and real robot. The second one is a visual RL framework trained entirely inside a world model learned from offline videos only, using a video-aligned reward and no online interactions. Project page: https://mlzxy.github.io/rla-wm
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Embodied and robot world models, one of 11 directions in World models. Builds on 71 of its 78 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
71 of 78 references.
Closest ancestors · 20
Same direction, newest first
- 2026Fast-WAM: Do World Action Models Need Test-time Future Imagination?17 Mar
- 2026Interactive World Simulator for Robot Policy Training and Evaluation9 Mar
- 2026PointWorld: Scaling 3D World Models for In-The-Wild Robotic Manipulation7 Jan
- 2025Motus: A Unified Latent Action World Model15 Dec
- 2025Semantic World Models22 Oct
- 2025GigaBrain-0: A World Model-Powered Vision-Language-Action Model22 Oct
- 2025Latent Action Pretraining Through World Modeling22 Sep
- 2025Planning with Reasoning using Vision Language World Model2 Sep
- 2025GWM: Towards Scalable Gaussian World Models for Robotic Manipulation25 Aug
- 2025ParticleFormer: A 3D Point Cloud World Model for Multi-Object, Multi-Material Robotic Manipulation29 Jun
- 2025WorldVLA: Towards Autoregressive Action World Model26 Jun
- 2025ManiGaussian++: General Robotic Bimanual Manipulation with Hierarchical Gaussian World Model24 Jun
8 more
- 2025GAF: Gaussian Action Field as a 4D Representation for Dynamic World Modeling in Robotic Manipulation17 Jun
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planning11 Jun
- 2025Sparse Imagination for Efficient Visual World Model Planning2 Jun
- 2025FLARE: Robot Learning with Implicit World Modeling21 May
- 2025AdaWorld: Learning Adaptable World Models with Latent Actions24 Mar
- 20243D-VLA: A 3D Vision-Language-Action Generative World Model14 Mar
- 2023Structured World Models from Human Videos21 Aug
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusion7 Mar
Other references · 51
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,177
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2017Neural Discrete Representation Learningcited by 4,712
- 2015Continuous control with deep reinforcement learningcited by 3,407
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,920
- 2018World Modelscited by 1,513
38 more
- 2025DINOv3cited by 1,414
- 2023LIBERO: Benchmarking Knowledge Transfer for Lifelong Robot Learningcited by 1,410
- 2023Mastering Diverse Domains through World Modelscited by 1,253
- 2024Structured 3D Latents for Scalable and Versatile 3D Generationcited by 770
- 2017Deep Reinforcement Learning that Matterscited by 754
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 512
- 2023DynamiCrafter: Animating Open-domain Images with Video Diffusion Priorscited by 471
- 2023Learning Interactive Real-World Simulatorscited by 458
- 2025UniVLA: Learning to Act Anywhere with Task-centric Latent Actionscited by 410
- 2022Temporal Difference Learning for Model Predictive Controlcited by 397
- 2024DINO-WM: World Models on Pre-trained Visual Features enable Zero-shot Planningcited by 386
- 2024Latent Action Pretraining from Videoscited by 321
- 2025Diffusion Transformers with Representation Autoencoderscited by 295
- 2021ManiSkill: Generalizable Manipulation Skill Benchmark with Large-Scale Demonstrationscited by 197
- 2025Open-Sora 2.0: Training a Commercial-Level Video Generation Model in 200kcited by 163
- 2023Video Language Planningcited by 162
- 2023Diffusion Schr"odinger Bridge Matchingcited by 153
- 2025VLA-Adapter: An Effective Paradigm for Tiny-Scale Vision-Language-Action Modelcited by 145
- 2025Training Agents Inside of Scalable World Modelscited by 136
- 2024Cameras as Rays: Pose Estimation via Ray Diffusioncited by 131
- 2026V-JEPA 2.1: Unlocking Dense Features in Video Self-Supervised Learningcited by 117
- 2025Reinforcement Learning with Action Chunkingcited by 107
- 2025Geometry Forcing: Marrying Video Diffusion and 3D Representation for Consistent World Modelingcited by 96
- 2025Vid2World: Crafting Video Diffusion Models to Interactive World Modelscited by 76
- 2018Simple random search provides a competitive approach to reinforcement learningcited by 75
- 2024Is Behavior Cloning All You Need? Understanding Horizon in Imitation Learningcited by 74
- 2022CROM: Continuous Reduced-Order Modeling of PDEs Using Implicit Neural Representationscited by 58
- 2025Back to the Features: DINO as a Foundation for Video World Modelscited by 58
- 2025Video-Bench: Human-Aligned Video Generation Benchmarkcited by 57
- 2025Robotic World Model: A Neural Network Simulator for Robust Policy Optimization in Roboticscited by 51
- 2024Schr"odinger Bridge Flow for Unpaired Data Translationcited by 47
- 2025RLVR-World: Training World Models with Reinforcement Learningcited by 40
- 2025What Do Latent Action Models Actually Learn?cited by 32
- 2023Gradient-based Planning with World Modelscited by 19
- 2024TOP-ERL: Transformer-based Off-Policy Episodic Reinforcement Learningcited by 19
- 2024PWM: Policy Learning with Multi-Task World Modelscited by 17
- 2025Precise Action-to-Video Generation Through Visual Action Promptscited by 15
- 2026A Frame is Worth One Token: Efficient Generative World Modeling with Delta Tokenscited by 10
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026SANTS: A State-Adaptive Scheduler for World Action Modelscited by 7
- 2026Rethinking Representations for World-Action Modelingcited by 1
- 2026OneLA: Learning Many Worlds from Latent Actionscited by 0
- 2026VITAL: From Visual Motion to Precise Actioncited by 0
- 2026VITAL: From Visual Motion to Precise Action14 Oct 2026
- 2026OneLA: Learning Many Worlds from Latent Actions14 Oct 2026
- 2026Rethinking Representations for World-Action Modeling29 Sep 2026
- 2026SANTS: A State-Adaptive Scheduler for World Action Models27 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026ResWM: Residual-Action World Model for Visual RL
- 2026DreamWAM: Beyond RGB Future Prediction for World Action Models
- 2026Foresight Without Seeing: Latent Futures for World Action Models
- 2026Chain of World: World Model Thinking in Latent Motion
- 2026World Tokens: Enhancing Embodied Policies with Training-Time World Modeling
- 2026What Makes Video World Model Latents Action-Relevant: Prediction over Reconstruction
- 2026PLaW-VLA: Predictive Latent World Modeling for Vision-Language-Action Policies
- 2026LaWAM: Latent World Action Models for Efficient Dynamics-Aware Robot Policies