VHA注意力方案提出:KV缓存再减半,长上下文大模型推理能否更省算力

Hugging Face Papers 收录的论文提出 Virtual Head Attention(VHA),试图用两条轻量线性变换,在 KV 缓存、训练与预填充算力和模型质量之间同时改善,而不是三选二。值得关注的原因在于,它声称推理时可折叠回标准 GQA-2,直接复用现有优化。

一句话看懂:Hugging Face Papers 收录的论文提出 Virtual Head Attention(VHA),试图用两条轻量线性变换,在 KV 缓存、训练与预填充算力和模型质量之间同时改善,而不是三选二。值得关注的原因在于,它声称推理时可折叠回标准 GQA-2,直接复用现有优化。

事件核心:发生了什么

根据论文摘要(2026 年 10 月 13 日发布,目前公开信息为摘要,不代表全文实验已核验或已上线产品),长上下文大模型的注意力设计长期受困于“不可能三角”:KV 缓存大小、训练与预填充 FLOPs、模型质量难以同时兼顾。GQA 质量好但缓存开销高;MLA 能压缩缓存却有质量损失;MFA 增加训练和预填充计算。VHA 的思路是:Q Premix 在每组 KV 内对查询做近似恒等变换,用减半的物理查询头恢复“虚拟头”多样性;Linear Postmix 以低秩 I + AB^T 残差结构融合头间特征。因为全是线性操作,推理时 Premix 可并入查询投影、Postmix 可并入输出投影,模型退化为标准 GQA-2 架构。摘要给出的实验中,4K 上下文、500 亿 token 预训练的四个规模上,VHA 均超过 GQA 基线(+0.05% 至 +0.81%);在 30B-A3B MoE 模型(5000 亿 token、8K 上下文)的 12 项基准上,比 MLA 和 GQA-4 分别高 0.70% 和 2.60%。

为什么重要

长上下文推理的成本大头之一是 KV 缓存占用和显存带宽。VHA 若成立,意味着在保持 GQA 式工程兼容性的同时进一步压缩缓存:摘要称稠密规模下相对 GQA-8 缓存减 4 倍,MoE 规模相对 GQA-4 减 2 倍,1.7B 规模、序列长度 4096 时训练与预填充 FLOPs 降 7.9%。只保留两个 KV 头,可让大模型的组大小达到 64,更充分调用 Tensor Core。对开源与闭源模型团队而言,这是一条“不推翻现有 GQA 推理栈”的压缩路线,落地门槛低于重写注意力内核。

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

对普通用户,短期内没有直接产品变化;若被模型厂商采用,可能体现为更长的上下文窗口或更低的推理价格。对开发者,关键吸引力是“部署时就是 GQA-2”,现有推理框架的 GQA 优化可直接复用,改造成本主要在训练侧。对企业和算力采购方,注意力压缩若稳定成立,有助于降低长上下文服务的显存需求。创作者使用长文档、代码库类 AI 应用时,可能间接受益于更低的长上下文成本。

GamsGo AI

AI 工具推荐

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

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

了解 GamsGo AI

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

值得关注的后续

一是主流开源模型或推理框架是否跟进实现 VHA;二是摘要外的完整实验、消融与长上下文稳定性是否经得起复现;三是 MLA、MFA 等路线的后续改进是否抵消其优势。以上评测均限于论文所述条件,并非永久排名。

来源:Hugging Face Papers

celebrityanime
celebrityanime
文章: 29340

发表回复

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