来源 · arXiv cs.CL
arxiv arXiv cs.CL · 18 天前 · 3 次浏览

PIVOT通过跨查询组共享前缀扫描来加速令牌级稀疏注意力

PIVOT(Proxy Indexing Via One full-prefix Traversal)是一种无需训练、可即插即用的DeepSeek稀疏注意力(DSA)索引器替代方案,它通过在邻近查询组间共享单次前缀扫描来减少计算冗余。PIVOT不再为每个查询单独对每个前置令牌进行评分,而是将查询组聚合为一个代理查询以获取候选集,随后从中为每个查询选出top-k个令牌。