English

X-Cache: Cross-Chunk Block Caching for Few-Step Autoregressive World Models Inference

Computer Vision and Pattern Recognition 2026-05-07 v2

Abstract

Real-time world simulation is becoming a key infrastructure for scalable evaluation and online reinforcement learning of autonomous driving systems. Recent driving world models built on autoregressive video diffusion achieve high-fidelity, controllable multi-camera generation, but their inference cost remains a bottleneck for interactive deployment. However, existing diffusion caching methods are designed for offline video generation with multiple denoising steps, and do not transfer to this scenario. Few-step distilled models have no inter-step redundancy left for these methods to reuse, and sequence-level parallelization techniques require future conditioning that closed-loop interactive generation does not provide. We present X-Cache, a training-free acceleration method that caches along a different axis: across consecutive generation chunks rather than across denoising steps. X-Cache maintains per-block residual caches that persist across chunks, and applies a dual-metric gating mechanism over a structure- and action-aware block-input fingerprint to independently decide whether each block should recompute or reuse its cached residual. To prevent approximation errors from permanently contaminating the autoregressive KV cache, X-Cache identifies KV update chunks (the forward passes that write clean keys and values into the persistent cache) and unconditionally forces full computation on these chunks, cutting off error propagation. We implement X-Cache on X-world, a production multi-camera action-conditioned driving world model built on multi-block causal DiT with few-step denoising and rolling KV cache. X-Cache achieves 71% block skip rate with 2.6x wall-clock speedup while maintaining minimum degradation.

Keywords

Cite

@article{arxiv.2604.20289,
  title  = {X-Cache: Cross-Chunk Block Caching for Few-Step Autoregressive World Models Inference},
  author = {Yixiao Zeng and Jianlei Zheng and Chaoda Zheng and Shijia Chen and Mingdian Liu and Tongping Liu and Tengwei Luo and Yu Zhang and Boyang Wang and Linkun Xu and Siyuan Lu and Bo Tian and Xianming Liu},
  journal= {arXiv preprint arXiv:2604.20289},
  year   = {2026}
}

Comments

Technical Report, update demonstration website

R2 v1 2026-07-01T12:29:55.703Z