mlsanityData to 14 Oct 2026

Arithmetic-Mean μP for Modern Architectures: A Unified Learning-Rate Scale for CNNs and ResNets

arXiv 2510.04327 ↗First posted 5 Oct 2025cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Choosing an appropriate learning rate remains a key challenge in scaling depth of modern deep networks. The classical maximal update parameterization (μP) enforces a fixed per-layer update magnitude, which is well suited to homogeneous multilayer perceptrons (MLPs) but becomes ill-posed in heterogeneous architectures where residual accumulation and convolutions introduce imbalance across layers. We introduce Arithmetic-Mean μP (AM-μP), which constrains not each individual layer but the network-wide average one-step pre-activation second moment to a constant scale. Combined with a residual-aware He fan-in initialization - scaling residual-branch weights by the number of blocks (Var[W]=c/(K· mathrmfan-in)) - AM-μP yields width-robust depth laws that transfer consistently across depths. We prove that, for one- and two-dimensional convolutional networks, the maximal-update learning rate satisfies η^star(L)propto L-3/2; with zero padding, boundary effects are constant-level as Ngg k. For standard residual networks with general conv+MLP blocks, we establish η^star(L)=Theta(L-3/2), with L the minimal depth. Empirical results across a range of depths confirm the -3/2 scaling law and enable zero-shot learning-rate transfer, providing a unified and practical LR principle for convolutional and deep residual networks without additional tuning overhead.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 16 of its 21 references that are in the map. First posted on arXiv 5 Oct 2025.

Builds on

16 of 21 references.

References · 16

  1. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
  2. 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
  3. 2020ReZero is All You Need: Fast Convergence at Large Depthcited by 264
  4. 2016Exponential expressivity in deep neural networks through transient chaoscited by 261
  5. 2019Fixup Initialization: Residual Learning Without Normalizationcited by 198
  6. 2016Deep Information Propagationcited by 195
  7. 2018Dynamical Isometry and a Mean Field Theory of CNNs: How to Train 10,000-Layer Vanilla Convolutional Neural Networkscited by 176
  8. 2015All you need is a good initcited by 124
  9. 2017Mean Field Residual Networks: On the Edge of Chaoscited by 83
  10. 2019How to Initialize your Network? Robust Initialization for WeightNorm & ResNetscited by 36
  11. 2022PASHA: Efficient HPO and NAS with Progressive Resource Allocationcited by 13
  12. 2017Deep Residual Networks and Weight Initializationcited by 12
4 more
  1. 2023Depth Dependence of μP Learning Rates in ReLU MLPscited by 9
  2. 2023Is One Epoch All You Need For Multi-Fidelity Hyperparameter Optimization?cited by 6
  3. 2024The Unreasonable Effectiveness Of Early Discarding After One Epoch In Neural Network Hyperparameter Optimizationcited by 5
  4. 2024Principled Architecture-aware Scaling of Hyperparameterscited by 4

Built on by

Not cited yet.