The Global Optimization Geometry of Shallow Linear Neural Networks

Zhihui Zhu, Daniel Soudry, Yonina C. Eldar, Michael B. Wakin

Research output: Contribution to journalArticlepeer-review

Abstract

We examine the squared error loss landscape of shallow linear neural networks. We show—with significantly milder assumptions than previous works—that the corresponding optimization problems have benign geometric properties: There are no spurious local minima, and the Hessian at every saddle point has at least one negative eigenvalue. This means that at every saddle point there is a directional negative curvature which algorithms can utilize to further decrease the objective value. These geometric properties imply that many local search algorithms (such as the gradient descent which is widely utilized for training neural networks) can provably solve the training problem with global convergence.

Original languageEnglish
Pages (from-to)279-292
Number of pages14
JournalJournal of Mathematical Imaging and Vision
Volume62
Issue number3
Early online date31 May 2019
DOIs
StatePublished - 1 Apr 2020

Keywords

  • Deep learning
  • Linear neural network
  • Optimization geometry
  • Spurious local minima
  • Strict saddle

All Science Journal Classification (ASJC) codes

  • Statistics and Probability
  • Modelling and Simulation
  • Condensed Matter Physics
  • Computer Vision and Pattern Recognition
  • Geometry and Topology
  • Applied Mathematics

Fingerprint

Dive into the research topics of 'The Global Optimization Geometry of Shallow Linear Neural Networks'. Together they form a unique fingerprint.

Cite this