How do grouped-query and multi-query attention speed up inference?
PICTURE THIS: A SENTENCE BECOMES TOKENS
The model does not read letters like humans. It reads these pieces, then predicts the next one.
Simple meaning
They share key and value heads across many query heads, shrinking the KV cache.
WHY — Tokens instead of words?
LLMs use tokens (not full words) because it helps them:
question about Attention.
trade-offs, and what you would actually do on a GenAI / LLM project - not buzzwords.
Each piece maps to a number the network can learn.
Fixed pieces are what transformers expect as input.
STEPS — What happens with tokens?
Before the model can read a sentence, it goes through these steps:
- 1They share key and
value heads across many query heads, shrinking the KV cache.
- 2Decode memory and bandwidth
drop, which raises tokens per second.
- 3Quality can dip slightly
versus full multi-head attention, so labs pick GQA as a serving compromise.
- 4Context mix
Attention looks at nearby tokens together.
- 5Next token
The model scores what should come next.
- 6Decode
IDs turn back into readable text.
EXAMPLE — See it in action
Let's see how a real sentence is tokenized (tokens may vary by model):
Note: Actual tokens and IDs depend on the tokenizer (e.g., GPT, Llama, etc.).
Key takeaway
They share key and value heads across many query heads, shrinking the KV cache. Decode memory and bandwidth drop, which raises tokens per second.