On the Depth Transfer of Neural Optimizers via L-Smoothness
Not on the map. Why?
Abstract (arXiv)
Scaling deep learning models efficiently requires predictable performance transfer across compute regimes. Establishing reliable hyperparameter transfer allows optimal training dynamics and structural choices to be discovered on low-cost proxy models, avoiding catastrophic failures during massive-scale training. Existing depth-transfer analyses are architecture-specific and heuristic: hyperparameter transfer is only justified through toy blocks, and both rely on infinite-depth dynamics, a specific initialization, and feature-learning criteria. Building on L-smoothness framework of Xu et al. (2026), we understand hyperparameter transfer as how do the Lipschitz and L-smoothness constants of the loss scale with the depth K? We show that inverse-depth residual branch scaling α=Theta(1/K) is sufficient to make both constants depth-independent at fixed width, uniformly over the bounded parameter domain and for arbitrary p→ q geometries. Because the bounds depend on each residual branch only through its Lipschitz and smoothness constants, the argument applies to general residual networks (including branches of block depth ≥ 2 such as transformer and ResNet blocks) without assumptions on initialization or correlations between layer weights. Since the smoothness constant controls the curvature penalty in the descent inequality, our bounds motivate a learning-rate scale without an additional depth-dependent curvature correction for updates bounded in the prescribed norm. This provides a geometric explanation of inverse-depth residual scaling and its application to the operator-norm geometries associated with AdamW, Muon, and MOGA. Learning-rate sweeps on GPT-2 (depths 2--128) and CIFAR-100 ResNets (8--32 residual blocks) confirm that Muon and MOGA exhibit stable learning-rate transfer under these prescriptions, with a common optimal rate across a 64-fold increase in depth.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 28 of its 39 references that are in the map. First posted on arXiv 9 Oct 2026.
Builds on
28 of 39 references.
References · 28
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
- 2023Symbolic Discovery of Optimization Algorithmscited by 434
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 192
- 2020Feature Learning in Infinite-Width Neural Networkscited by 190
- 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 139
- 2023A Spectral Condition for Feature Learningcited by 109
- 2025Fantastic Pretraining Optimizers and Where to Find Themcited by 100
- 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 97
16 more
- 2025Understanding Gradient Orthogonalization for Deep Learning via Non-Euclidean Trust-Region Optimizationcited by 92
- 2025A Note on the Convergence of Muoncited by 83
- 2025Don't be lazy: CompleteP enables compute-efficient deep transformerscited by 78
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 72
- 2024nGPT: Normalized Transformer with Representation Learning on the Hyperspherecited by 70
- 2024Scalable Optimization in the Modular Normcited by 68
- 2023Depthwise Hyperparameter Transfer in Residual Networks: Dynamics and Scaling Limitcited by 66
- 2020Tensor Programs III: Neural Matrix Lawscited by 49
- 2025Completed Hyperparameter Transfer across Modules, Width, Depth, Batch and Durationcited by 27
- 2026Fantastic Pretraining Optimizers and Where to Find Them II: Hyperball Optimizationcited by 16
- 2026RMNP: Row-Momentum Normalized Preconditioning for Scalable Matrix-Based Optimizationcited by 15
- 2025Memory-Efficient LLM Pretraining via Minimalist Optimizer Designcited by 13
- 2026Hyperparameter Transfer with Mixture-of-Expert Layerscited by 13
- 2026On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfercited by 12
- 2026Nora: Normalized Orthogonal Row Alignment for Scalable Matrix Optimizercited by 5
- 2026Learning Rate Transfer in Normalized Transformerscited by 3
Built on by
Not cited yet.