ΕπόμενοΕπόμενος οδηγός
Προσοχή πολλαπλών ερωτημάτων
Γλώσσα 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 κάθε πρωί της εβδομάδας, γραμμένα σε απλή αγγλική. Δωρεάν για πάντα, χωρίς διαφημίσεις.
Ένα email κάθε εργάσιμη ημέρα. Κατάργηση συνδρομής με ένα κλικ. Δεν πουλάμε ή μοιραζόμαστε ποτέ τη διεύθυνσή σας.
Δοκιμάστε τον εαυτό σας
Άμεση ανάδραση σε κάθε απάντηση, και πιστοποιητικό με επαληθεύσιμο ID μόλις περάσετε το μάθημα.
Υποστηρίξτε τη δωρεάν εκπαίδευση 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.
Η πολλαπλή κεφαλή δίνει σε κάθε κεφαλή το δικό της KV. πολλαπλών ερωτημάτων μοιράζεται ένα 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