论文提出 Virtual Head Attention:同时压缩 KV cache 并降低训练与 prefill FLOPs
PaddlePaddle 团队在 arXiv 发布 Virtual Head Attention(VHA)论文,通过 Q Premix 和 Linear Postmix 两个线性操作同时改善 KV cache 大小、训练与 prefill FLOPs 和模型质量,推理时可折叠进投影层化为标准 GQA-2 架构。在 4K 上下文 50B token 预训练的四个规模上均超过 GQA 基线(+0.05% 至 +0.81%),KV cache 相比 GQA-8 减少 4 倍;在 30B-A3B MoE 模型上比 MLA 和 GQA-4 分别高 +0.70% 和 +2.60%,长上下文 RULER 平均分 84.4 对 GQA 的 73.0。
推荐理由:论文给出 VHA 在多个规模和 MoE 模型上对 GQA、MLA 的对比数据,KV cache 压缩和 FLOPs 降幅都有具体数字可查。