JSSA 稀疏注意力:128K 上下文解码吞吐达全 KV 的 6.43 倍

一篇新论文提出 Joint Subspace Sparse Attention(JSSA),用共享低维查询—键子空间做 token 级选择,并配套免聚集的注意力内核,在 128K 上下文下把解码吞吐做到全 KV 注意力的最高 6.43 倍,同时精度接近全 KV。

一句话看懂:一篇新论文提出 Joint Subspace Sparse Attention(JSSA),用共享低维查询—键子空间做 token 级选择,并配套免聚集的注意力内核,在 128K 上下文下把解码吞吐做到全 KV 注意力的最高 6.43 倍,同时精度接近全 KV。

事件核心:发生了什么

Hugging Face Papers 在 2026 年 10 月 13 日收录的论文摘要显示,JSSA 针对长上下文 LLM 解码中占主导的 KV-cache 流量问题。现有稀疏注意力方法大多在较粗粒度上操作,token 级选择虽然更细,但面临两个难题:低维打分可能丢失排序所需的查询—键结构,而把分散选中的 token 聚集起来又会抵消延迟收益。JSSA 的做法是学习一个共享低维查询—键子空间,通过最小化 token 分数保持误差来提升选择质量,并用可解的边际目标得到闭式谱解;执行侧则开发了免聚集的 token 级索引注意力内核,在标准 GQA KV-cache 布局下支持每个 KV 头独立选择。摘要称,在标准 GQA、滑动窗口/全局混合注意力和 MLA 模型上,JSSA 均接近全 KV 精度,且没有出现竞争方法常见的生成长度膨胀和架构相关失败。在 NVIDIA B200 GPU 上,128K 上下文时解码吞吐最高为全 KV 注意力的 6.43 倍;在 700B 规模大模型上达到模型原生 DSA 的 96%–99% 解码吞吐,推理精度相当,且无需模型训练。

为什么重要

长上下文推理的成本瓶颈主要集中在 KV-cache 的读取与搬运,而不是算力本身。JSSA 如果成立,意味着 token 级稀疏选择可以同时兼顾精度和实际延迟,而不是停留在算法指标上。它覆盖 GQA、混合注意力和 MLA 三类主流架构,说明这条路线不是绑定单一模型的技巧。免聚集内核和 GQA 布局的兼容性,也让它更贴近现有推理栈的工程约束。对正把上下文推向 128K 甚至更长的开源与闭源模型而言,这类工作直接影响单位 token 的推理成本和吞吐上限。

对用户/开发者/创作者的影响

目前公开信息仅为论文摘要,尚无产品化或 API 落地声明。对开发者来说,值得关注的是它是否会被主流推理框架(如 vLLM、SGLang 等)吸收为可选注意力后端,以及是否支持现成的 GQA 权重而不需要重训。对企业和云厂商而言,6.43 倍吞吐若能在真实服务中复现,会直接压低长上下文推理的算力账单。对创作者和普通用户,短期体验变化有限,但长文档问答、代码库理解和多轮长对话类 AI 应用的响应速度和可承受上下文长度可能间接受益。评测结论仅限摘要给出的模型与任务条件,不宜当作永久排名。

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

值得关注的后续

一,是否开源内核或提供与主流推理引擎的集成,这决定它能否从论文走向部署。二,在更广泛模型和真实线上流量下,精度与吞吐是否仍接近摘要描述。三,与 DSA 等既有稀疏注意力方案的对比是否会在完整实验和第三方复现中保持,尤其是在 700B 级模型上的表现。

来源:Hugging Face Papers

celebrityanime
celebrityanime
文章: 29202

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注