Head-pruning studies report that many attention heads can be removed with negligible loss. Pick the correct interpretation.
Pretraining over-provisions heads. Many can be pruned post-hoc with small quality loss, but training with fewer heads from scratch hits worse final quality. The extra heads pay off at train time.
Picture a relay team training with 16 runners. After many races, you notice only 8 actually run the final routes; the other 8 are bench helpers, warming up, scouting, sharing tactics. Once the team is trained, you can race with just the 8 runners and barely lose. But if you had only ever signed 8 runners, the team never would have developed the same tactics, because the bench helpers were what made the training useful. Attention heads work the same way: the extras help while learning, and you can drop them after.
Detailed answer & concept explanation~8 min readEverything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example. Click to expand.
Everything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example. Click to expand.
Everything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example.
Everything important, quickly.
60s: state the finding (30-50% of heads prunable post-training), give the over-provisioning interpretation, note the train from scratch failure as the key nuance, mention that prunable heads are identified empirically, connect to GQA / MLA as structured forms of the same insight.
Real products, models, and research that use this idea.
- Michel et al. 2019 'Are Sixteen Heads Really Better Than One?' showed 30-50% of heads prunable on BERT and Transformer NMT.
- Voita et al. 2019 connected head pruning to attention specialization, identifying syntactic, positional, and topic head patterns.
- GQA in Llama-2 70B is a structured form of the same insight applied to the KV side: many K, V heads were redundant.
- MLA in DeepSeek V4 takes the head-redundancy lesson further with learned-latent compression of K, V across heads.
- Modern LLM design (Llama 4 Maverick, Mistral Large 3, Qwen 3.5, Gemma 4) still uses 32-64+ heads despite the prunability finding, because training quality dominates inference slack.
What an interviewer would ask next. Try answering before peeking at the approach.
QWhy does training with fewer heads from scratch produce worse quality than pruning after?
QHow does the over-provisioning argument relate to GQA's KV head reduction?
Don't say thisRed flags and common mistakes that signal junior thinking. Click to expand.
Red flags and common mistakes that signal junior thinking. Click to expand.
Concluding 'pruning works, so train with fewer heads.' Empirically that fails: smaller from scratch models hit worse quality. The redundancy is a training-time enabler, not a wasted capacity.
The night-before-the-interview bullets. Scan these on the way to the call.
Primary sources. Skim if you want the original framing.
Same topic, related formats. Practice these next.