We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Batch Normalization Biases Residual Blocks Towards the Identity Function in Deep Networks

Abstract: Batch normalization dramatically increases the largest trainable depth of residual networks, and this benefit has been crucial to the empirical success of deep residual networks on a wide range of benchmarks. We show that this key benefit arises because, at initialization, batch normalization downscales the residual branch relative to the skip connection, by a normalizing factor on the order of the square root of the network depth. This ensures that, early in training, the function computed by normalized residual blocks in deep networks is close to the identity function (on average). We use this insight to develop a simple initialization scheme that can train deep residual networks without normalization. We also provide a detailed empirical study of residual networks, which clarifies that, although batch normalized networks can be trained with larger learning rates, this effect is only beneficial in specific compute regimes, and has minimal benefits when the batch size is small.
Comments: Camera-ready version of NeurIPS 2020
Subjects: Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV); Machine Learning (stat.ML)
Cite as: arXiv:2002.10444 [cs.LG]
  (or arXiv:2002.10444v3 [cs.LG] for this version)

Submission history

From: Soham De [view email]
[v1] Mon, 24 Feb 2020 18:43:03 GMT (603kb,D)
[v2] Thu, 18 Jun 2020 12:20:43 GMT (712kb,D)
[v3] Wed, 9 Dec 2020 10:18:10 GMT (712kb,D)

Link back to: arXiv, form interface, contact.