Abstract:The rapid development of deep neural networks (DNNs) in recent years can be attributed to the various techniques that address gradient explosion and vanishing. In order to understand the principle behind these techniques and develop new methods, plenty of metrics have been proposed to identify networks that are free of gradient explosion and vanishing. However, due to the diversity of network components and complex serial-parallel hybrid connections in modern DNNs, the evaluation of existing metrics usually requires strong assumptions, complex statistical analysis, or has limited application fields, which constraints their spread in the community. In this paper, inspired by the Gradient Norm Equality and dynamical isometry, we first propose a novel metric called Block Dynamical Isometry, which measures the change of gradient norm in individual blocks. Because our Block Dynamical Isometry is norm-based, its evaluation needs weaker assumptions compared with the original dynamical isometry. To mitigate challenging derivation, we propose a highly modularized statistical framework based on free probability. Our framework includes several key theorems to handle complex serial-parallel hybrid connections and a library to cover the diversity of network components. Besides, several sufficient conditions for prerequisites are provided. Powered by our metric and framework, we analyze extensive initialization, normalization, and network structures. We find that our Block Dynamical Isometry is a universal philosophy behind them. Then, we improve some existing methods based on our analysis, including an activation function selection strategy for initialization techniques, a new configuration for weight normalization, a depth-aware way to derive coefficients in SeLU, and initialization/weight normalization in DenseNet. Moreover, we propose a novel normalization technique named second moment normalization, which has 30 percent fewer computation overhead than batch normalization without accuracy loss and has better performance under micro batch size. Last but not least, our conclusions and methods are evidenced by extensive experiments on multiple models over CIFAR-10 and ImageNet.

A generic shift-norm-activation approach for deep learning

Normalized Activation Function: Toward Better Convergence

Evolving Normalization-Activation Layers

Beyond BatchNorm: Towards a Unified Understanding of Normalization in Deep Learning

Normalization Techniques in Training DNNs: Methodology, Analysis and Application

Unified Normalization for Accelerating and Stabilizing Transformers

On the Importance of Normalisation Layers in Deep Learning with Piecewise Linear Activation Units

Adaptive Parametric Activation

UN-η: an Offline Adaptive Normalization Method for Deploying Transformers

An Investigation of the Impact of Normalization Schemes on GCN Modelling

ANAct: Adaptive Normalization for Activation Functions

Activated Gradients for Deep Neural Networks

Batchless Normalization: How to Normalize Activations Across Instances with Minimal Memory Requirements

Redefining The Self-Normalization Property

A Comprehensive and Modularized Statistical Framework for Gradient Norm Equality in Deep Neural Networks.

SSN: Learning Sparse Switchable Normalization via SparsestMax

Online Normalization for Training Neural Networks

Enhancing Neural Network Representations with Prior Knowledge-Based Normalization

Improving Generalization of Deep Neural Networks by Optimum Shifting

CHAIN: Enhancing Generalization in Data-Efficient GANs via lipsCHitz continuity constrAIned Normalization

Large-scale Self-Normalizing Neural Networks