Repository logo
 

Archetypal landscapes for deep neural networks.

Accepted version
Peer-reviewed

Loading...
Thumbnail Image

Type

Article

Change log

Authors

Verpoort, Philipp C  ORCID logo  https://orcid.org/0000-0003-1319-5006
Lee, Alpha A 

Abstract

The predictive capabilities of deep neural networks (DNNs) continue to evolve to increasingly impressive levels. However, it is still unclear how training procedures for DNNs succeed in finding parameters that produce good results for such high-dimensional and nonconvex loss functions. In particular, we wish to understand why simple optimization schemes, such as stochastic gradient descent, do not end up trapped in local minima with high loss values that would not yield useful predictions. We explain the optimizability of DNNs by characterizing the local minima and transition states of the loss-function landscape (LFL) along with their connectivity. We show that the LFL of a DNN in the shallow network or data-abundant limit is funneled, and thus easy to optimize. Crucially, in the opposite low-data/deep limit, although the number of minima increases, the landscape is characterized by many minima with similar loss values separated by low barriers. This organization is different from the hierarchical landscapes of structural glass formers and explains why minimization procedures commonly employed by the machine-learning community can navigate the LFL successfully and reach low-lying solutions.

Description

Keywords

deep learning, energy landscapes, neural networks, optimization, statistical mechanics

Journal Title

Proc Natl Acad Sci U S A

Conference Name

Journal ISSN

0027-8424
1091-6490

Volume Title

117

Publisher

Proceedings of the National Academy of Sciences

Rights

All rights reserved
Sponsorship
A.A.L. was supported by the Winton Program for the Physics of Sustainability. P.C.V. and D.J.W. were supported by the Engineering and Physical Sciences Research Council.
Relationships
Is supplemented by: