Causal-rCM: A Unified Teacher-Forcing and Self-Forcing Open Recipe for Autoregressive Diffusion Distillation in Streaming Video Generation and Interactive World Models
TLDR
Extends rCM diffusion distillation to autoregressive video diffusion, achieving SOTA streaming video generation and interactive world models via teacher-forcing and self-forcing.
Reasoning
The paper presents a novel unified recipe combining teacher-forcing and self-forcing for diffusion distillation in autoregressive video generation, achieving state-of-the-art performance and 10x faster convergence. However, it relies solely on synthetic data, lacking real-world validation, and the abstract cuts off, leaving some details unclear.
Read-first score
Read-first score 59.5, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 54.
Field roles
Rank sensitivity
Stability: volatile; rank range: 504.
Keyword Scores
Deep Analysis
Innovations
- Extending rCM, an advanced diffusion distillation framework, to autoregressive video diffusion.
- Showing that teacher-forcing CM is the best complement to self-forcing DMD as an initialization strategy.
- First implementation of teacher-forcing-based continuous-time consistency models (sCM/MeanFlow) for autoregressive video diffusion, enabled by a custom-mask FlashAttention-2 JVP kernel, achieving 10x faster convergence compared to discrete-time CMs.
- Introducing Causal-rCM, a unified, scalable algorithm-infrastructure open recipe for diffusion distillation and causal training.
- Achieving state-of-the-art streaming video generation performance in both frame-wise and chunk-wise settings using only synthetic data.
- Applying Causal-rCM to Cosmos 3, an advanced omnimodal world foundation model, enabling an interactive world model.
Methodology
The paper extends the rCM diffusion distillation framework to autoregressive video diffusion, leveraging teacher-forcing (offline forward-divergence causal training) and self-forcing (on-policy reverse-divergence refinement). It implements continuous-time consistency models (sCM/MeanFlow) using a custom-mask FlashAttention-2 JVP kernel for efficient training. The method is evaluated on streaming video generation (frame-wise and chunk-wise) and action-conditioned world models, using only synthetic data for training.
Key Results
The distilled 2-step causal Wan2.1-1.3B model achieves a VBench-T2V score of 84.63 with only 1 or 2 sampling steps, achieving state-of-the-art streaming video generation performance.