Theoretical Analysis of Auto Rate-Tuning by Batch Normalization
Not on the map. Why?
Abstract (arXiv)
Batch Normalization (BN) has become a cornerstone of deep learning across diverse architectures, appearing to help optimization as well as generalization. While the idea makes intuitive sense, theoretical analysis of its effectiveness has been lacking. Here theoretical support is provided for one of its conjectured properties, namely, the ability to allow gradient descent to succeed with less tuning of learning rates. It is shown that even if we fix the learning rate of scale-invariant parameters (e.g., weights of each layer with BN) to a constant (say, 0.3), gradient descent still approaches a stationary point (i.e., a solution where gradient is zero) in the rate of T-1/2 in T iterations, asymptotically matching the best bound for gradient descent with well-tuned learning rates. A similar result with convergence rate T-1/4 is also shown for stochastic gradient descent.
Key facts. Cited by 65 papers in the map (data through 14 Oct 2026). First posted on arXiv 10 Dec 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 16; 2022: 13; 2023: 12; 2024: 7; 2025: 10; 2026: 7.
65 papers.
Sort:
By area:
- 2022Understanding Gradient Descent on Edge of Stability in Deep Learningcited by 128
- 2022Vision Transformers provably learn spatial structurecited by 93
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2021On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)cited by 73
- 2025Transformers without Normalizationcited by 70
- 2022GradMax: Growing Neural Networks using Gradient Informationcited by 66
- 2021Spectral Normalisation for Deep Reinforcement Learning: an Optimisation Perspectivecited by 65
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2024Normalization and effective learning rates in reinforcement learningcited by 58
- 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Examplecited by 56
- 2021Noether's Learning Dynamics: Role of Symmetry Breaking in Neural Networkscited by 50
- 2021GradInit: Learning to Initialize Neural Networks for Stable and Efficient Trainingcited by 44
38 more
- 2023Solving Regularized Exp, Cosh and Sinh Regression Problemscited by 44
- 2024How to set AdamW's weight decay as you scale model and dataset sizecited by 43
- 2022Towards understanding how momentum improves generalization in deep learningcited by 40
- 2023DoWG Unleashed: An Efficient Universal Parameter-Free Gradient Descent Methodcited by 35
- 2024Residual Connections and Normalization Can Provably Prevent Oversmoothing in GNNscited by 34
- 2021Batch Normalization Orthogonalizes Representations in Deep Random Networkscited by 29
- 2022Robust Training of Neural Networks Using Scale Invariant Architecturescited by 29
- 2023Normalization Layers Are All That Sharpness-Aware Minimization Needscited by 27
- 2021Beyond BatchNorm: Towards a Unified Understanding of Normalization in Deep Learningcited by 25
- 2021Demystifying Batch Normalization in ReLU Networks: Equivalent Convex Optimization Models and Implicit Regularizationcited by 21
- 2021On the Periodic Behavior of Neural Network Training with Batch Normalization and Weight Decaycited 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
- 2022Rethinking Normalization Methods in Federated Learningcited by 19
- 2023TWINS: A Fine-Tuning Framework for Improved Transferability of Adversarial Robustness and Generalizationcited by 18
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrizationcited by 16
- 2023Latent State Models of Training Dynamicscited by 16
- 2022Toward Equation of Motion for Deep Neural Networks: Continuous-time Gradient Descent and Discretization Error Analysiscited by 13
- 2024From Attention to Activation: Unravelling the Enigmas of Large Language Modelscited by 13
- 2025Stronger Normalization-Free Transformerscited by 13
- 2021Proxy-Normalizing Activations to Match Batch Normalization while Removing Batch Dependencecited by 12
- 2021UniDrop: A Simple yet Effective Technique to Improve Transformer without Extra Costcited by 11
- 2022UDC: Unified DNAS for Compressible TinyML Modelscited by 10
- 2022Amortized Proximal Optimizationcited by 9
- 2024On the Nonlinearity of Layer Normalizationcited by 9
- 2023The Implicit Bias of Batch Normalization in Linear Models and Two-layer Linear Convolutional Neural Networkscited by 8
- 2025What Can Grokking Teach Us About Learning Under Nonstationarity?cited by 8
- 2021State estimation with limited sensors -- A deep learning based approachcited by 7
- 2021Sandwich Batch Normalization: A Drop-In Replacement for Feature Distribution Heterogeneitycited by 7
- 2023Partial Fine-Tuning: A Successor to Full Fine-Tuning for Vision Transformerscited by 7
- 2024Where Do Large Learning Rates Lead Us?cited by 6
- 2025On Learning Verifiers and Implications to Chain-of-Thought Reasoningcited by 6
- 2021Batch Normalization Preconditioning for Neural Network Trainingcited by 5
- 2021Accelerating Training of Batch Normalization: A Manifold Perspectivecited by 4
- 2022Nonlinear Advantage: Trained Networks Might Not Be As Complex as You Thinkcited by 4
- 2023On the Weight Dynamics of Deep Normalized Networkscited by 4
- 2025A Minimalist Example of Edge-of-Stability and Progressive Sharpeningcited by 4
- 2023On the Training Instability of Shuffling SGD with Batch Normalizationcited by 3
50 most cited of 65 All 65 (JSON)
- 2026When Does Scale-Invariant Optimization Become Unstable? An Exact Schedule Law with Weight Decay8 Sep 2026
- 2026Improving Neural Network Training by Decoupling the Magnitude and Direction of Weight Vectors24 Jun 2026
- 2026A Link between Shock-wave Theory and Symmetry-reduced Stochastic Gradient Descent for Artificial Neural Networks16 Jun 2026
- 2026Placing Degree Scales After LayerNorm12 Jun 2026
- 2026Low-Rank Decay for Grokking in Scale-Invariant Transformers: A Spectral-Geometric View3 Jun 2026
- 2026Do Deep Networks Forget Initialization? A Forgetting-Time View of Practical Inductive Bias27 May 2026
- 2026A Mechanism Study of Delayed Loss Spikes in Batch-Normalized Linear Models18 Apr 2026
- 2025Stronger Normalization-Free Transformers11 Dec 2025
- 2025Can Stationary Distributions of Scale-Invariant Neural Networks Be Described by the Thermodynamics of an Ideal Gas?10 Nov 2025
- 2025The Hidden Power of Normalization Layers in Neural Networks: Exponential Capacity Control2 Nov 2025
- 2025Weight Decay may matter more than muP for Learning Rate Transfer in Practice21 Oct 2025
- 2025Position: Many generalization measures for deep learning are fragile21 Oct 2025
38 more
- 2025What Can Grokking Teach Us About Learning Under Nonstationarity?26 Jul 2025
- 2025On Learning Verifiers and Implications to Chain-of-Thought Reasoning28 May 2025
- 2025Parallel Layer Normalization for Universal Approximation19 May 2025
- 2025Transformers without Normalization13 Mar 2025
- 2025A Minimalist Example of Edge-of-Stability and Progressive Sharpening4 Mar 2025
- 2024Where Do Large Learning Rates Lead Us?29 Oct 2024
- 2024From Attention to Activation: Unravelling the Enigmas of Large Language Models22 Oct 2024
- 2024Normalization and effective learning rates in reinforcement learning1 Jul 2024
- 2024Residual Connections and Normalization Can Provably Prevent Oversmoothing in GNNs5 Jun 2024
- 2024On the Nonlinearity of Layer Normalization3 Jun 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
- 2023Partial Fine-Tuning: A Successor to Full Fine-Tuning for Vision Transformers25 Dec 2023
- 2023Large Learning Rates Improve Generalization: But How Large Are We Talking About?19 Nov 2023
- 2023Latent State Models of Training Dynamics18 Aug 2023
- 2023The Implicit Bias of Batch Normalization in Linear Models and Two-layer Linear Convolutional Neural Networks20 Jun 2023
- 2023Normalization Layers Are All That Sharpness-Aware Minimization Needs7 Jun 2023
- 2023On the Weight Dynamics of Deep Normalized Networks1 Jun 2023
- 2023DoWG Unleashed: An Efficient Universal Parameter-Free Gradient Descent Method25 May 2023
- 2023Solving Regularized Exp, Cosh and Sinh Regression Problems28 Mar 2023
- 2023Convergence of variational Monte Carlo simulation and scale-invariant pre-training21 Mar 2023
- 2023TWINS: A Fine-Tuning Framework for Improved Transferability of Adversarial Robustness and Generalization20 Mar 2023
- 2023On the Training Instability of Shuffling SGD with Batch Normalization24 Feb 2023
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrization14 Feb 2023
- 2022Nonlinear Advantage: Trained Networks Might Not Be As Complex as You Think30 Nov 2022
- 2022Toward Equation of Motion for Deep Neural Networks: Continuous-time Gradient Descent and Discretization Error Analysis28 Oct 2022
- 2022Vision Transformers provably learn spatial structure13 Oct 2022
- 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Example7 Oct 2022
- 2022Rethinking Normalization Methods in Federated Learning7 Oct 2022
- 2022Training Scale-Invariant Neural Networks on the Sphere Can Happen in Three Regimes8 Sep 2022
- 2022Towards understanding how momentum improves generalization in deep learning13 Jul 2022
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reduction14 Jun 2022
- 2022Understanding Gradient Descent on Edge of Stability in Deep Learning19 May 2022
- 2022Amortized Proximal Optimization28 Feb 2022
- 2022Robust Training of Neural Networks Using Scale Invariant Architectures2 Feb 2022
- 2022UDC: Unified DNAS for Compressible TinyML Models15 Jan 2022
- 2022GradMax: Growing Neural Networks using Gradient Information13 Jan 2022
- 2021A Riemannian Mean Field Formulation for Two-layer Neural Networks with Batch Normalization17 Oct 2021
50 newest of 65 All 65 (JSON)
- 2022Understanding Gradient Descent on Edge of Stability in Deep Learningcited by 128
- 2022Vision Transformers provably learn spatial structurecited by 93
- 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
- 2021On the Validity of Modeling SGD with Stochastic Differential Equations (SDEs)cited by 73
- 2025Transformers without Normalizationcited by 70
- 2022GradMax: Growing Neural Networks using Gradient Informationcited by 66
- 2021Spectral Normalisation for Deep Reinforcement Learning: an Optimisation Perspectivecited by 65
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2022Understanding Edge-of-Stability Training Dynamics with a Minimalist Examplecited by 56
- 2021Noether's Learning Dynamics: Role of Symmetry Breaking in Neural Networkscited by 50
- 2021GradInit: Learning to Initialize Neural Networks for Stable and Efficient Trainingcited by 44
- 2023Solving Regularized Exp, Cosh and Sinh Regression Problemscited by 44
12 most cited of 62 All 62 (JSON)
- 2026When Does Scale-Invariant Optimization Become Unstable? An Exact Schedule Law with Weight Decay8 Sep 2026
- 2026A Link between Shock-wave Theory and Symmetry-reduced Stochastic Gradient Descent for Artificial Neural Networks16 Jun 2026
- 2026Placing Degree Scales After LayerNorm12 Jun 2026
- 2026Low-Rank Decay for Grokking in Scale-Invariant Transformers: A Spectral-Geometric View3 Jun 2026
- 2026Do Deep Networks Forget Initialization? A Forgetting-Time View of Practical Inductive Bias27 May 2026
- 2026A Mechanism Study of Delayed Loss Spikes in Batch-Normalized Linear Models18 Apr 2026
- 2025Stronger Normalization-Free Transformers11 Dec 2025
- 2025Can Stationary Distributions of Scale-Invariant Neural Networks Be Described by the Thermodynamics of an Ideal Gas?10 Nov 2025
- 2025The Hidden Power of Normalization Layers in Neural Networks: Exponential Capacity Control2 Nov 2025
- 2025Weight Decay may matter more than muP for Learning Rate Transfer in Practice21 Oct 2025
- 2025Position: Many generalization measures for deep learning are fragile21 Oct 2025
- 2025What Can Grokking Teach Us About Learning Under Nonstationarity?26 Jul 2025
12 newest of 62 All 62 (JSON)