返回全部动态

DoSA:点稀疏注意力,无损加速 Top-K 选择

原标题:Paper page - DoSA: Dots Sparse Attention

Hugging Face Daily Papers一手来源研究质量 83

AI 摘要

该论文提出 DoSA(Dots Sparse Attention),一种无损稀疏注意力方法,通过分数上下界证书提前拒绝候选 token,仅对仍符合条件的候选进行完整打分,从而保持精确的 Top-K 选择。其 GPU 实现使完整 Indexer 相比 SGLang 提速最高 2.74 倍、相比 LiteTopK 提速 1.69 倍。在 GLM-5.2 和 DeepSeek-V3.2 上的端到端评估中,首 token 延迟分别相比两个基线最高提速 1.94 倍和 1.50 倍。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

DoSA: Dots Sparse Attention Abstract Sparse attention visits few keys, yet finding them can still require scoring every key with every indexer head. We introduce DoSA (Dots Sparse Attention), a lossless method for making this computation conditional while preserving exact token selection. Its central observation is that Top-K selection requires sufficient evidence to reject a candidate, not necessarily its full score. A seed set supplies a lower bound on the winning threshold. Nonnegative head p


发布时间:—
抓取时间:2026-10-14 10:26
来源机构:Hugging Face
阅读原文huggingface.co