mlsanityData to 14 Oct 2026

Negligible in Size, Significant in Effect: On Scale Vectors in Large Language Models

arXiv 2605.26895 ↗First posted 26 May 2026cs.LG cs.AI stat.MLcited by 4

Not on the map. Why?

Abstract (arXiv)

Normalization layers in modern large language models (LLMs) consist of a deterministic normalization operation and a learnable scale vector. While the normalization operation has been extensively studied, the scale vector remains poorly understood despite its ubiquitous use. In this work, we present a systematic study of scale vectors in LLMs from the perspectives of expressivity, optimization, and architectural structure. First, we show empirically that although scale vectors constitute only a negligible fraction of model parameters, removing them substantially degrades LLM pre-training. Our theory further shows that, in Pre-Norm architectures, scale vectors do not increase expressivity; instead, they improve optimization through a self-amplifying preconditioning effect on subsequent linear mappings. Second, we investigate the role of weight decay for scale vectors. By distinguishing Input-Norm and Output-Norm layers, we theoretically show that weight decay is beneficial for the former but harmful for the latter, due to their distinct roles in optimization and expressivity. Third, motivated by this understanding, we propose three lightweight and complementary improvements to scale vectors: branch-specific heterogeneity, improved placement around linear mappings, and magnitude-direction reparameterization. Both theory and experiments show that each improvement yields consistent gains. Finally, we combine these improvements into a unified scale-vector strategy and evaluate it through extensive LLM pre-training experiments on dense and mixture-of-experts models ranging from 0.12B to 2B parameters, across multiple optimizers and learning rate schedules, under industrial-scale token budgets. The unified strategy consistently achieves lower terminal loss than well-tuned baselines and exhibits more favorable scaling behavior, while adding negligible parameter and computational overhead.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 43 of its 50 references that are in the map. First posted on arXiv 26 May 2026.

Builds on

43 of 50 references.

References · 43

  1. 2017Attention Is All You Needcited by 60,361
  2. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  3. 2025Qwen3 Technical Reportcited by 8,820
  4. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  5. 2016Layer Normalizationcited by 5,669
  6. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  7. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  8. 2024Qwen2 Technical Reportcited by 2,120
  9. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
  10. 2019Root Mean Square Layer Normalizationcited by 1,480
  11. 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
  12. 2021Scaling Vision Transformerscited by 999
31 more
  1. 2020On Layer Normalization in the Transformer Architecturecited by 922
  2. 2019Introduction to Online Convex Optimizationcited by 803
  3. 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
  4. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
  5. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 723
  6. 2023Scaling Vision Transformers to 22 Billion Parameterscited by 694
  7. 2025Muon is Scalable for LLM Trainingcited by 428
  8. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
  9. 20242 OLMo 2 Furiouscited by 340
  10. 2018On the Optimization of Deep Networks: Implicit Acceleration by Overparameterizationcited by 284
  11. 2017Non-convex learning via Stochastic Gradient Langevin Dynamics: a nonasymptotic analysiscited by 243
  12. 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 178
  13. 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
  14. 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 142
  15. 2021Label Noise SGD Provably Prefers Flat Global Minimizerscited by 121
  16. 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 107
  17. 2020Shape Matters: Understanding the Implicit Bias of the Noise Covariancecited by 88
  18. 2025Transformers without Normalizationcited by 70
  19. 2022The alignment property of SGD noise and how it helps select flat minima: A stability analysiscited by 64
  20. 2020Noise and Fluctuation of Finite Learning Rate Stochastic Gradient Descentcited by 37
  21. 2025The Sharpness Disparity Principle in Transformers for Accelerating Language Model Pre-Trainingcited by 30
  22. 2021Power-law escape rate of SGDcited by 29
  23. 2021Stochastic gradient descent with noise of machine learning type. Part II: Continuous time analysiscited by 25
  24. 2026A Unified View of Attention and Residual Sinks: Outlier-Driven Rescaling is Essential for Transformer Trainingcited by 23
  25. 2024Parameter Symmetry and Noise Equilibrium of Stochastic Gradient Descentcited by 19
  26. 2025HybridNorm: Towards Stable and Efficient Transformer Training via Hybrid Normalizationcited by 16
  27. 2024Improving Generalization and Convergence by Enhancing Implicit Regularizationcited by 14
  28. 2025GradPower: Powering Gradients for Faster Language Model Pre-Trainingcited by 8
  29. 2026Accelerating LLM Pre-Training through Flat-Direction Dynamics Enhancementcited by 8
  30. 2026Post-LayerNorm Is Back: Stable, ExpressivE, and Deepcited by 7
  31. 2025SeeDNorm: Self-Rescaled Dynamic Normalizationcited by 6

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024Scaling Laws for Post Training Quantized Large Language Models
  2. 2025LLMs on the Line: Data Determines Loss-to-Loss Scaling Laws
  3. 2026Understanding the Weight Averaging Mechanism in LLM Training for Post-Training Quantization
  4. 2025Compression Scaling Laws:Unifying Sparsity and Quantization
  5. 2024Rethinking Conventional Wisdom in Machine Learning: From Generalization to Scaling
  6. 2025Low-Precision Training of Large Language Models: Methods, Challenges, and Opportunities
  7. 2025Generalizing Scaling Laws for Dense and Sparse Large Language Models
  8. 2025Understanding Post-Training Structural Changes in Large Language Models