视频生成稀疏计算¶
介绍注意力稀疏化、动态选择及稀疏算子的适用条件。
前置知识: 注意力机制、GPU 计算。
使用步骤: 区分理论稀疏率与实际执行结构 → 选择静态或动态稀疏模式 → 评估选择开销、运行速度和条件遵循。
稀疏计算概述¶
视频 Diffusion Transformer(DiT)将时间、空间和条件信息编码为长序列。当 token 数为 \(N\)、隐维为 \(d\) 时,密集自注意力中注意力分数的主要计算量随 \(N^2d\) 增长,线性投影与 FFN 的计算复杂度则为 \(O(Nd^2)\)。现有方法主要从两个方面降低计算量:一是减少 token 数量,从而同时降低注意力、MLP 和激活的开销;二是减少注意力连边,仅计算部分 token 对之间的交互。
近期研究主要关注三个相互关联的问题:如何识别必要的 token 或连接,如何将稀疏模式组织为适合 GPU 执行的布局,以及如何根据提示词、网络层、注意力头和去噪时间步动态调整稀疏模式。
1. 基本概念¶
| 对象 | 主要变化 | 参数量是否变化 | 是否属于本章重点 |
|---|---|---|---|
| weight / channel / head / layer pruning | 权重或网络结构 | 是或形成稀疏参数 | 否,见模型剪枝综述 |
| token merge / prune | 运行时序列长度 \(N\) | 否 | 是 |
| attention edge sparsity | \(QK^\top\) 中实际参与计算的 token 对 | 否 | 是 |
| linear / recurrent mixer | 以线性或状态机制替代原有交互算子 | 可能需预训练或后训练 | 作为相关方向讨论 |
不同方法需要采用相应的评测指标。Token 缩减方法应报告实际序列长度 \(N\)、合并位置以及选择、聚合和分散操作的开销;注意力稀疏方法则应报告实际稠密度、分块大小、检索或重排成本和所用稀疏算子。理论 FLOPs 的下降并不足以说明实际运行时延随之降低。
2. 方法分类¶
2.1 Token Merge / Prune:缩短运行时序列¶
Token merge/prune 方法首先估计可合并的空间或时间单元,再在部分 layer 中使用较短序列进行计算。ADAPTOR 根据局部运动信号自适应地减少 temporal token,使静态区域与动态区域获得不同的计算预算 [1]。
这类方法可以同时降低 attention、FFN 和 activation 的开销,但过早合并可能损害小物体、文字、手部、遮挡关系和快速运动的建模。对于接收轨迹、mask、ControlNet 或局部编辑条件的模型,token map 的变化还需与控制坐标保持一致。
2.2 Training-free Sparse Attention:利用预训练模型中的连接冗余¶
Sparse Video-Gen 观察到不同 head 分别侧重空间或时间交互,采用 online profiling 对 head 进行分类,并协同设计布局变换与定制 kernel [2]。AdaSpa 进一步考虑稀疏模式随 input、layer 和 head 的变化,采用 online precise search、LSE 复用以及 head-adaptive block 预算 [4]。RAPID 利用相邻 denoising step 之间 sparsity pattern 的稳定性复用模式,同时保留 inter-step adaptation [7]。
这类方法需要在计算预算与连接质量之间取得平衡。部分工作在最初若干 step 或敏感 layer 中保留 dense attention,并在较稳定的区间提高稀疏度,以减少重要连接被误删的风险。
2.3 语义重排与路由:兼顾连接选择与执行效率¶
Sparse VideoGen2 指出,当重要 token 分散在序列的不同位置时,GPU 仍可能因 padding 和非连续访存产生额外开销。该方法利用语义聚类与 permutation 将重要 token 重排为较密集的 block,再通过动态 kernel 进行计算 [3]。VORTA 在多种 sparse attention 变体之间进行路由,以兼顾局部和长距离交互 [5]。VMoBA 则通过 1D/2D/3D block partition、global block selection 和阈值预算,建模不同 head 的时空局部性 [10]。
上述方法同时涉及稀疏模式选择和系统实现。若语义选择无法转换为高效的执行布局,收益可能仅停留在理论 FLOPs;若布局规则但 token 选择不准确,则可能损害长程身份一致性、遮挡后的主体重现以及镜头内的关系建模。
2.4 可训练稀疏机制与后训练转换¶
Training-free 方法便于应用于已有模型,但 selector 受原始 attention 分布的限制。VSA 将 coarse tile 选择和 fine token attention 纳入可微 kernel,使稀疏计算可同时用于训练与推理 [6]。BLADE 联合训练 adaptive block sparsity 与 step distillation;其实验表明,先将模型蒸馏为少步模型、再独立进行稀疏化,未必能够补偿两种近似产生的联合误差 [8]。QuantSparse 则对稀疏 attention 与低比特模型进行联合恢复,进一步显示稀疏误差与量化误差之间存在耦合 [13]。
LinVideo 选择性地将部分 softmax attention 转换为 linear attention,并利用 distribution matching 恢复性能 [12]。SANA-Video 则在骨干设计阶段引入 block linear DiT [11]。前者属于已有 checkpoint 的后训练转换,后者属于新架构设计;两者均不宜与 training-free sparse mask 直接归为同一类实验。
2.5 分布式稀疏计算:通信与负载均衡¶
DSA 对 sparse attention、多 GPU 切分、调度和负载均衡进行联合设计 [9]。在多卡环境中,如果部分 head 或 block 保留更多连接,整个 attention 的时延将受到最慢设备的限制;若动态稀疏进一步引入 all-to-all 或频繁的数据重分布,通信开销可能抵消理论上的计算节省。
因此,多卡实验需要同时报告设备数量、互连方式、各卡的 density 分布、通信时间、straggler 和 strong-scaling efficiency,而不能只给出全局平均 sparsity。
3. 方法演进¶
| 阶段 | 代表工作 | 主要问题 | 方法进展 |
|---|---|---|---|
| 运动感知 token 缩减 | ADAPTOR(2025) | 哪些时间 token 可被合并 | 从静态规则转向视频内容自适应 |
| online pattern + kernel | Sparse Video-Gen(2025) | 分类空间/时间 head | 将 pattern 与可执行 layout 同时设计 |
| 精确搜索与动态预算 | AdaSpa、VORTA(2025) | input/head/layer 变化与长距交互 | 由固定 mask 转向动态稀疏模式 |
| 语义重排 | Sparse VideoGen2(2025) | 关键 token 分散导致的 GPU 浪费 | 区分 identification accuracy 与 hardware efficiency |
| 训练期稀疏 | VSA(2025) | 训练和推理共用可微 kernel | 将稀疏计算纳入模型训练 |
| 跨步模式复用 | RAPID(2026) | 降低 search 自身的开销 | 结合 attention sparsity 与 denoising-time cache |
| 联合压缩 | BLADE、QuantSparse(2026) | 稀疏与少步/量化的联合误差 | 对多种压缩误差进行联合优化 |
| 线性化与分布式执行 | LinVideo、SANA-Video、DSA(2026) | 替换算子与跨卡执行 | 扩展至 architecture/system co-design |
这些工作并非在相同实验条件下逐年改进。较新的方法通常同时引入语义聚类、稀疏 kernel、少步蒸馏或多卡调度,因此其整体加速比不能仅归因于 sparsity。
4. 代表方法¶
| 工作 | 主要对象 | 是否需要新 checkpoint | 动态性 | 实现实际加速的条件 | 主要质量风险 |
|---|---|---|---|---|---|
| ADAPTOR [1] | temporal token | 否/可选调参 | 局部运动自适应 | selector 之后实际缩短序列 | 小物体、快运动、条件坐标错位 |
| Sparse Video-Gen [2] | spatial/temporal head 的连边 | 否 | 在线分类 | layout transform + custom kernel | pattern 错分和长距交互丢失 |
| Sparse VideoGen2 [3] | 语义关键 token 对 | 否 | 在线聚类与 top-p | permutation + dynamic block kernel | 聚类误差、重排开销 |
| AdaSpa [4] | head-adaptive block | 否 | input/layer/head/step 自适应 | fused search + LSE reuse | search step 与 density 对新 prompt 的适应性下降 |
| VORTA [5] | 多种 sparse mixer | 否 | 按 layer/step 路由 | 路由开销与各算子 kernel | 路由不当损害全局上下文 |
| VSA [6] | tile 内 critical tokens | 是 | 通过训练学习 | 可微 block kernel | 预训练成本与对特定骨干的依赖 |
| RAPID [7] | block mask | 否 | 跨 step 适配 | 通过 mask 复用降低搜索开销 | 跨步复用的 pattern 不再适用 |
| BLADE [8] | block attention + NFE | 是 | 内容自适应 | sparsity-aware distillation | 稀疏与少步近似共同造成质量损失 |
| DSA [9] | 分布式 sparse block | 否/实现绑定 | 动态调度 | 负载均衡与通信重叠 | straggler、跨卡重分布 |
| LinVideo [12] | softmax 层转 linear attention | 是(后训练) | 选层转换 | 线性 attention kernel | 复杂运动和全局依赖表达不足 |
5. 适用条件与限制¶
5.1 共识¶
- 稀疏模式具有输入和层次依赖性。 prompt、layer、head、视频长度和 denoising step 均可能改变重要连接的分布。
- 实际加速取决于 algorithm、layout 与 kernel 的协同。 非连续的理想 mask 可能比密度稍高但便于 block 化执行的 mask 更慢。
- 不同去噪阶段需要不同的计算预算。 语义形成、稳定生成和细节恢复阶段对 attention 的需求不同,固定 density 通常难以兼顾各阶段。
- 视频质量需要多维评估。 除平均图像质量外,还应考察小物体、遮挡、快速运动、镜头变化和长程身份一致性等典型失效现象。
5.2 争议¶
- 近似误差与生成质量。 相对于 dense output 的 PSNR 可用于检测近似误差,但不能替代感知质量、文本遵循和跨 seed 多样性评测。
- Training-free 方法的迁移性。 这类方法无需重新训练模型,但仍会受到特定 attention topology、RoPE、mask 和 kernel 的限制,其跨模型迁移能力尚无一致结论。
- 线性 attention 与稀疏 attention 的关系。 线性化具有较明确的复杂度优势,但不同层的可替换性和表达能力并不均匀。现有研究更倾向于将其与 sparse/local/global attention 混合使用。
- 多种加速方法的组合。 稀疏化会改变缓存相似性、量化分布和多卡负载,因此组合收益需要通过逐项消融和端到端实验确定。
6. 性能测试¶
为保证实验可比性,应固定或完整报告以下设置:
checkpoint / prompt set / seeds / sampler / NFE / CFG
resolution / frames / FPS / batch / precision
GPU / interconnect / software / compile / warm-up
token count before and after / density / block size
selector / clustering / permutation / gather-scatter latency
attention-kernel / DiT-forward / DiT-only / end-to-end latency
peak VRAM / communication / per-device load balance
frame quality / motion / text / consistency / diversity / failures
评测结果可分为三个层次:
- 近似保真度:在相同提示词和随机种子下,相对于密集计算基线的 PSNR、LPIPS、SSIM 或特征差异;
- 生成质量:文本、美学、运动、主体/背景一致性、多样性和人评;
- 系统效率:计入选择操作的预热后 p50/p95、峰值显存、吞吐、通信和单视频能耗。
若仅报告 attention kernel 的加速比,应明确该结果不包含 MLP、VAE、text encoder、聚类和布局转换的耗时。若实验比较的是 50-step dense baseline 与 4-step sparse student,则所得结果反映的是多项技术的组合收益。
7. 开放问题¶
- 由稀疏率转向误差预算。 后续 selector 可进一步预测删除特定 token/edge 对运动、文本遵循和长程一致性的边际影响。
- 面向控制条件的 token 缩减。 轨迹、视差、姿态、mask 和音频等条件可能需要分别设置不可合并的 token 集合。
- 统一的动态布局接口。 现有方法通常依赖特定的 Triton/CUDA 算子和分块大小,迁移到不同 GPU 或模型骨干时仍有较高成本。
- 稀疏化与 cache、quantization、parallelism 的联合校准。 后续研究需在统一质量约束下考察不同技术的组合 Pareto 前沿,而非分别优化单个模块。
- 可重复的长视频压力测试。 应公开镜头切换、镜头环绕、多主体交互、小物体重现和跨分钟身份一致性等测试的退化曲线,并减少对精选演示样例的依赖。
8. 延伸阅读¶
- 先读 ADAPTOR,理解减 token 与减 attention edge 的区别 [1]。
- 再对照 Sparse Video-Gen、AdaSpa 和 Sparse VideoGen2,比较 pattern discovery、精确搜索、语义重排与 kernel 设计 [2] [3] [4]。
- 阅读 VORTA 和 VMoBA,理解路由与混合 sparse topology [5] [10]。
- 对照 VSA 与 RAPID,比较训练所得稀疏模式和跨 denoising step 稀疏复用 [6] [7]。
- 最后阅读 BLADE、QuantSparse、LinVideo 和 DSA,了解联合压缩、线性化与分布式执行 [8] [9] [12] [13]。
返回压缩与推理加速综述导航。有关 GPU kernel、多卡计算与 serving 的讨论见系统与部署综述。
资料版本¶
手册结构修订:2026-09-20。原资料覆盖日期:2026-09-02。动态资源状态以条目日期和官方入口为准;未标注本仓库复现的实验数字均按其引用来源理解。
参考文献¶
[1] ADAPTOR: Adaptive Token Reduction for Video Diffusion Transformers. Elia Peruzzo et al. CVPR Workshops. 2025.
[2] Sparse Video-Gen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsity. Haocheng Xi et al. ICML. 2025.
[3] Sparse VideoGen2: Accelerate Video Generation with Sparse Attention via Semantic-Aware Permutation. Shuo Yang et al. NeurIPS. 2025.
[4] Training-free and Adaptive Sparse Attention for Efficient Long Video Generation. Yifei Xia et al. ICCV. 2025.
[5] VORTA: Efficient Video Diffusion via Routing Sparse Attention. Wenhao Sun et al. NeurIPS. 2025.
[6] Faster Video Diffusion with Trainable Sparse Attention. Peiyuan Zhang et al. NeurIPS. 2025.
[7] RAPID: Reusing Attention Sparsity with Inter-step Adaptation for Efficient Video Diffusion. Shangran Lin, Lu Lu, Jian Chen, Qiang Liu. CVPR. 2026.
[8] BLADE: Block-Sparse Attention Meets Step Distillation for Efficient Video Generation. Youping Gu et al. ICLR. 2026.
[9] DSA: Efficient Inference for Video Generation Models via Distributed Sparse Attention. Shenggui Li et al. ICLR. 2026.
[10] VMoBA: Mixture-of-Block Attention for Video Diffusion Models. Jianzong Wu, Liang Hou, Haotian Yang, Ye Tian, Pengfei Wan, Di Zhang, Yunhai Tong. ICLR. 2026.
[11] SANA-Video: Efficient Video Generation with Block Linear Diffusion Transformer. Junsong Chen et al. ICLR. 2026.
[12] LinVideo: A Post-Training Framework towards O(n) Attention in Efficient Video Generation. Yushi Huang et al. CVPR. 2026.
[13] QuantSparse: Comprehensively Compressing Video Diffusion Transformer with Model Quantization and Attention Sparsification. Weilun Feng et al. ICLR. 2026.