Back to Search Start Over

Mean Field Analysis of Deep Neural Networks.

Authors :
Sirignano, Justin
Spiliopoulos, Konstantinos
Source :
Mathematics of Operations Research; Feb2022, Vol. 47 Issue 1, p120-152, 33p
Publication Year :
2022

Abstract

We analyze multilayer neural networks in the asymptotic regime of simultaneously (a) large network sizes and (b) large numbers of stochastic gradient descent training iterations. We rigorously establish the limiting behavior of the multilayer neural network output. The limit procedure is valid for any number of hidden layers, and it naturally also describes the limiting behavior of the training loss. The ideas that we explore are to (a) take the limits of each hidden layer sequentially and (b) characterize the evolution of parameters in terms of their initialization. The limit satisfies a system of deterministic integro-differential equations. The proof uses methods from weak convergence and stochastic analysis. We show that, under suitable assumptions on the activation functions and the behavior for large times, the limit neural network recovers a global minimum (with zero loss for the objective function). [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
0364765X
Volume :
47
Issue :
1
Database :
Complementary Index
Journal :
Mathematics of Operations Research
Publication Type :
Academic Journal
Accession number :
155256226
Full Text :
https://doi.org/10.1287/moor.2020.1118