Kling 提出 ALLOT:免训练按输出分配块稀疏注意力省算力

KlingTeam · hf · 2026-10-22

KlingAI 与 UCAS、CASIA、北大团队提出 ALLOT,一种免训练的块稀疏注意力分配方法,用于视频扩散 Transformer 的加速。

背景与发现:全 3D 注意力开销大,现有块稀疏方法只决定「哪些块参与计算」,却忽视了每层每步保留多少块对质量与延迟同样关键;不同层和去噪步的稀疏化对最终保真度影响差异巨大,且最敏感的 layer-step 对在不同输入间几乎不变。

方法:ALLOT 把每个 layer-step 的延迟预算视作分配单元,基于输出测试(Output Tests)来分配块稀疏密度,在给定延迟预算下最大化输出保真度。

原文链接 →

「多模态」频道最新

更多「多模态」频道 AI 资讯 →