Keep the Future, Drop the Rollout: RIFT for World Action Models
On the map: World models → Embodied and robot world models
Abstract (arXiv)
World action models (WAMs) condition robot actions on predicted futures, but iterative video rollout increases deployment latency. We ask whether action generation requires the evolving rollout trajectory or only its future representation. Across four WAMs on 40 simulated robotic manipulation tasks, paired closed-loop interventions show that blocking access to the future cache or reassigning its values changes execution and reduces success. Yet in the evaluated co-denoising settings, reusing one fixed final-clean key/value (K/V) cache throughout action denoising nearly preserves unmodified execution, with 1.7--1.9 cm end-effector average displacement error. Obtaining this cache still requires iterative video generation. We therefore propose RIFT (Rollout-free Imagination via Future Tokens), which uses learned anticipation tokens to construct a complete future K/V cache in one backbone pass. On LIBERO, RIFT achieves 98.8% overall success, outperforming all evaluated rollout-based methods while yielding a 3.1--9.2× inference speedup. Without further training, it achieves 81.1% overall success on the out-of-distribution LIBERO-Plus benchmark, a +9.7 percentage-point improvement over the strongest evaluated baseline. On RoboTwin, it achieves 92.9% and 92.6% success on clean and randomized scenes, respectively, the highest among the evaluated methods. On real-world manipulation tasks, RIFT achieves 45.3% average success, a +6.0 percentage-point improvement over Fast-WAM-Joint. These results support rollout-free future conditioning without iterative video generation at deployment.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Embodied and robot world models, one of 11 directions in World models. Builds on 42 of its 50 references that are in the map. First posted on arXiv 12 Aug 2026.
Builds on
42 of 50 references.
Closest ancestors · 12
Same direction, newest first
- 2026Being-H0.7: A Latent World-Action Model from Egocentric Videos30 Apr
- 2026Privileged Foresight Distillation: Zero-Cost Future Correction for World Action Models28 Apr
- 2026World Action Models are Zero-shot Policies17 Feb
- 2026Causal World Modeling for Robot Control29 Jan
- 2025Motus: A Unified Latent Action World Model15 Dec
- 2025RynnVLA-002: A Unified Vision-Language-Action and World Model21 Nov
- 2025Dual-Stream Diffusion for World-Model Augmented Vision-Language-Action Model31 Oct
- 2025FLARE: Robot Learning with Implicit World Modeling21 May
- 2025DreamGen: Unlocking Generalization in Robot Learning through Video World Models19 May
- 2025Unified World Models: Coupling Video and Action Diffusion for Pretraining on Large Robotic Datasets3 Apr
- 2024RoboDreamer: Learning Compositional World Models for Robot Imagination18 Apr
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusion7 Mar
Other references · 30
- 2023RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Controlcited by 2,993
- 2024OpenVLA: An Open-Source Vision-Language-Action Modelcited by 2,777
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,536
- 2024π0: A Vision-Language-Action Flow Model for General Robot Controlcited by 2,297
- 2025π0.5: a Vision-Language-Action Model with Open-World Generalizationcited by 1,653
- 2023LIBERO: Benchmarking Knowledge Transfer for Lifelong Robot Learningcited by 1,338
- 2019Mastering Atari, Go, Chess and Shogi by Planning with a Learned Modelcited by 1,327
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2025GR00T N1: An Open Foundation Model for Generalist Humanoid Robotscited by 1,167
- 2019Attention is not Explanationcited by 916
18 more
- 2024RDT-1B: a Diffusion Foundation Model for Bimanual Manipulationcited by 670
- 2025SmolVLA: A Vision-Language-Action Model for Affordable and Efficient Roboticscited by 507
- 2023Learning Universal Policies via Text-Guided Video Generationcited by 506
- 2023TD-MPC2: Scalable, Robust World Models for Continuous Controlcited by 482
- 2025RoboTwin 2.0: A Scalable Data Generator and Benchmark with Strong Domain Randomization for Robust Bimanual Robotic Manipulationcited by 477
- 2019Learning by Cheatingcited by 361
- 2023Unleashing Large-Scale Video Generative Pre-training for Visual Robot Manipulationcited by 353
- 2024GR-2: A Generative Video-Language-Action Model with Web-Scale Knowledge for Robot Manipulationcited by 290
- 2024Video Prediction Policy: A Generalist Robot Policy with Predictive Visual Representationscited by 279
- 2026Cosmos Policy: Fine-Tuning Video Models for Visuomotor Control and Planningcited by 274
- 2025DreamVLA: A Vision-Language-Action Model Dreamed with Comprehensive World Knowledgecited by 203
- 2025DexVLA: Vision-Language Model with Plug-In Diffusion Expert for General Robot Controlcited by 191
- 2024Gen2Act: Human Video Generation in Novel Scenarios enables Generalizable Robot Manipulationcited by 142
- 2025mimic-video: Video-Action Models for Generalizable Robot Control Beyond VLAscited by 133
- 2025Genie Envisioner: A Unified World Foundation Platform for Robotic Manipulationcited by 129
- 2025RoboCOIN: An Open-Sourced Bimanual Robotic Data Collection for Integrated Manipulationcited by 72
- 2025Vidar: Embodied Video Diffusion Model for Generalist Manipulationcited by 58
- 2026EvoScene-VLA: Evolving Scene Beliefs Inside the Action Decoder for Chunked Robot Controlcited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Is the Future Compatible? Diagnosing Dynamic Consistency in World Action Models
- 2026FutureWorlds: Learning Robotic World Models from Alternative Futures
- 2026Latent Action as Intention Enables Efficient Future Imagination for World Action Models
- 2026Rethink Before You Execute: Adaptive Execution for World Action Models
- 2026Foresight Without Seeing: Latent Futures for World Action Models
- 2026When to Trust Imagination: Adaptive Action Execution for World Action Models
- 2026Faster-WAM: Efficient Inference-Time Future Conditioning for Robust World Action Models
- 2026Persistent Robot World Models: Stabilizing Multi-Step Rollouts via Reinforcement Learning