小红书DoSA加速稀疏注意力索引器,1M上下文首token延迟降37%-48%

机器之心 · wechat · 2026-10-20

小红书Dots团队自研DoSA(Dots Sparse Attention),针对稀疏注意力中负责筛选token的索引器瓶颈做加速,在保持原始Top-K选择结果的前提下,测试中提前过滤约97%的候选。

问题:DeepSeek-V3.2采用的DSA先由索引器打分选出Top-K再由主注意力计算,但索引器本身成为瓶颈。约1M token上下文下,Dots3-Note-preview、GLM-5.2和DeepSeek-V3.2的Indexer耗时分别占预填充时间的81.6%、73.0%和90.8%。

方法:保留原始评分函数,用数学上下界证明哪些计算可省略。先选种子候选并用全部索引头重打分得到Top-K门槛下界,再为每个候选构造不低于真实得分的上界(保留少量重要头精确计算,正负权重头分别保守估计),只有上界严格低于门槛下界才排除,其余仍用全部索引头精算。工程上组织为规则化筛选、候选精算、紧凑写回三阶段,利用查询间候选重叠减少数据搬运。

结果:GLM-5.2完整Indexer微基准60种配置全部快于SGLang和LiteTopK,1M上下文下相对SGLang加速2.472.74倍。端到端TTFT在约1M上下文下,Dots3-Note-preview、GLM-5.2、DeepSeek-V3.2较各自SGLang基线分别降低约37%、46%、48%。

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

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →