Skip to content
简体中文

切换主题

// ◎ Transformer 与注意力

多查询注意力 MQA

所有注意力头共享一份 K/V,大幅缩小 KV Cache。

Multi-Query Attention (MQA) 2019 进阶 扩展

通俗解释

标准多头注意力每个头都有自己的 K、V;MQA 只保留一份,解码时读取的显存少很多,速度更快,代价是少量质量下降。

用在哪里:PaLM、Falcon

时间轴 · 前置 → 当前 → 后继

2017 Transformer 与…2019 多查询注意力 MQA2023 分组查询注意力 GQA

01原始论文

Fast Transformer Decoding: One Write-Head is All You Need

Noam Shazeer · 2019

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

左边先学;右边是学完它之后可以学的。

先学

多查询注意力 MQA

学完可以学

相关:多头潜在注意力 MLA

Transformer 与自注…KV Cache分组查询注意力 GQA多头潜在注意力 MLA 多查询注意力 MQA

左:前置 · 右:后继 · 虚线:相关。看完整关系图 →

03学习资料

官方文档、图解、课程和参考实现,链接都核对过。

04在课程里哪里学

给 AI 读 / 数据说明

数据更新于:AI 算法图谱与课程体系(北京时间)