Als nächstesNächster Leitfaden
Aufmerksamkeit bei mehreren Abfragen
Sprach-KI
Sprach-KI-GUIDE
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads.
Hören Sie sich diesen Leitfaden anKI-erzählt · ~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.
Kostenloser Newsletter
Drei verifizierte KI-Geschichten jeden Werktagmorgen, in einfachem Englisch geschrieben. Für immer kostenlos, keine Werbung.
Eine E‑Mail jeden Werktag. Abmelden mit einem Klick. Wir verkaufen oder teilen Ihre Adresse nie.
Testen Sie sich selbst
Sofortiges Feedback zu jeder Antwort und ein teilbares Zertifikat mit verifizierbarer ID, sobald Sie einen Kurs bestanden haben.
Unterstützen Sie kostenlose KI-Bildung. AI Understanding ist eine 501(c)(3) gemeinnützige Organisation – keine Werbung, keine Paywall, nie. Machen Sie eine Spende
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 behält separate Abfrageköpfe bei, lässt jedoch zu, dass jede Gruppe einen Satz von Schlüssel- und Wertköpfen gemeinsam nutzt, wodurch der KV-Cache verkleinert wird.
Multi-Head gibt jedem Kopf seinen eigenen KV; Multi-Abfrage teilt einen KV für alle Köpfe; GQA liegt dazwischen und einige KV-Gruppen.
Die Einsparungen zeigen sich zum Zeitpunkt der Generierung, wo das Lesen des KV-Cache die Hauptkosten für die Speicherbandbreite darstellt.
32 Abfrageköpfe geteilt durch 8 gemeinsam genutzte KV-Gruppen führen zu einer etwa vierfachen Reduzierung der zwischengespeicherten Schlüssel und Werte.
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.
Lerne weiter
Weitere Leitfäden zu diesem Thema ausgewählt
Als nächstesNächster Leitfaden
Aufmerksamkeit bei mehreren Abfragen
Sprach-KI