On the Quality of the Initial Basin in Overspecified Neural Networks

By Itay Safran, Ohad Shamir. 2015

131 citations recorded by the source corpus.

Deep learning, in the form of artificial neural networks, has achieved remarkable practical success in recent years, for a variety of difficult machine learning applications. However, a theoretical explanation for this remains a major open problem, since training neural networks involves optimizing a highly non-convex objective function, and is known to be computationally hard in the worst case. In this work, we study the \emph{geometric} structure of the associated non-convex objective function, in the context of ReLU networks and starting from a random initialization of the network parameters. We identify some conditions under which it becomes more favorable to optimization, in the sense of (i) High probability of initializing at a point from which there is a monotonically decreasing path to a global minimum; and (ii) High probability of initializing at a basin (suitably defined) with a small minimal objective value. A common theme in our results is that such properties are more likely to hold for larger ("overspecified") networks, which accords with some recent empirical and theoretical observations.

Subfields: Artificial Intelligence, Computational Mechanics

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jason D. Lee
  2. Simon Shaolei Du
  3. Yuanzhi Li
  4. Tengyu Ma
  5. Zeyuan Allen-Zhu
  6. Quanquan Gu
  7. Ruslan Salakhutdinov
  8. Percy Liang

In Computational Mechanics

  1. Jason D. Lee
  2. Simon Shaolei Du
  3. Ohad Shamir
  4. Wei Hu
  5. Zeyuan Allen-Zhu
  6. Tengyu Ma
  7. Yuanzhi Li
  8. Sanjeev Arora

Citations in the corpus