NastępnyNastępny poradnik
Uwaga dotycząca wielu zapytań
Językowa sztuczna inteligencja
PRZEWODNIK Językowy AI
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads.
Posłuchaj tego przewodnikaOpowiadane przez sztuczną inteligencję · ~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.
Darmowy newsletter
Trzy zweryfikowane historie AI każdego poranka w dni robocze, napisane prostym angielskim. Bezpłatne na zawsze, bez reklam.
Jedno e‑mail każdego dnia roboczego. Wypisz się jednym kliknięciem. Nigdy nie sprzedajemy ani nie udostępniamy Twojego adresu.
Sprawdź się
Natychmiastowa informacja zwrotna na każde pytanie oraz udostępnialny certyfikat z weryfikowalnym ID po zaliczeniu kursu.
Wspieraj darmową edukację AI. AI Understanding jest organizacją non-profit 501(c)(3) — bez reklam, bez płatnej ściany, zawsze. Przekaż darowiznę
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 utrzymuje oddzielne głowice zapytań, ale pozwala każdej grupie współdzielić jeden zestaw kluczy i wartości, zmniejszając pamięć podręczną KV.
Funkcja wielogłowicowa nadaje każdej głowicy własny współczynnik KV; wiele zapytań dzieli jeden KV dla wszystkich głowic; GQA znajduje się pomiędzy kilkoma grupami KV.
Oszczędności pojawiają się w czasie generowania, gdzie odczyt pamięci podręcznej KV jest dominującym kosztem przepustowości pamięci.
32 głowice zapytań podzielone przez 8 współdzielonych grup KV daje w przybliżeniu czterokrotną redukcję kluczy i wartości w pamięci podręcznej.
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.
Ucz się dalej
Wybrano więcej przewodników na ten temat
NastępnyNastępny poradnik
Uwaga dotycząca wielu zapytań
Językowa sztuczna inteligencja