Attention Sparsity is Input-Stable: Training-Free Sparse Attention for Video Generation via Offline Sparsity Profiling and Online QK Co-Clustering
Organizations: SKLCCSE, School of Computer Science and Engineering, Beihang University · Zhongguancun Academy · School of Computer Science, Peking University · Beihang University ·
Abstract
Diffusion Transformers (DiTs) achieve strong video generation quality but suffer from high inference cost due to dense 3D attention, motivating sparse attention techniques for improving efficiency. However, existing training-free sparse attention methods for video generation still face two unresolved limitations: ignoring layer heterogeneity in attention pruning and ignoring query-key coupling in block partitioning, which hinder a better quality-speedup trade-off. In this work, we uncover a critical insight: attention sparsity is an intrinsic layer-wise property, with only minor variation across different inputs. Motivated by this observation, we propose SVOO, a training-free sparse attention framework for fast video generation via offline layer-wise sparsity profiling and online bidirectional co-clustering. Specifically, SVOO adopts a two-stage paradigm: (i) offline layer-wise sensitivity profiling to derive intrinsic per-layer pruning levels, and (ii) online block-wise sparse attention via a bidirectional co-clustering algorithm. Extensive experiments on seven widely used video generation models demonstrate that SVOO achieves a superior quality-speedup trade-off over state-of-the-art methods, delivering up to 1.93x speedup while maintaining a PSNR of up to 29 dB on Wan2.1. Code is available at: https://github.com/Mutual-Luo/SVOO.
Figures & tables
| Model | Config | Baseline | Quality | Efficiency | |||||||
| PSNR | SSIM | LPIPS | ImageQual | AesQual | SubConsist | BackConsist | Latency | Speedup | |||
| Wan2.1 | 1.3B-720P-T2V | Origin | - | - | - | 66.58% | 64.47% | 96.74% | 97.30% | 417s | 1.00 |
| SpargeAttn | 25.137 | 0.801 | 0.234 | 63.08% | 62.75% | 95.79% | 97.09% | 288s | 1.45 | ||
| SVG1 | 25.712 | 0.811 | 0.230 | 63.26% | 61.14% | 94.27% | 91.16% | 266s | 1.56 | ||
| SVG2 | 29.268 | 0.886 | 0.127 | 61.83% | 60.15% | 95.88% | 96.47% | 241s | 1.73 | ||
| Radial | 26.305 | 0.829 | 0.182 | 63.67% | 62.76% | 96.56% | 97.07% | 257s | 1.62 | ||
| Model | Config | Baseline | Quality | Efficiency | |||||||
| PSNR | SSIM | LPIPS | ImageQual | AesQual | SubConsist | BackConsist | Latency | Speedup | |||
| Wan2.1 | 14B-720P-I2V | Origin | - | - | - | 71.52% | 61.86% | 94.98% | 95.37% | 1658s | 1.00 |
| SpargeAttn | 21.557 | 0.691 | 0.297 | 70.83% | 61.77% | 95.81% | 95.04% | 1124s | 1.48 | ||
| SVG1 | 24.262 | 0.825 | 0.174 | 70.93% | 61.02% | 94.42% | 94.51% | 1047s | 1.58 | ||
| SVG2 | 27.324 | 0.856 | 0.125 | 70.76% | 61.45% | 95.72% | 94.93% | 998s | 1.66 | ||
| Radial | 23.673 | 0.759 | 0.189 | 70.91% | 61.62% | 94.52% | 95.36% | 1046s | 1.58 | ||
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.