English

Sparse Forcing: Native Trainable Sparse Attention for Real-time Autoregressive Diffusion Video Generation

Computer Vision and Pattern Recognition 2026-04-24 v1 Machine Learning

Abstract

We introduce Sparse Forcing, a training-and-inference paradigm for autoregressive video diffusion models that improves long-horizon generation quality while reducing decoding latency. Sparse Forcing is motivated by an empirical observation in autoregressive diffusion rollouts: attention concentrates on a persistent subset of salient visual blocks, forming an implicit spatiotemporal memory in the KV cache, and exhibits a locally structured block-sparse pattern within sliding windows. Building on this observation, we propose a trainable native sparsity mechanism that learns to compress, preserve, and update these persistent blocks while restricting computation within each local window to a dynamically selected local neighborhood. To make the approach practical at scale for both training and inference, we further propose Persistent Block-Sparse Attention (PBSA), an efficient GPU kernel that accelerates sparse attention and memory updates for low-latency, memory-efficient decoding. Experiments show that Sparse Forcing improves the VBench score by +0.26 over Self-Forcing on 5-second text-to-video generation while delivering a 1.11-1.17x decoding speedup and 42% lower peak KV-cache footprint. The gains are more pronounced on longer-horizon rollouts, delivering improved visual quality with +0.68 and +2.74 VBench improvements, and 1.22x and 1.27x speedups on 20-second and 1-minute generations, respectively.

Keywords

Cite

@article{arxiv.2604.21221,
  title  = {Sparse Forcing: Native Trainable Sparse Attention for Real-time Autoregressive Diffusion Video Generation},
  author = {Boxun Xu and Yuming Du and Zichang Liu and Siyu Yang and Ziyang Jiang and Siqi Yan and Rajasi Saha and Albert Pumarola and Wenchen Wang and Peng Li},
  journal= {arXiv preprint arXiv:2604.21221},
  year   = {2026}
}
R2 v1 2026-07-01T12:31:47.401Z