arXiv · 2602.00440
DISK: Dynamic Inference SKipping for World Models
Abstract
We present DISK, a training-free adaptive inference method for autoregressive world models. DISK coordinates two coupled diffusion transformers for video and ego-trajectory via dual-branch controllers with cross-modal skip decisions, preserving motion-appearance consistency without retraining. We extend higher-order latent-difference skip testing to the autoregressive chain-of-forward regime and propagate controller statistics through rollout loops for long-horizon stability. When integrated into closed-loop driving rollouts on 1500 NuPlan and NuScenes samples using an NVIDIA L40S GPU, DISK achieves 2x speedup on trajectory diffusion and 1.6x speedup on video diffusion while maintaining L2 planning error, visual quality (FID/FVD), and NAVSIM PDMS scores, demonstrating practical long-horizon video-and-trajectory prediction at substantially reduced cost.
Explore related subjects
Keep this discovery
Anugunj Naman, Gaibo Zhang, Ayushman Singh, Yaguang Zhang. 2026-01-31. DISK: Dynamic Inference SKipping for World Models. https://arxiv.org/abs/2602.00440
Cite the original work for its findings. Save a collection to share your selection of sources.