小红书开源DoSA:无损索引器加速,百万token首字延迟最高降48%

小红书技术REDtech · wechat · 2026-10-21

小红书Dots团队发布DoSA,解决稀疏注意力(如DeepSeek-V3.2的DSA)中索引器成为长上下文预填充瓶颈的问题:在约1M上下文测试中,Indexer占预填充GPU时间的73%91%。

核心思路:不同于层间复用、减少索引头等可能改变Top-K选择的近似路线,DoSA在保持原始评分函数的前提下,用分数上下界提前排除不可能进入Top-K的候选——用种子候选的第K大分数构造门槛下界,再为每个候选构造有数学保证的得分上界,仅排除上界严格低于门槛者,实现无损筛选。

效果:GLM-5.2微基准中约97%的查询-key候选组合跳过全头精算;Indexer相对SGLang加速1.122.74倍;约1M token下,Dots3-Note-preview、GLM-5.2和DeepSeek-V3.2的首token延迟相对各自SGLang基线分别减少约37%、46%和48%。代码已开源(dosa-sglang),论文arXiv:2610.13990。

所属事件:小红书开源DoSA:无损加速稀疏注意力索引器,首token延迟最高降48%(2 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →