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

Darshil Doshi, Tianyu He, Andrey Gromov

Deep neural networks are notorious for defying theoretical treatment. However, when the number of parameters in each layer tends to infinity, the network function is a Gaussian pro…