WorldCache: Accelerating World Models for Free via Heterogeneous Token Caching
On the map: World models → Latent predictive world models
Abstract (arXiv)
Diffusion-based world models have shown strong potential for unified world simulation, but the iterative denoising remains too costly for interactive use and long-horizon rollouts. While feature caching can accelerate inference without training, we find that policies designed for single-modal diffusion transfer poorly to world models due to two world-model-specific obstacles: token heterogeneity from multi-modal coupling and spatial variation, and non-uniform temporal dynamics where a small set of hard tokens drives error growth, making uniform skipping either unstable or overly conservative. We propose WorldCache, a caching framework tailored to diffusion world models. We introduce textitCurvature-guided Heterogeneous Token Prediction, which uses a physics-grounded curvature score to estimate token predictability and applies a Hermite-guided damped predictor for chaotic tokens with abrupt direction changes. We also design textitChaotic-prioritized Adaptive Skipping, which accumulates a curvature-normalized, dimensionless drift signal and recomputes only when bottleneck tokens begin to drift. Experiments on diffusion world models show that WorldCache delivers up to 3.7× end-to-end speedups while maintaining 98% rollout quality, demonstrating the vast advantages and practicality of WorldCache in resource-constrained scenarios. Our code is released in https://github.com/FofGofx/WorldCache.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Latent predictive world models, one of 11 directions in World models. Builds on 40 of its 49 references that are in the map. First posted on arXiv 6 Mar 2026.
Builds on
40 of 49 references.
References · 40
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2022Scalable Diffusion Models with Transformerscited by 5,931
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,598
- 2024HunyuanVideo: A Systematic Framework For Large Video Generative Modelscited by 1,597
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2022Diffusion Models in Vision: A Surveycited by 1,008
- 2020Mastering Atari with Discrete World Modelscited by 990
- 2025Cosmos World Foundation Model Platform for Physical AIcited by 833
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
28 more
- 2024Genie: Generative Interactive Environmentscited by 706
- 2023A Comprehensive Overview of Large Language Modelscited by 639
- 2024Sora: A Review on Background, Technology, Limitations, and Opportunities of Large Vision Modelscited by 467
- 2023DeepCache: Accelerating Diffusion Models for Freecited by 346
- 2024Navigation World Modelscited by 262
- 2024Timestep Embedding Tells: It's Time to Cache for Video Diffusion Modelcited by 245
- 2025Sparse VideoGen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsitycited by 204
- 2025GAIA-2: A Controllable Multi-View Generative World Model for Autonomous Drivingcited by 178
- 2024SageAttention: Accurate 8-Bit Attention for Plug-and-play Inference Accelerationcited by 158
- 2026Advancing Open-source World Modelscited by 144
- 2024Accelerating Diffusion Transformers with Token-wise Feature Cachingcited by 143
- 2025WorldScore: A Unified Evaluation Benchmark for World Generationcited by 143
- 2024Learning-to-Cache: Accelerating Diffusion Transformer via Layer Cachingcited by 138
- 2024FORA: Fast-Forward Caching in Diffusion Transformer Accelerationcited by 127
- 2025From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeerscited by 124
- 2024Adaptive Caching for Faster Video Generation with Diffusion Transformerscited by 100
- 2025Aether: Geometric-Aware Unified World Modelingcited by 94
- 2025Voyager: Long-Range and World-Consistent Video Diffusion for Explorable 3D Scene Generationcited by 93
- 2025Less is Enough: Training-Free Video Diffusion Acceleration via Runtime-Adaptive Cachingcited by 39
- 2024MPQ-DM: Mixed Precision Quantization for Extremely Low Bit Diffusion Modelscited by 20
- 2025Q-VDiT: Towards Accurate Quantization and Distillation of Video-Generation Diffusion Transformerscited by 20
- 2025Compute Only 16 Tokens in One Timestep: Accelerating Diffusion Transformers with Cluster-Driven Feature Cachingcited by 17
- 2025Quantized Visual Geometry Grounded Transformercited by 16
- 2025Accelerating Diffusion Transformer via Increment-Calibrated Caching with Channel-Aware Singular Value Decompositioncited by 12
- 2025MPQ-DMv2: Flexible Residual Mixed Precision Quantization for Low-Bit Diffusion Models with Temporal Distillationcited by 12
- 2025S2Q-VDiT: Accurate Quantized Video Diffusion Transformer with Salient Data and Sparse Token Distillationcited by 12
- 2025HERO: Hierarchical Extrapolation and Refresh for Efficient World Modelscited by 8
- 2025QuantSparse: Comprehensively Compressing Video Diffusion Transformer with Model Quantization and Attention Sparsificationcited by 5
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Video Generation Models as World Models: Efficient Paradigms, Architectures and Algorithmscited by 5
- 2026Multi-Resolution Flow Matching: Training-Free Diffusion Acceleration via Staged Samplingcited by 2
- 2026Sparse-WAM: Accelerating World Action Models via Action-Guided Sparse Imaginationcited by 0
- 2026WAM-Cache: Staleness-Bounded KV Reuse for Efficient World Action Modelscited by 0
- 2026WAM-Cache: Staleness-Bounded KV Reuse for Efficient World Action Models8 Oct 2026
- 2026Sparse-WAM: Accelerating World Action Models via Action-Guided Sparse Imagination30 Sep 2026
- 2026Multi-Resolution Flow Matching: Training-Free Diffusion Acceleration via Staged Sampling2 Jul 2026
- 2026Video Generation Models as World Models: Efficient Paradigms, Architectures and Algorithms30 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2026SpectralCache: Accelerating Diffusion-Based World Models via Spectral Feature Caching
- 2026OTCache: Optimal Transport for Geometry-Aware Caching in Diffusion Models
- 2026OnlineCache: Learning Dynamic Caching Policies with Error Correction for Efficient Diffusion Inference
- 2026WorldDynCache: Risk-Controlled Latent Dynamics Approximation for Diffusion World Model
- 2026WorldCache: Content-Aware Caching for Accelerated Video World Models
- 2025A Survey on Cache Methods in Diffusion Models: Toward Efficient Multi-Modal Generation
- 2025No Cache Left Idle: Accelerating diffusion model via Extreme-slimming Caching
- 2026Budget-Constrained Step-Level Diffusion Caching