СледваСледващо ръководство
Внимание за множество заявки
Езиков AI
Език AI РЪКОВОДСТВО
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads.
Чуйте това ръководствоAI-разказан · ~1 min
The original GQA paper studies this as a tradeoff between multi-head and multi-query attention for decoder inference.
The GQA paper proposes grouping query heads so multiple query heads share key/value heads. MHA has one key/value head per query head; MQA shares a single key/value head across all queries; GQA lies between them. The paper evaluates GQA on language modeling and machine translation tasks and reports quality closer to MHA with inference speed closer to MQA in its experiments. It also reports uptraining existing MHA models using about 5% of original pretraining compute. These claims are limited to the paper’s models, tasks and setup.
Comparing MHA, GQA and MQA head sharing.
Estimating KV-cache size from the number of query heads and key/value groups.
Reading the paper’s reported quality/speed tradeoff on evaluated language-model tasks.
Reviewing the paper’s uptraining recipe for converting an MHA checkpoint to GQA.
Benchmark head-sharing choices on the target model and hardware. The original paper’s quality/speed results do not establish universal serving speedups or negligible quality loss across all systems.
Безплатен бюлетин
Три проверени AI истории всяка делнична сутрин, написани на достъпен език. Безплатно завинаги, без реклами.
Един имейл всеки делничен ден. Отписване с едно кликване. Никога не продаваме и не споделяме вашия имейл адрес.
Тествайте себе си
Незабавна обратна връзка за всеки отговор и сертификат за споделяне с подлежащ на проверка идентификационен номер, след като преминете курс.
Подкрепете безплатното AI образование. AI Understanding е 501(c)(3) с нестопанска цел — без реклами, без платен достъп, никога. Направете дарение
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads. The original GQA paper studies this as a tradeoff between multi-head and multi-query attention for decoder inference.
GQA поддържа отделни глави на заявки, но позволява на всяка група от тях да споделя един набор от глави на ключове и стойности, свивайки KV кеша.
Multi-head дава на всяка глава свой собствен KV; multi-query споделя един KV за всички глави; GQA се намира между няколко KV групи.
Спестяванията се показват по време на генериране, където четенето на KV кеша е доминиращият разход за честотна лента на паметта.
32 глави на заявки, разделени на 8 споделени KV групи, дават грубо четирикратно намаление на кешираните ключове и стойности.
The GQA paper reports quality closer to multi-head attention (MHA) with inference speed closer to multi-query attention (MQA) in its evaluated tasks. This is an empirical result, not a universal guarantee.
Продължавай да учиш
Още ръководства, избрани за тази тема
СледваСледващо ръководство
Внимание за множество заявки
Езиков AI