Discussion on Per-Head Scaling in Multi-Head Attention
stochasticchasm · x · 2026-07-16
This reply discusses a parameterization detail in multi-head attention:
- The other party noted that in many LLMs, qnorm is simply applied to qproj(x), and the q/k vectors of each head are not scaled by \sqrt(dhead).
- The question raised here is why we don't use "per head independent gain" (e.g., nn.RMSNorm((nhead, dhead), ...)) to achieve per head gain.
- The response suggests this is related to the muP concept.
- Another reply adds that explaining attention's 1/d scaling based on per-head qk normalization could itself be a novel idea.
Related event: Researchers Debate 1/d-Style Attention Scaling in Modern LLMs(6 posts)→
More from Research
- Structural ensembles beat single predictions in TCR:pMHC generalization study — quaidmorris · 2026-07-22
- RSS launches under OMSF to push structural biology data modeling at scale — MoAlQuraishi · 2026-07-22
- enFoldX tops 8 neoantigen scans and an unseen-peptide benchmark — quaidmorris · 2026-07-22
- enFoldX reaches AUC 0.82 on human VDJdb and transfers to mouse at 0.76 — quaidmorris · 2026-07-22
- enFoldX gains accuracy as AF3 ensemble disagreement rises for non-binders — quaidmorris · 2026-07-22
- A 3D ray plot shows how hard this Jacobian counterexample is to read — moultano · 2026-07-22