SıradakiSonraki rehber
Çoklu Sorgu Dikkati
Dil Yapay Zeka
Dil AI KILAVUZU
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads.
Bu kılavuzu dinleyinAI tarafından anlatılan · ~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.
Ücretsiz bülten
Hafta içi her sabah sade İngilizce yazılmış, doğrulanmış üç yapay zeka hikayesi. Sonsuza kadar ücretsiz, reklam yok.
Hafta içi her gün bir e-posta. Tek tıklamayla aboneliğinizi iptal edin. Adresinizi asla satmayız veya paylaşmayız.
Kendinizi test edin
Her yanıtla ilgili anında geri bildirim ve bir kursu geçtiğinizde doğrulanabilir bir kimliğe sahip paylaşılabilir bir sertifika.
Ücretsiz AI eğitimini destekleyin. AI Understanding, bir 501(c)(3) kâr amacı gütmeyen kuruluştur; hiçbir zaman reklam yoktur, ödeme duvarı yoktur. Bağış yap
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 ayrı sorgu kafaları tutar ancak her grubun bir anahtar ve değer kafası kümesini paylaşmasına izin vererek KV önbelleğini daraltır.
Çok kafalı her kafaya kendi KV'sini verir; çoklu sorgu tüm kafalar için bir KV'yi paylaşır; GQA birkaç KV grubunun arasında yer alır.
Tasarruflar, KV önbelleğinin okunmasının baskın bellek bant genişliği maliyeti olduğu üretim zamanında ortaya çıkıyor.
32 sorgu kafasının 8 paylaşılan KV grubuna bölünmesi, önbelleğe alınan anahtar ve değerlerde kabaca dört kat azalma sağlar.
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.
Öğrenmeye devam et
Bu konu için daha fazla rehber seçildi
SıradakiSonraki rehber
Çoklu Sorgu Dikkati
Dil Yapay Zeka