mlsanityData to 14 Oct 2026

Learning Less Is More: Premature Upper-Layer Attention Specialization Hurts Language Model Pretraining

arXiv 2605.10504 ↗First posted 11 May 2026cs.CLcited by 2

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  4. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  5. 2020Scaling Laws for Neural Language Modelscited by 6,586
  6. 2016Layer Normalizationcited by 5,669
  7. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
  8. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  9. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,512
  10. 2016Gaussian Error Linear Units (GELUs)cited by 3,502
  11. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  12. 2022BLOOM: A 176B-Parameter Open-Access Multilingual Language Modelcited by 1,984
26 more
  1. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
  2. 2020GLU Variants Improve Transformercited by 1,804
  3. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  4. 2019Root Mean Square Layer Normalizationcited by 1,480
  5. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  6. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  7. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  8. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 1,003
  9. 2020On Layer Normalization in the Transformer Architecturecited by 922
  10. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
  11. 2019Are Sixteen Heads Really Better than One?cited by 816
  12. 2016Language Modeling with Gated Convolutional Networkscited by 702
  13. 2023Scaling Vision Transformers to 22 Billion Parameterscited by 694
  14. 2024OLMo: Accelerating the Science of Language Modelscited by 613
  15. 2021Knowledge Neurons in Pretrained Transformerscited by 601
  16. 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
  17. 2020Query-Key Normalization for Transformerscited by 280
  18. 2020Understanding the Difficulty of Training Transformerscited by 210
  19. 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 178
  20. 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 161
  21. 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
  22. 2024The Evolution of Statistical Induction Heads: In-Context Learning Markov Chainscited by 118
  23. 2021Do Transformer Modifications Transfer Across Implementations and Applications?cited by 87
  24. 2021NormFormer: Improved Transformer Pretraining with Extra Normalizationcited by 83
  25. 2025PolyPythias: Stability and Outliers across Fifty Language Model Pre-Training Runscited by 35
  26. 2025Controlling changes to attention logitscited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.