Learning Less Is More: Premature Upper-Layer Attention Specialization Hurts Language Model Pretraining
Not on the map. Why?
Abstract (arXiv)
A causal-decoder block is hierarchical: lower layers build the residual basis that upper layers attend over. We identify a failure mode in GPT pretraining: upper layers commit to sharp attention patterns before lower-layer features stabilize. We call this premature upper-layer attention specialization. Temporarily slowing only upper-layer Q/K projections during early training improves final perplexity and downstream accuracy without altering other parameters; it prevents upper attention from collapsing onto an immature residual basis. In LLaMA-style blocks, the same intervention is nearly unnecessary. Through ablations, we isolate multiplicative gated FFNs (not RMSNorm or bias removal) as the component that suppresses the upstream residual writes driving the failure. A pathwise analysis unifies both findings: the learning-rate intervention reduces a step-size factor, while gated FFNs reduce a residual-energy factor on the same growth pathway. Our results identify upper-layer Q/K timing as a concrete interaction point between decoder architecture and optimization.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 38 of its 42 references that are in the map. First posted on arXiv 11 May 2026.
Builds on
38 of 42 references.
References · 38
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2016Layer Normalizationcited by 5,669
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
- 2016Gaussian Error Linear Units (GELUs)cited by 3,502
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2022BLOOM: A 176B-Parameter Open-Access Multilingual Language Modelcited by 1,984
26 more
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
- 2020GLU Variants Improve Transformercited by 1,804
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
- 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 1,003
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
- 2019Are Sixteen Heads Really Better than One?cited by 816
- 2016Language Modeling with Gated Convolutional Networkscited by 702
- 2023Scaling Vision Transformers to 22 Billion Parameterscited by 694
- 2024OLMo: Accelerating the Science of Language Modelscited by 613
- 2021Knowledge Neurons in Pretrained Transformerscited by 601
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
- 2020Query-Key Normalization for Transformerscited by 280
- 2020Understanding the Difficulty of Training Transformerscited by 210
- 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 178
- 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 161
- 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
- 2024The Evolution of Statistical Induction Heads: In-Context Learning Markov Chainscited by 118
- 2021Do Transformer Modifications Transfer Across Implementations and Applications?cited by 87
- 2021NormFormer: Improved Transformer Pretraining with Extra Normalizationcited by 83
- 2025PolyPythias: Stability and Outliers across Fifty Language Model Pre-Training Runscited by 35
- 2025Controlling changes to attention logitscited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.