Sharp description of local minima in the loss landscape of high-dimensional two-layer ReLU neural networks
Abstract
Lay Summary
Training a neural network means searching through many possible settings of its weights, and this search can get stuck in places that look good locally but are not the best solution. We study this problem in a controlled setting: a simple two-layer neural network learns a target function generated by another network of the same kind. This lets us replace the huge space of weights with a small set of summary quantities that exactly track the network's error and learning dynamics for Gaussian data. Using this reduced description, we identify hierarchical families of suboptimal solutions and explain why gradient-based training often ends at a small number of predictable error levels. We then show how adding extra hidden units changes which suboptimal solutions are stable and how often training reaches the correct solution. The picture is more nuanced than "larger networks always have an easy landscape": overparameterisation removes some traps while other, more structured traps can remain. These results provide a mathematical explanation of why extra model capacity can help optimisation, and a warning that common simplifications may miss important finite-size effects in neural-network training.