SelanjutnyaPanduan berikutnya
Perhatian Multi-Kueri
Bahasa AI
PANDUAN AI Bahasa
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads.
Dengarkan panduan iniDiceritakan oleh 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.
Buletin gratis
Tiga cerita AI terverifikasi setiap pagi hari kerja, ditulis dalam bahasa Inggris sederhana. Gratis selamanya, tanpa iklan.
Satu email setiap hari kerja. Berhenti berlangganan dalam satu klik. Kami tidak pernah menjual atau membagikan alamat Anda.
Uji diri Anda sendiri
Umpan balik instan untuk setiap jawaban, dan sertifikat yang dapat dibagikan dengan ID yang dapat diverifikasi setelah Anda lulus kursus.
Mendukung pendidikan AI gratis. AI Understanding adalah organisasi nirlaba 501(c)(3) — tanpa iklan, tanpa paywall, selamanya. Berikan sumbangan
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 menyimpan kepala kueri yang terpisah tetapi memungkinkan setiap grup berbagi satu set kepala kunci dan nilai, sehingga memperkecil cache KV.
Multi-head memberi setiap kepala KVnya sendiri; multi-kueri membagikan satu KV untuk semua kepala; GQA berada di antara beberapa grup KV.
Penghematan muncul pada waktu pembangkitan, dimana pembacaan cache KV merupakan biaya bandwidth memori yang dominan.
32 kepala kueri yang dibagi dengan 8 grup KV bersama memberikan pengurangan sekitar empat kali lipat pada kunci dan nilai yang di-cache.
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.
Teruslah belajar
Panduan lainnya dipilih untuk topik ini
SelanjutnyaPanduan berikutnya
Perhatian Multi-Kueri
Bahasa AI