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.

Criticality & Deep Learning II: Momentum Renormalisation Group

Deep learning and the renormalization group

Criticality & Deep Learning I: Generally Weighted Nets

Mutual information, neural networks and the renormalization group

Critical Initialization of Wide and Deep Neural Networks through Partial Jacobians: General Theory and Applications

A Deep Dive into the Connections Between the Renormalization Group and Deep Learning in the Ising Model

Is Deep Learning a Renormalization Group Flow?

Optimization and Stabilization of Functional Renormalization Group Flows

Optimal Real-Space Renormalization-Group Transformations with Artificial Neural Networks

An exact mapping between the Variational Renormalization Group and Deep Learning

Random Reshuffling with Momentum for Nonconvex Problems: Iteration Complexity and Last Iterate Convergence

Renormalization Group: Applications in Statistical Physics

Conformal Symplectic and Relativistic Optimization

On discretisation drift and smoothness regularisation in neural network training

Continuous Time Analysis of Momentum Methods

Interpreting Deep Learning by Establishing a Rigorous Corresponding Relationship with Renormalization Group

Random Scaling and Momentum for Non-smooth Non-convex Optimization

Laplacian Renormalization Group: An introduction to heterogeneous coarse-graining

Fast renormalizing the structures and dynamics of ultra-large systems via random renormalization group

Positive-Negative Momentum: Manipulating Stochastic Gradient Noise to Improve Generalization

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