A continuaciónSiguiente guía
Atención multiconsulta
Idioma IA
GUÍA de IA en idiomas
Grouped-Query Attention (GQA) shares key/value heads across groups of query heads.
Escuche esta guíaNarrado por IA · ~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.
Boletín gratuito
Tres noticias verificadas sobre IA cada mañana de lunes a viernes, explicadas con claridad. Gratis para siempre y sin anuncios.
Un correo cada día laborable. Date de baja con un clic. Nunca vendemos ni compartimos tu dirección.
Ponte a prueba
Corrección inmediata en cada respuesta y un certificado para compartir, con un ID verificable, cuando apruebas un curso.
Apoya la educación gratuita sobre IA. AI Understanding es una organización sin fines de lucro 501(c)(3): sin anuncios ni muros de pago, nunca. Hacer una donación
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 mantiene cabezales de consulta separados, pero permite que cada grupo comparta un conjunto de cabezales de clave y valor, lo que reduce la caché de KV.
El cabezal múltiple le da a cada cabezal su propio KV; la consulta múltiple comparte un KV para todos los cabezales; GQA se encuentra en el medio con algunos grupos de KV.
Los ahorros aparecen en el momento de la generación, donde la lectura de la caché KV es el costo de ancho de banda de memoria dominante.
32 cabezales de consulta divididos por 8 grupos KV compartidos ofrecen aproximadamente una reducción de cuatro veces en claves y valores almacenados en caché.
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.
sigue aprendiendo
Más guías seleccionadas para este tema.
A continuaciónSiguiente guía
Atención multiconsulta
Idioma IA