Lost in Backpropagation: The LM Head is a Gradient Bottleneck
By Nathan Godey, Yoav Artzi
This paper reveals that the language model head (projecting from dimension D to vocabulary size V) creates not just an expressivity bottleneck but a severe optimization bottleneck, suppressing 95-99% of gradient norm during backpropagation. The authors provide theoretical analysis and empirical evidence showing this compression fundamentally alters training feedback for most parameters.