English

Extended critical regimes of deep neural networks

Machine Learning 2022-03-25 v1 Disordered Systems and Neural Networks Statistical Mechanics Artificial Intelligence Machine Learning

Abstract

Deep neural networks (DNNs) have been successfully applied to many real-world problems, but a complete understanding of their dynamical and computational principles is still lacking. Conventional theoretical frameworks for analysing DNNs often assume random networks with coupling weights obeying Gaussian statistics. However, non-Gaussian, heavy-tailed coupling is a ubiquitous phenomenon in DNNs. Here, by weaving together theories of heavy-tailed random matrices and non-equilibrium statistical physics, we develop a new type of mean field theory for DNNs which predicts that heavy-tailed weights enable the emergence of an extended critical regime without fine-tuning parameters. In this extended critical regime, DNNs exhibit rich and complex propagation dynamics across layers. We further elucidate that the extended criticality endows DNNs with profound computational advantages: balancing the contraction as well as expansion of internal neural representations and speeding up training processes, hence providing a theoretical guide for the design of efficient neural architectures.

Keywords

Cite

@article{arxiv.2203.12967,
  title  = {Extended critical regimes of deep neural networks},
  author = {Cheng Kevin Qu and Asem Wardak and Pulin Gong},
  journal= {arXiv preprint arXiv:2203.12967},
  year   = {2022}
}