Negligible in Size, Significant in Effect: On Scale Vectors in Large Language Models
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
- 2017Attention Is All You Needcited by 60,361
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2025Qwen3 Technical Reportcited by 8,820
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2016Layer Normalizationcited by 5,669
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2024Qwen2 Technical Reportcited by 2,120
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2021Scaling Vision Transformerscited by 999
31 more
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2019Introduction to Online Convex Optimizationcited by 803
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 723
- 2023Scaling Vision Transformers to 22 Billion Parameterscited by 694
- 2025Muon is Scalable for LLM Trainingcited by 428
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 426
- 20242 OLMo 2 Furiouscited by 340
- 2018On the Optimization of Deep Networks: Implicit Acceleration by Overparameterizationcited by 284
- 2017Non-convex learning via Stochastic Gradient Langevin Dynamics: a nonasymptotic analysiscited by 243
- 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 178
- 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
- 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 142
- 2021Label Noise SGD Provably Prefers Flat Global Minimizerscited by 121
- 2021What Happens after SGD Reaches Zero Loss? --A Mathematical Frameworkcited by 107
- 2020Shape Matters: Understanding the Implicit Bias of the Noise Covariancecited by 88
- 2025Transformers without Normalizationcited by 70
- 2022The alignment property of SGD noise and how it helps select flat minima: A stability analysiscited by 64
- 2020Noise and Fluctuation of Finite Learning Rate Stochastic Gradient Descentcited by 37
- 2025The Sharpness Disparity Principle in Transformers for Accelerating Language Model Pre-Trainingcited by 30
- 2021Power-law escape rate of SGDcited by 29
- 2021Stochastic gradient descent with noise of machine learning type. Part II: Continuous time analysiscited by 25
- 2026A Unified View of Attention and Residual Sinks: Outlier-Driven Rescaling is Essential for Transformer Trainingcited by 23
- 2024Parameter Symmetry and Noise Equilibrium of Stochastic Gradient Descentcited by 19
- 2025HybridNorm: Towards Stable and Efficient Transformer Training via Hybrid Normalizationcited by 16
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularizationcited by 14
- 2025GradPower: Powering Gradients for Faster Language Model Pre-Trainingcited by 8
- 2026Accelerating LLM Pre-Training through Flat-Direction Dynamics Enhancementcited by 8
- 2026Post-LayerNorm Is Back: Stable, ExpressivE, and Deepcited by 7
- 2025SeeDNorm: Self-Rescaled Dynamic Normalizationcited by 6
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Improving Neural Network Training by Decoupling the Magnitude and Direction of Weight Vectorscited by 3
- 2026Placing Degree Scales After LayerNormcited by 0
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretrainingcited by 0
- 2026LayerRoPE: Dynamic Depth-wise Magnitude & Angular Superpositioncited by 0
- 2026LayerRoPE: Dynamic Depth-wise Magnitude & Angular Superposition6 Oct 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026Improving Neural Network Training by Decoupling the Magnitude and Direction of Weight Vectors24 Jun 2026
- 2026Placing Degree Scales After LayerNorm12 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2024Scaling Laws for Post Training Quantized Large Language Models
- 2025LLMs on the Line: Data Determines Loss-to-Loss Scaling Laws
- 2026Understanding the Weight Averaging Mechanism in LLM Training for Post-Training Quantization
- 2025Compression Scaling Laws:Unifying Sparsity and Quantization
- 2024Rethinking Conventional Wisdom in Machine Learning: From Generalization to Scaling
- 2025Low-Precision Training of Large Language Models: Methods, Challenges, and Opportunities
- 2025Generalizing Scaling Laws for Dense and Sparse Large Language Models
- 2025Understanding Post-Training Structural Changes in Large Language Models