arXiv:2602.12675cs.LGcs.AI2026-02被引 16

改进稀疏线性注意力,实现高速视频生成且不损失质量

SLA2: Sparse-Linear Attention with Learnable Routing and QAT

  • 用可学习路由动态决定每处注意力用稀疏还是线性方式
  • 实现97%注意力稀疏度,速度提升18.6倍,质量保持不变
  • 支持低比特量化,适合部署在资源受限设备上

稀疏线性注意力(SLA)通过结合稀疏与线性注意力加速扩散模型,在视频生成中表现优异。然而,(i) SLA 依赖基于注意力权重大小的启发式划分,可能次优;(ii) 经形式化分析发现,SLA 与直接分解为稀疏和线性注意力存在不一致。为此提出 SLA2:(I) 引入可学习路由,动态决定每项注意力应采用稀疏或线性方式;(II) 提出更准确的稀疏-线性注意力公式,使用可学习比例融合两类分支;(III) 设计稀疏+低比特注意力,通过量化感知微调引入低比特注意力以降低量化误差。实验表明,在视频扩散模型上,SLA2 可实现 97% 的注意力稀疏度,带来 18.6 倍的注意力加速,同时保持生成质量。

原文摘要 · Abstract (English)

Sparse-Linear Attention (SLA) combines sparse and linear attention to accelerate diffusion models and has shown strong performance in video generation. However, (i) SLA relies on a heuristic split that assigns computations to the sparse or linear branch based on attention-weight magnitude, which can be suboptimal. Additionally, (ii) after formally analyzing the attention error in SLA, we identify a mismatch between SLA and a direct decomposition into sparse and linear attention. We propose SLA2, which introduces (I) a learnable router that dynamically selects whether each attention computation should use sparse or linear attention, (II) a more faithful and direct sparse-linear attention formulation that uses a learnable ratio to combine the sparse and linear attention branches, and (III) a sparse + low-bit attention design, where low-bit attention is introduced via quantization-aware fine-tuning to reduce quantization error. Experiments show that on video diffusion models, SLA2 can achieve 97% attention sparsity and deliver an 18.6x attention speedup while preserving generation quality.

注意力机制视频生成加速推理量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。