跳到正文
我的
arXiv · cs.CL· Ruitian Wang·· 1 天前AI 评分40

RecSA:以对数成本实现百万 token 长度外推的递归稀疏注意力

Recursive Sparse Attention: Million-Token Length Extrapolation at Logarithmic Cost

来源与原文

arXiv · cs.CL

作者:Ruitian Wang

发布时间:(北京时间)

阅读原始来源 · arxiv.org ↗
完整链接与内容说明

https://arxiv.org/abs/2611.07232v1

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

RecSA(Recursive Sparse Attention)通过在 KV cache 上构建树结构、每层用共享 pooler 蒸馏出的 key 汇总子树,使每个 query 的选择成本降至 O(log L),仅需读取 O(log L) 条目。基于 Qwen3-4B 初始化并在原模型 32K 训练长度上训练 0.62B tokens,RecSA 在 RULER 上从 131K 到 2M 各长度均超越所有基线,1M 处得分 60.5,而最佳基线 Dense + DCA 为 40.0。在单张 H200 GPU 上 524K 上下文时,其优化 kernel 使 prefill 加速 3.2x、decoding 加速 3.8x,端到端快于稠密注意力。

来源:arXiv · cs.CL · arxiv.org

这条消息,你怎么看?

把新闻带到 MickerBook,和社区伙伴聊聊;也可以去 AIHER 看看模型背后的角色与梗。