LM Head Gradient Bottleneck Suppresses 95-99% of Gradient Norm
July 13, 2026
The output layer in language models creates an optimization bottleneck where backpropagating V-dimensional gradients through a rank-D linear layer compresses signal. Empirical tests show the bottleneck suppresses 95-99% of the gradient norm, making trivial patterns unlearnable and altering update directions for the majority of parameters.
HOW THIS AFFECTS YOU
●
researcherYou should account for this compression when designing training dynamics or scaling vocabulary sizes.