Multi-Head Attention Residuals Enhance Transformer Depth Routing
heghbalz · x · 2026-08-02
Traditional Transformers propagate information across depth using a single additive residual stream, forcing all feature subspaces to share one routing query. This compromise degrades performance as model width grows.
The paper Multi-Head Attention Residuals (MHAR) introduces a new architecture to fix this:
- Mechanism: Reshapes the routing query into H per-subspace heads, each with its own softmax over depth history.
- Zero Parameters: It is essentially a block-diagonal reshape, adding zero parameters and negligible compute.
- Performance: Outperforms standard Transformers at 100M, 350M, and 1B scales, with validation loss improvements increasing with model size.
- Optimal Heads: The head count H is a real design axis. H=4 or H=8 yields optimal results, while over-splitting (H=16) degrades performance.
Related event: MHAR Improves Transformer Residuals to Reduce Validation Loss(2 posts)→
More from Research
- Netflix details its production LLM judge: hundreds of thousands of recommendations scored weekly — omarsar0 · 2026-08-24
- Nature Comment: Provenance, not interpretability, grounds trust in autonomous science — gabepgomes · 2026-08-24
- New Architecture RHEA: Train 1B Model on 8GB VRAM — zemondza · 2026-08-24
- Trained two 16M-param models to do generative CAD with real physics — debreuil · 2026-08-24
- Claude model helps discover complex structure on S^6, solving 60-year-old math problem — Singularitarian · 2026-08-24
- Study: Agents read instructions/notes 60.5% of the time, rarely touch API docs — dair_ai · 2026-08-24