English

Width and Depth Limits Commute in Residual Networks

Machine Learning 2023-08-11 v2 Machine Learning

Abstract

We show that taking the width and depth to infinity in a deep neural network with skip connections, when branches are scaled by 1/depth1/\sqrt{depth} (the only nontrivial scaling), result in the same covariance structure no matter how that limit is taken. This explains why the standard infinite-width-then-depth approach provides practical insights even for networks with depth of the same order as width. We also demonstrate that the pre-activations, in this case, have Gaussian distributions which has direct applications in Bayesian deep learning. We conduct extensive simulations that show an excellent match with our theoretical findings.

Keywords

Cite

@article{arxiv.2302.00453,
  title  = {Width and Depth Limits Commute in Residual Networks},
  author = {Soufiane Hayou and Greg Yang},
  journal= {arXiv preprint arXiv:2302.00453},
  year   = {2023}
}

Comments

24 pages, 8 figures. arXiv admin note: text overlap with arXiv:2210.00688

R2 v1 2026-06-28T08:29:06.353Z