Geometry of Energy Landscapes and the Optimizability of Deep Neural Networks

Simon Becker, Yao Zhang, and Alpha A. Lee
Phys. Rev. Lett. 124, 108301 – Published 10 March 2020
PDFHTMLExport Citation

Abstract

Deep neural networks are workhorse models in machine learning with multiple layers of nonlinear functions composed in series. Their loss function is highly nonconvex, yet empirically even gradient descent minimization is sufficient to arrive at accurate and predictive models. It is hitherto unknown why deep neural networks are easily optimizable. We analyze the energy landscape of a spin glass model of deep neural networks using random matrix theory and algebraic geometry. We analytically show that the multilayered structure holds the key to optimizability: Fixing the number of parameters and increasing network depth, the number of stationary points in the loss function decreases, minima become more clustered in parameter space, and the trade-off between the depth and width of minima becomes less severe. Our analytical results are numerically verified through comparison with neural networks trained on a set of classical benchmark datasets. Our model uncovers generic design principles of machine learning models.

  • Figure
  • Figure
  • Figure
  • Figure
  • Received 9 August 2018
  • Revised 13 November 2019
  • Accepted 6 February 2020

DOI:https://doi.org/10.1103/PhysRevLett.124.108301

© 2020 American Physical Society

Physics Subject Headings (PhySH)

Interdisciplinary PhysicsStatistical Physics & ThermodynamicsNetworks

Authors & Affiliations

Simon Becker1, Yao Zhang2,1, and Alpha A. Lee2,*

  • 1Department of Applied Mathematics and Theoretical Physics, University of Cambridge, Wilberforce Road, Cambridge CB3 0WA, United Kingdom
  • 2Cavendish Laboratory, University of Cambridge, Cambridge CB3 0HE, United Kingdom

  • *aal44@cam.ac.uk

Article Text (Subscription Required)

Click to Expand

Supplemental Material (Subscription Required)

Click to Expand

References (Subscription Required)

Click to Expand
Issue

Vol. 124, Iss. 10 — 13 March 2020

Reuse & Permissions
Access Options
Author publication services for translation and copyediting assistance advertisement

Authorization Required


×
×

Images

×

Sign up to receive regular email alerts from Physical Review Letters

Log In

Cancel
×

Search


Article Lookup

Paste a citation or DOI

Enter a citation
×