cs.LGOct 9, 2026

On the Depth Transfer of Neural Optimizers via L-Smoothness

Authors: Maximilian Wong, Chenyu Xu, Jiajin Li, Yiping Lu

Organizations: University of Chicago · Peking University · Sauder School of Business University of British Columbia · Beijing International Center for Mathematical Research Peking University

Abstract

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 LL-smoothness framework of Xu et al. (2026), we understand hyperparameter transfer as how do the Lipschitz and LL-smoothness constants of the loss scale with the depth KK? We show that inverse-depth residual branch scaling α=Θ(1/K)α=Θ(1/K) is sufficient to make both constants depth-independent at fixed width, uniformly over the bounded parameter domain and for arbitrary p→qp\to 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\geq 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.

Figures & tables

Appendix figures & tables2 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. On the Width Scaling of Neural Optimizers Under Matrix Operator Norms I: Row/Column Normalization and Hyperparameter Transfer

    Mar 10, 2026Ruihan Xu, Jiajin Li, Yiping LuNeural Network OptimizationHyperparameter Transfer

  2. Loop-μμP: Residual Scaling and Hyperparameter Transfer in Looped Transformers

    Jun 16, 2026Shaowen Wang, Bingrui Li, Tansheng Zhu +4Residual LearningHyperparameter Transfer

  3. Quantifying Hyperparameter Transfer and the Importance of Embedding Layer Learning Rate

    May 20, 2026Dayal Singh Kalra, Maissam BarkeshliLanguage Model Scaling LawsHyperparameter Transfer