On the Stability of the Jacobian Matrix in Deep Neural Networks
Deep neural networks are known to suffer from exploding or vanishing gradients as depth increases, a phenomenon closely tied to the spectral behavior of the input-output Jacobian. Prior work has identified critical initialization schemes that ensure Jacobian stability, but these analyses are typically restricted to fully connected networks with i.i.d. weights. In this work, we go significantly beyond these limitations: we establish a general stability theorem for deep neural networks that accommodates sparsity (such as that introduced by pruning) and non-i.i.d., weakly correlated weights (e.g. induced by training). Our results rely on recent advances in random matrix theory, and provide rigorous guarantees for spectral stability in a much broader class of network models. This extends the theoretical foundation for initialization schemes in modern neural networks with structured and dependent randomness.
Code (0)
등록된 구현이 없습니다.
Similar Papers 제목 키워드 기반
Extending Jacobian matrix in proving stability for nonlinear systems with one equilibrium point such as compressor
Global stability of the systems has always been vital of importance; however, this concept has not yet been sufficiently developed for the nonlinear systems. This paper extends the Jacobian matrix so that this method be …
Bounds on transient instability for complex ecosystems
Stability is a desirable property of complex ecosystems. If a community of interacting species is at a stable equilibrium point then it is able to withstand small perturbations to component species' abundances without su…
Advancing Nonlinear System Stability Analysis with Hessian Matrix Analysis
This paper introduces an innovative method for ensuring global stability in a broad array of nonlinear systems. The novel approach enhances the traditional analysis based on Jacobian matrices by incorporating the Taylor …
RecurJac: An Efficient Recursive Algorithm for Bounding Jacobian Matrix of Neural Networks and Its Applications
The Jacobian matrix (or the gradient for single-output networks) is directly related to many important properties of neural networks, such as the function landscape, stationary points, (local) Lipschitz constants and rob…
Training on the Edge of Stability Is Caused by Layerwise Jacobian Alignment
During neural network training, the sharpness of the Hessian matrix of the training loss rises until training is on the edge of stability. As a result, even nonstochastic gradient descent does not accurately model the un…