English

HASTE: Training-Free Video Diffusion Acceleration via Head-Wise Adaptive Sparse Attention

Computer Vision and Pattern Recognition 2026-05-15 v1 Artificial Intelligence

Abstract

Diffusion-based video generation has advanced substantially in visual fidelity and temporal coherence, but practical deployment remains limited by the quadratic complexity of full attention. Training-free sparse attention is attractive because it accelerates pretrained models without retraining, yet existing online top-pp sparse attention still spends non-negligible cost on mask prediction and applies shared thresholds despite strong head-level heterogeneity. We show that these two overlooked factors limit the practical speed-quality trade-off of training-free sparse attention in Video DiTs. To address them, we introduce a head-wise adaptive framework with two plug-in components: Temporal Mask Reuse, which skips unnecessary mask prediction based on query-key drift, and Error-guided Budgeted Calibration, which assigns per-head top-pp thresholds by minimizing measured model-output error under a global sparsity budget. On Wan2.1-1.3B and Wan2.1-14B, our method consistently improves XAttention and SVG2, achieving up to 1.93 times speedup at 720P while maintaining competitive video quality and similarity metrics.

Keywords

Cite

@article{arxiv.2605.14513,
  title  = {HASTE: Training-Free Video Diffusion Acceleration via Head-Wise Adaptive Sparse Attention},
  author = {Xuzhe Zheng and Yuexiao Ma and Jing Xu and Xiawu Zheng and Rongrong Ji and Fei Chao},
  journal= {arXiv preprint arXiv:2605.14513},
  year   = {2026}
}
R2 v1 2026-07-22T07:11:50.377Z