Three Mechanisms of Weight Decay Regularization
Not on the map. Why?
Abstract (arXiv)
Weight decay is one of the standard tricks in the neural network toolbox, but the reasons for its regularization effect are poorly understood, and recent results have cast doubt on the traditional interpretation in terms of L2 regularization. Literal weight decay has been shown to outperform L2 regularization for optimizers for which they differ. We empirically investigate weight decay for three optimization algorithms (SGD, Adam, and K-FAC) and a variety of network architectures. We identify three distinct mechanisms by which weight decay exerts a regularization effect, depending on the particular optimization algorithm and architecture: (1) increasing the effective learning rate, (2) approximately regularizing the input-output Jacobian norm, and (3) reducing the effective damping coefficient for second-order optimization. Our results provide insight into how to improve the regularization of neural networks.
Key facts. Cited by 113 papers in the map (data through 14 Oct 2026). First posted on arXiv 29 Oct 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 20; 2022: 26; 2023: 23; 2024: 17; 2025: 10; 2026: 17.
113 papers.
Sort:
By area:
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 713
- 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 402
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
- 2023Explaining grokking through circuit efficiencycited by 111
- 2021Scalable Marginal Likelihood Estimation for Model Selection in Deep Learningcited by 106
- 2022Characterizing and overcoming the greedy nature of learning in multi-modal deep neural networkscited by 95
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2023Why Do We Need Weight Decay in Modern Deep Learning?cited by 72
- 2021Quasi-Global Momentum: Accelerating Decentralized Deep Learning on Heterogeneous Datacited by 69
- 2021SimPLE: Similar Pseudo Label Exploitation for Semi-Supervised Classificationcited by 68
- 2022SGD with Large Step Sizes Learns Sparse Featurescited by 68
- 2021Initialization and Regularization of Factorized Neural Layerscited by 58
38 more
- 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 58
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2023Rotational Equilibrium: How Weight Decay Balances Learning Across Neural Networkscited by 53
- 2021Noether's Learning Dynamics: Role of Symmetry Breaking in Neural Networkscited by 50
- 2024How to set AdamW's weight decay as you scale model and dataset sizecited by 43
- 2025Power Lines: Scaling Laws for Weight Decay and Batch Size in LLM Pre-trainingcited by 43
- 2024Approaching Deep Learning through the Spectral Dynamics of Weightscited by 31
- 2022Gradient Descent on Neurons and its Link to Approximate Second-Order Optimizationcited by 30
- 2022Robust Training of Neural Networks Using Scale Invariant Architecturescited by 29
- 2024Weight decay induces low-rank attention layerscited by 27
- 2024Prompt Learning via Meta-Regularizationcited by 26
- 2023Grokking in Linear Estimators -- A Solvable Model that Groks without Understandingcited by 25
- 2021Tensor Normal Training for Deep Learning Modelscited by 24
- 2023MoMo: Momentum Models for Adaptive Learning Ratescited by 24
- 2024Rethinking Conventional Wisdom in Machine Learning: From Generalization to Scalingcited by 23
- 2025Why Gradients Rapidly Increase Near the End of Trainingcited by 22
- 2021On the Periodic Behavior of Neural Network Training with Batch Normalization and Weight Decaycited by 20
- 2023Revisiting the Fragility of Influence Functionscited by 20
- 2025Weight Decay may matter more than muP for Learning Rate Transfer in Practicecited by 20
- 2022Training Scale-Invariant Neural Networks on the Sphere Can Happen in Three Regimescited by 19
- 2026Fantastic Pretraining Optimizers and Where to Find Them II: Hyperball Optimizationcited by 16
- 2023A Stochastic Proximal Polyak Step Sizecited by 15
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Widthcited by 15
- 2022A Data-Augmentation Is Worth A Thousand Samples: Exact Quantification From Analytical Augmented Sample Momentscited by 14
- 2024How Much Can We Forget about Data Contamination?cited by 14
- 2025Provable Separations between Memorization and Generalization in Diffusion Modelscited by 14
- 2022Improving Robustness with Adaptive Weight Decaycited by 13
- 2022Toward Equation of Motion for Deep Neural Networks: Continuous-time Gradient Descent and Discretization Error Analysiscited by 13
- 2021Proxy-Normalizing Activations to Match Batch Normalization while Removing Batch Dependencecited by 12
- 2022Improving Pre-trained Language Model Fine-tuning with Noise Stability Regularizationcited by 12
- 2022Semi-Discrete Normalizing Flows through Differentiable Tessellationcited by 11
- 2021Edge of chaos as a guiding principle for modern neural network trainingcited by 10
- 2023Stochastic Marginal Likelihood Gradients using Neural Tangent Kernelscited by 10
- 2023HyperSparse Neural Networks: Shifting Exploration to Exploitation through Adaptive Regularizationcited by 9
- 2024On the Nonlinearity of Layer Normalizationcited by 9
- 2022A Mini-Block Fisher Method for Deep Neural Networkscited by 8
- 2022Cyclical Focal Losscited by 8
- 2022Scale-invariant Bayesian Neural Networks with Connectivity Tangent Kernelcited by 7
50 most cited of 113 All 113 (JSON)
- 2026Signal-Noise Factorization Isolates Nuisance Variation into Removable Subspaces30 Sep 2026
- 2026When Does Scale-Invariant Optimization Become Unstable? An Exact Schedule Law with Weight Decay8 Sep 2026
- 2026From the Loss Landscape to Diverse Feature Learning in Neural Networks28 Aug 2026
- 2026Effective Learning Rate Governs Loss Dynamics in Language Model Pretraining25 Aug 2026
- 2026Hyperball May Not Be a Free Lunch24 Jul 2026
- 2026Weight-norm Criticality: A Mechanism for Loss Spikes Induced by the Normalization and Weight Decay23 Jul 2026
- 2026How Far Can Sharpness and Complexity Jointly Explain Generalization?27 Jun 2026
- 2026Fantastic Pretraining Optimizers and Where to Find Them II: Hyperball Optimization15 Jun 2026
- 2026Learning Sparse Compositional Functions with Norm-Constrained Neural Networks25 May 2026
- 2026Demystifying Manifold Constraints in LLM Pre-training6 May 2026
- 2026SAGE: Sign-Adaptive Gradient for Memory-Efficient LLM Optimization9 Apr 2026
- 2026Statistical and structural identifiability in representation learning12 Mar 2026
38 more
- 2026Quantifying LLM Attention-Head Stability: Implications for Circuit Universality17 Feb 2026
- 2026Weight Decay Improves Language Model Plasticity11 Feb 2026
- 2026TruKAN: Towards More Efficient Kolmogorov-Arnold Networks Using Truncated Power Functions2 Feb 2026
- 2026The Effect of Mini-Batch Noise on the Implicit Bias of Adam2 Feb 2026
- 2026The Geometric Mechanics of Contrastive Representation Learning: Alignment Potentials, Entropic Dispersion, and Cross-modal Divergence27 Jan 2026
- 2025Correction of Decoupled Weight Decay9 Dec 2025
- 2025Can Stationary Distributions of Scale-Invariant Neural Networks Be Described by the Thermodynamics of an Ideal Gas?10 Nov 2025
- 2025Provable Separations between Memorization and Generalization in Diffusion Models5 Nov 2025
- 2025Weight Decay may matter more than muP for Learning Rate Transfer in Practice21 Oct 2025
- 2025Efficient Hyperparameter Tuning via Trajectory Invariance Principle29 Sep 2025
- 2025GLAD: Generalizable Tuning for Vision-Language Models17 Jul 2025
- 2025Recovering Plasticity of Neural Networks via Soft Weight Rescaling7 Jul 2025
- 2025Why Gradients Rapidly Increase Near the End of Training2 Jun 2025
- 2025Power Lines: Scaling Laws for Weight Decay and Batch Size in LLM Pre-training19 May 2025
- 2025Towards Accurate Binary Spiking Neural Networks: Learning with Adaptive Gradient Modulation Mechanism20 Feb 2025
- 2024Preventing Model Collapse in Deep Canonical Correlation Analysis by Noise Regularization1 Nov 2024
- 2024Weight decay induces low-rank attention layers31 Oct 2024
- 2024Data Diversity as Implicit Regularization: How Does Diversity Shape the Weight Space of Deep Neural Networks?18 Oct 2024
- 2024FusionLLM: A Decentralized LLM Training System on Geo-distributed GPUs with Adaptive Compression16 Oct 2024
- 2024The Fair Language Model Paradox15 Oct 2024
- 2024How Much Can We Forget about Data Contamination?4 Oct 2024
- 2024Stochastic Subsampling With Average Pooling25 Sep 2024
- 2024Rethinking Conventional Wisdom in Machine Learning: From Generalization to Scaling23 Sep 2024
- 2024Approaching Deep Learning through the Spectral Dynamics of Weights21 Aug 2024
- 2024Clustering and Alignment: Understanding the Training Dynamics in Modular Addition18 Aug 2024
- 2024Revealing the Utilized Rank of Subspaces of Learning in Neural Networks5 Jul 2024
- 2024On the Nonlinearity of Layer Normalization3 Jun 2024
- 2024Classifying Overlapping Gaussian Mixtures in High Dimensions: From Optimal Classifiers to Neural Nets28 May 2024
- 2024How to set AdamW's weight decay as you scale model and dataset size22 May 2024
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimization5 Apr 2024
- 2024Prompt Learning via Meta-Regularization1 Apr 2024
- 2024Neural Networks with (Low-Precision) Polynomial Approximations: New Insights and Techniques for Accuracy Improvement17 Feb 2024
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Width19 Dec 2023
- 2023Analyzing and Improving the Training Dynamics of Diffusion Models5 Dec 2023
- 2023Random Linear Projections Loss for Hyperplane-Based Optimization in Neural Networks21 Nov 2023
- 2023Improving Deep Learning Optimization through Constrained Parameter Regularization15 Nov 2023
- 2023Grokking in Linear Estimators -- A Solvable Model that Groks without Understanding25 Oct 2023
- 2023Why Do We Need Weight Decay in Modern Deep Learning?6 Oct 2023
50 newest of 113 All 113 (JSON)
- 2024Prompt Learning via Meta-Regularizationcited by 26
- 2025GLAD: Generalizable Tuning for Vision-Language Modelscited by 3
- 2025GLAD: Generalizable Tuning for Vision-Language Models17 Jul 2025
- 2024Prompt Learning via Meta-Regularization1 Apr 2024
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 713
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
- 2023Explaining grokking through circuit efficiencycited by 111
- 2021Scalable Marginal Likelihood Estimation for Model Selection in Deep Learningcited by 106
- 2022Characterizing and overcoming the greedy nature of learning in multi-modal deep neural networkscited by 95
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2023Why Do We Need Weight Decay in Modern Deep Learning?cited by 72
- 2021Quasi-Global Momentum: Accelerating Decentralized Deep Learning on Heterogeneous Datacited by 69
- 2021SimPLE: Similar Pseudo Label Exploitation for Semi-Supervised Classificationcited by 68
- 2022SGD with Large Step Sizes Learns Sparse Featurescited by 68
- 2023Revisiting Weighted Aggregation in Federated Learning with Neural Networkscited by 58
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
12 most cited of 104 All 104 (JSON)
- 2026Signal-Noise Factorization Isolates Nuisance Variation into Removable Subspaces30 Sep 2026
- 2026When Does Scale-Invariant Optimization Become Unstable? An Exact Schedule Law with Weight Decay8 Sep 2026
- 2026From the Loss Landscape to Diverse Feature Learning in Neural Networks28 Aug 2026
- 2026Effective Learning Rate Governs Loss Dynamics in Language Model Pretraining25 Aug 2026
- 2026Hyperball May Not Be a Free Lunch24 Jul 2026
- 2026Weight-norm Criticality: A Mechanism for Loss Spikes Induced by the Normalization and Weight Decay23 Jul 2026
- 2026How Far Can Sharpness and Complexity Jointly Explain Generalization?27 Jun 2026
- 2026Fantastic Pretraining Optimizers and Where to Find Them II: Hyperball Optimization15 Jun 2026
- 2026Learning Sparse Compositional Functions with Norm-Constrained Neural Networks25 May 2026
- 2026Demystifying Manifold Constraints in LLM Pre-training6 May 2026
- 2026SAGE: Sign-Adaptive Gradient for Memory-Efficient LLM Optimization9 Apr 2026
- 2026Statistical and structural identifiability in representation learning12 Mar 2026
12 newest of 104 All 104 (JSON)