大模型-attention汇总解析之-MQA

MQA,即 "M ulti-Q uery Attention",是减少 KV Cache 的一次的一种大胆尝试,首次提出自《Fast Transformer Decoding: One Write-Head is All You Need》, 在2019 年减少 KV Cache 就已经是研究人员非常关注的一个课题了。MQA 的思路很简单,直接让所有 Attention Head 共享同一个 K、V。

论文中伪代码,

Todo : k , v 如何选择?实际上是多头共同使用一个k, v 缓存。

一般的multi head attention 的qkv的头的数量都一样,而multi query attention的q的头数量保持不变,k,v的头数量都变为1。

论文中的计算结果:

这样子就直接将 KV Cache 减少到了原来的1/h ,这是非常可观的, 已经简洁的不能再简洁了。使用 MQA 的模型包括 PaLM 、StarCoder、Gemini 等。

相关推荐
BeforeEasy20 小时前
关于大模型工具调用技术的总结
llm·agent·工具调用·function_call·tool_use
龙骑士baby20 小时前
重建 AI 认知第 1 篇:基础认知——一张地图看懂 AI Landscape
深度学习·ai·大模型·llm·ai生态
龙侠九重天21 小时前
Embedding 模型深度使用——语义搜索与聚类
人工智能·深度学习·数据挖掘·大模型·llm·embedding·聚类
吴佳浩1 天前
炸裂!一家创业公司声称打破了 Transformer 七年魔咒
人工智能·llm
冬奇Lab1 天前
RAG 系列(二十三):多模态 RAG——图片、表格也能检索
人工智能·llm
Cosolar2 天前
RAG语义丢失?全链路优化通关宝典✅
人工智能·面试·llm
带娃的IT创业者2 天前
LLM半年巨变:从模型混战到AI代理“爪“时代的崛起
人工智能·llm·ai编程·基准测试·技术趋势·ai代理
xrz5785pixel2 天前
▎ 让本地模型在 Codex 里调用工具:我把两个不兼容的 API 翻译了
llm
花千树-0102 天前
SubAgent 基础:拥有自主工具的子代理
java·langchain·llm·agent·langgraph·subagent·harness
qcx232 天前
【AI Daily】每日AI日报
人工智能·llm·agent·daily