We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: Theoretical insights into the optimization landscape of over-parameterized shallow neural networks

Abstract: In this paper we study the problem of learning a shallow artificial neural network that best fits a training data set. We study this problem in the over-parameterized regime where the number of observations are fewer than the number of parameters in the model. We show that with quadratic activations the optimization landscape of training such shallow neural networks has certain favorable characteristics that allow globally optimal models to be found efficiently using a variety of local search heuristics. This result holds for an arbitrary training data of input/output pairs. For differentiable activation functions we also show that gradient descent, when suitably initialized, converges at a linear rate to a globally optimal model. This result focuses on a realizable model where the inputs are chosen i.i.d. from a Gaussian distribution and the labels are generated according to planted weight coefficients.
Comments: A mistake in the argument of Proposition 7.1 in the previous version of this manuscript was fixed
Subjects: Machine Learning (cs.LG); Information Theory (cs.IT); Optimization and Control (math.OC); Machine Learning (stat.ML)
Cite as: arXiv:1707.04926 [cs.LG]
  (or arXiv:1707.04926v3 [cs.LG] for this version)

Submission history

From: Mahdi Soltanolkotabi [view email]
[v1] Sun, 16 Jul 2017 18:13:51 GMT (50kb)
[v2] Thu, 8 Nov 2018 00:11:02 GMT (133kb,D)
[v3] Wed, 24 Aug 2022 01:32:05 GMT (136kb,D)

Link back to: arXiv, form interface, contact.