Symmetry & critical points for a model shallow neural network
Abstract
Using methods based on the analysis of real analytic functions, symmetry and equivariant bifurcation theory, we obtain sharp results on families of critical points of spurious minima that occur in optimization problems associated with fitting two-layer ReLU networks with k hidden neurons. The main mathematical result proved is to obtain power series representations of families of critical points of spurious minima in terms of 1 /√{ k } (coefficients independent of k). We also give a path based formulation that naturally connects the critical points with critical points of an associated linear, but highly singular, optimization problem. These critical points closely approximate the critical points in the original problem.
The mathematical theory is used to derive results on the original problem in neural nets. For example, precise estimates for several quantities that show that not all spurious minima are alike. In particular, we show that while the loss function at certain types of spurious minima decays to zero like k-1, in other cases the loss converges to a strictly positive constant.- Publication:
-
Physica D Nonlinear Phenomena
- Pub Date:
- December 2021
- DOI:
- 10.1016/j.physd.2021.133014
- arXiv:
- arXiv:2003.10576
- Bibcode:
- 2021PhyD..42733014A
- Keywords:
-
- Student-teacher network;
- ReLU activation;
- Critical points;
- Spurious minima;
- Power series representation;
- Symmetry breaking;
- Computer Science - Machine Learning;
- Mathematics - Dynamical Systems;
- Mathematics - Optimization and Control;
- Statistics - Machine Learning
- E-Print:
- doi:10.1016/j.physd.2021.133014