English

Gradient Descent Finds Global Minima of Deep Neural Networks

Machine Learning 2019-05-30 v4 Artificial Intelligence Computer Vision and Pattern Recognition Optimization and Control Machine Learning

Abstract

Gradient descent finds a global minimum in training deep neural networks despite the objective function being non-convex. The current paper proves gradient descent achieves zero training loss in polynomial time for a deep over-parameterized neural network with residual connections (ResNet). Our analysis relies on the particular structure of the Gram matrix induced by the neural network architecture. This structure allows us to show the Gram matrix is stable throughout the training process and this stability implies the global optimality of the gradient descent algorithm. We further extend our analysis to deep residual convolutional neural networks and obtain a similar convergence result.

Keywords

Cite

@article{arxiv.1811.03804,
  title  = {Gradient Descent Finds Global Minima of Deep Neural Networks},
  author = {Simon S. Du and Jason D. Lee and Haochuan Li and Liwei Wang and Xiyu Zhai},
  journal= {arXiv preprint arXiv:1811.03804},
  year   = {2019}
}

Comments

ICML 2019