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: Proxy-Normalizing Activations to Match Batch Normalization while Removing Batch Dependence

Abstract: We investigate the reasons for the performance degradation incurred with batch-independent normalization. We find that the prototypical techniques of layer normalization and instance normalization both induce the appearance of failure modes in the neural network's pre-activations: (i) layer normalization induces a collapse towards channel-wise constant functions; (ii) instance normalization induces a lack of variability in instance statistics, symptomatic of an alteration of the expressivity. To alleviate failure mode (i) without aggravating failure mode (ii), we introduce the technique "Proxy Normalization" that normalizes post-activations using a proxy distribution. When combined with layer normalization or group normalization, this batch-independent normalization emulates batch normalization's behavior and consistently matches or exceeds its performance.
Comments: NeurIPS 2021 camera-ready
Subjects: Machine Learning (cs.LG); Computer Vision and Pattern Recognition (cs.CV); Machine Learning (stat.ML)
Cite as: arXiv:2106.03743 [cs.LG]
  (or arXiv:2106.03743v6 [cs.LG] for this version)

Submission history

From: Antoine Labatie [view email]
[v1] Mon, 7 Jun 2021 16:08:48 GMT (741kb,D)
[v2] Fri, 11 Jun 2021 13:50:18 GMT (748kb,D)
[v3] Mon, 1 Nov 2021 17:47:35 GMT (1055kb,D)
[v4] Wed, 3 Nov 2021 17:54:04 GMT (1062kb,D)
[v5] Sat, 15 Jan 2022 14:53:50 GMT (1068kb,D)
[v6] Sun, 3 Apr 2022 15:42:50 GMT (1067kb,D)

Link back to: arXiv, form interface, contact.