Geometry of energy landscapes and the optimizability of deep neural networks

By Simon Becker, Yao Zhang, Alpha Albert Lee. 2018

33 citations recorded by the source corpus.

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.

Subfields: Artificial Intelligence, Computational Mathematics, Statistical and Nonlinear Physics

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. Medhi Bennis
  2. Jihong Park
  3. Sumudu Samarakoon
  4. Merouane Debbah
  5. Anis Elgabli
  6. H. Vincent Poor
  7. Seong-Lyun Kim
  8. Walid Saad

In Statistical and Nonlinear Physics

  1. Merouane Debbah
  2. Medhi Bennis
  3. Sumudu Samarakoon

Citations in the corpus