201

Analytic Characterization of the Hessian in Shallow ReLU Models: A Tale of Symmetry

Abstract

We consider the optimization problem associated with fitting two-layers ReLU networks with kk neurons. We leverage the rich symmetry structure to analytically characterize the Hessian and its spectral density at various families of spurious local minima. In particular, we prove that for standard dd-dimensional Gaussian inputs with dkd\ge k: (a) of the dkdk eigenvalues corresponding to the weights of the first layer, dkO(d)dk - O(d) concentrate near zero, (b) Ω(d)\Omega(d) of the remaining eigenvalues grow linearly with kk. Although this phenomenon of extremely skewed spectrum has been observed many times before, to the best of our knowledge, this is the first time it has been established rigorously. Our analytic approach uses techniques, new to the field, from symmetry breaking and representation theory, and carries important implications for our ability to argue about statistical generalization through local curvature.

View on arXiv
Comments on this paper