Grouped-Query Attention (GQA) is an attention query layout grouping query heads to share a single Key and Value head, reducing the memory footprint of the KV cache.
Key to managing sequence memory and token weights during lightweight model inference, mobile device deployment, and llama 3 architectures; optimizing Grouped-Query Attention prevents attention processing bottlenecks and keeps execution latencies low.
Grouped-Query Attention (GQA) is an optimization technique for Transformer models that speeds up inference by grouping query heads to share a single key-value head. GQA acts as a middle ground between Multi-Head Attention (MHA) and Multi-Query Attention (MQA), delivering near-MHA quality while significantly reducing memory bandwidth usage and accelerating KV cache operations.
GQA dramatically shrinks the KV cache size, improving throughput with minor quality tradeoffs.
GQA offers a balance, providing higher model quality than MQA and better generation speed than MHA.
We currently have no direct coverage articles matching "Grouped-Query Attention". Explore trending global AI topics below instead.