研究人员推出了一种新颖的方法,称为分组值注意力(GVA),以减少 Transformer 模型中 KV 缓存的内存占用。GVA 通过存储分组值并使用学习到的线性映射重建内容键来实现这一点,该映射可以在推理过程中被吸收到查询中。与分组查询注意力(GQA)相比,该方法显著减少了持久缓存标量,减少了约 45-47%,同时保持了接近 GQA 的基准准确性。该团队正在开发自定义解码内核以进一步提高推理速度,并计划开源发布。 AI
影响 这项研究可能带来更高效的 LLM 推理,从而实现更快的处理速度和更低的硬件要求。
排序理由 该集群包含一篇详细介绍提高 AI 模型效率的新技术方法的论文。[lever_c_demoted from research: ic=1 ai=1.0]
- arXiv
- FineWeb-Edu
- GQA
- Grouped Query Attention
- Grouped Value Attention
- Hugging Face
- KV cache
- Rope
- Transformer++
- Vishesh Tripathi
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →