Skip to content
简体中文

切换主题

// ◎ Transformer 与注意力

分组查询注意力 GQA

把注意力头分组,每组共享一份 K/V,在 MHA 与 MQA 间折中。

Grouped-Query Attention (GQA) 2023 进阶 核心

通俗解释

例如 32 个查询头分成 8 组,每组共享一份 K/V,KV Cache 缩小 4 倍而质量几乎不降。它是当前开源模型最常用的注意力形式。

用在哪里:Llama 2 70B、Llama 3、Mistral、Qwen 等

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

2019 多查询注意力 MQA2023 分组查询注意力 GQA2024 多头潜在注意力 MLA

01原始论文

GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpoints

Joshua Ainslie, James Lee-Thorp, Michiel de Jong, et al. · 2023

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

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

先学

分组查询注意力 GQA

学完可以学

相关:KV Cache

多查询注意力 MQA多头潜在注意力 MLAKV Cache 分组查询注意力 GQA

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

03学习资料

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

04在课程里哪里学

所在学习路线:从零理解一个大模型

给 AI 读 / 数据说明

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