Alert button

Understanding the Generalization Benefit of Normalization Layers: Sharpness Reduction

Jun 14, 2022
Kaifeng Lyu, Zhiyuan Li, Sanjeev Arora

Figure 1 for Understanding the Generalization Benefit of Normalization Layers: Sharpness Reduction
Figure 2 for Understanding the Generalization Benefit of Normalization Layers: Sharpness Reduction
Figure 3 for Understanding the Generalization Benefit of Normalization Layers: Sharpness Reduction
Figure 4 for Understanding the Generalization Benefit of Normalization Layers: Sharpness Reduction

Share this with someone who'll enjoy it:

Normalization layers (e.g., Batch Normalization, Layer Normalization) were introduced to help with optimization difficulties in very deep nets, but they clearly also help generalization, even in not-so-deep nets. Motivated by the long-held belief that flatter minima lead to better generalization, this paper gives mathematical analysis and supporting experiments suggesting that normalization (together with accompanying weight-decay) encourages GD to reduce the sharpness of loss surface. Here "sharpness" is carefully defined given that the loss is scale-invariant, a known consequence of normalization. Specifically, for a fairly broad class of neural nets with normalization, our theory explains how GD with a finite learning rate enters the so-called Edge of Stability (EoS) regime, and characterizes the trajectory of GD in this regime via a continuous sharpness-reduction flow.

* 68 pages, many figures  
View paper onarxiv icon

Share this with someone who'll enjoy it: