Maximal Initial Learning Rates in Deep ReLU Networks
Not on the map. Why?
Abstract (arXiv)
Training a neural network requires choosing a suitable learning rate, which involves a trade-off between speed and effectiveness of convergence. While there has been considerable theoretical and empirical analysis of how large the learning rate can be, most prior work focuses only on late-stage training. In this work, we introduce the maximal initial learning rate ηast - the largest learning rate at which a randomly initialized neural network can successfully begin training and achieve (at least) a given threshold accuracy. Using a simple approach to estimate ηast, we observe that in constant-width fully-connected ReLU networks, ηast behaves differently from the maximum learning rate later in training. Specifically, we find that ηast is well predicted as a power of depth × width, provided that (i) the width of the network is sufficiently large compared to the depth, and (ii) the input layer is trained at a relatively small learning rate. We further analyze the relationship between ηast and the sharpness λ1 of the network at initialization, indicating they are closely though not inversely related. We formally prove bounds for λ1 in terms of depth × width that align with our empirical results.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 33 of its 44 references that are in the map. First posted on arXiv 14 Dec 2022.
Builds on
33 of 44 references.
References · 33
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 413
- 2017Don't Decay the Learning Rate, Increase the Batch Sizecited by 313
- 2016On the Expressive Power of Deep Neural Networkscited by 302
- 2019PyHessian: Neural Networks Through the Lens of the Hessiancited by 291
- 2017Train longer, generalize better: closing the generalization gap in large batch training of neural networkscited by 278
- 2017Three Factors Influencing Minima in SGDcited by 270
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 206
- 2017Super-Convergence: Very Fast Training of Neural Networks Using Large Learning Ratescited by 196
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 183
21 more
- 2021The Principles of Deep Learning Theorycited by 169
- 2019Complexity of Linear Regions in Deep Networkscited by 166
- 2019Towards Explaining the Regularization Effect of Initial Large Learning Rate in Training Neural Networkscited by 162
- 2019Deep ReLU Networks Have Surprisingly Few Activation Patternscited by 160
- 2020The Break-Even Point on Optimization Trajectories of Deep Neural Networkscited by 149
- 2022Understanding Gradient Descent on Edge of Stability in Deep Learningcited by 126
- 2019Finite Depth and Width Corrections to the Neural Tangent Kernelcited by 114
- 2020The Early Phase of Neural Network Trainingcited by 114
- 2018How to Start Training: The Effect of Initialization and Architecturecited by 109
- 2018Which Neural Net Architectures Give Rise To Exploding and Vanishing Gradients?cited by 94
- 2018Universal Statistics of Fisher Information in Deep Neural Networks: Mean Field Approachcited by 89
- 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 88
- 2022Understanding the unstable convergence of gradient descentcited by 84
- 2017Failures of Gradient-Based Deep Learningcited by 79
- 2022Analyzing Sharpness along GD Trajectory: Progressive Sharpening and Edge of Stabilitycited by 61
- 2020Reconciling Modern Deep Learning with Traditional Optimization Analyses: The Intrinsic Learning Ratecited by 59
- 2018Products of Many Large Random Matrices and Gradients in Deep Neural Networkscited by 49
- 2021Large Learning Rate Tames Homogeneity: Convergence and Balancing Effectcited by 48
- 2020On the infinite width limit of neural networks with a standard parameterizationcited by 38
- 2021Deep ReLU Networks Preserve Expected Lengthcited by 17
- 2019The Effect of Network Width on Stochastic Gradient Descent and Generalization: an Empirical Studycited by 16
Built on by
Citing papers per year: 2023: 3; 2024: 3; 2025: 3; 2026: 2.
11 papers.
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfercited by 23
- 2023Phase diagram of early training dynamics in deep neural networks: effect of the learning rate, depth, and widthcited by 19
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Widthcited by 14
- 2025The Butterfly Effect: Neural Network Training Trajectories Are Highly Sensitive to Initial Conditionscited by 9
- 2023Catapult Dynamics and Phase Transitions in Quadratic Netscited by 8
- 2024Principled Architecture-aware Scaling of Hyperparameterscited by 4
- 2024ExpTest: Loss-Curve Hypothesis Testing for Autonomous Learning-Rate Selection in Deep Neural Networkscited by 0
- 2025Enlightenment Period Improving DNN Performancecited by 0
- 2025Beyond Gaussian Initializations: Signal Preserving Weight Initialization for Odd-Sigmoid Activationscited by 0
- 2026A Journey to the Edge of Stabilitycited by 0
- 2026Convergence of Practical Muon with Finite Newton-Schulz Iterations and Nesterov Momentumcited by 0
- 2026Convergence of Practical Muon with Finite Newton-Schulz Iterations and Nesterov Momentum30 Sep 2026
- 2026A Journey to the Edge of Stability26 Sep 2026
- 2025Beyond Gaussian Initializations: Signal Preserving Weight Initialization for Odd-Sigmoid Activations27 Sep 2025
- 2025The Butterfly Effect: Neural Network Training Trajectories Are Highly Sensitive to Initial Conditions16 Jun 2025
- 2025Enlightenment Period Improving DNN Performance2 Apr 2025
- 2024ExpTest: Loss-Curve Hypothesis Testing for Autonomous Learning-Rate Selection in Deep Neural Networks25 Nov 2024
- 2024Super Consistency of Neural Network Landscapes and Learning Rate Transfer27 Feb 2024
- 2024Principled Architecture-aware Scaling of Hyperparameters27 Feb 2024
- 2023On the Parameterization of Second-Order Optimization Effective Towards the Infinite Width19 Dec 2023
- 2023Phase diagram of early training dynamics in deep neural networks: effect of the learning rate, depth, and width23 Feb 2023
- 2023Catapult Dynamics and Phase Transitions in Quadratic Nets18 Jan 2023