Suboptimal Local Minima Exist for Wide Neural Networks with Smooth Activations: For Generic Inputs and Any Depth and Width, Some Output Data Gives the Squared Loss a Sub-Optimal Local MinimumResearch Paper
Motivation
Training a neural network means minimizing a non-convex loss over its weights. Local search methods can only certify local optimality, so a recurring question in the theory of neural network optimization is whether every local minimum of the training loss is a global minimum. A local minimum that is not global is called a sub-optimal (or bad) local minimum.
A line of positive results suggested that sufficiently wide networks have no bad local minima. Yu and Chen (IEEE Trans. Neural Networks 6(5), 1995, Theorem 3) stated that for one-hidden-layer sigmoid networks with at least as many hidden neurons as training samples, every local minimum is global; the survey of Bianchini and Gori (Neurocomputing, 1996) lists this as a main result of the area. Later work proved weaker landscape properties for deep over-parameterized networks, such as the absence of sub-optimal valleys or basins (Nguyen, Mukkamala and Hein, arXiv:1809.10749; Nguyen, arXiv:1901.07417), and Nguyen, Mukkamala and Hein conjectured that "no bad local minima" holds under mild conditions.
Timeline of related results, as attributed in the source paper:
- 1995: Yu and Chen state that wide one-hidden-layer sigmoid networks have no bad local minima.
- 1996: Auer, Herbster and Warmuth construct exponentially many local minima for a single neuron.
- 2016: Kawaguchi (NeurIPS 2016) shows that deep linear networks have no bad local minima.
- 2018: Safran and Shamir (ICML 2018) find spurious local minima in two-layer ReLU networks with Gaussian inputs; Yun, Sra and Jadbabaie (arXiv:1802.03487) construct bad local minima for networks with piecewise linear activations.
- 2019–2020: Ding, Li and Sun (arXiv:1911.01413) construct bad local minima for networks of any depth and width with almost every smooth activation, which also refutes the 1995 claim.
Setting
A fully connected network with hidden layers has widths , input width and output width . Its parameters are
Given an activation and a data matrix whose columns are the samples, the post-activations are and
and the network output is , with no output bias. For targets , the empirical loss is
Two assumptions are made.
- Assumption 1 (data). , and the vectors , the rows , and the Hadamard products for are linearly independent in . Given the dimension condition, this holds for all outside a set of measure zero.
- Assumption 2 (activation). For some and , is twice continuously differentiable on and are all nonzero. Sigmoid, softplus, ELU and Swish satisfy it.
Formalization targets
Goal: Theorem 1
For every satisfying Assumption 1 with , every satisfying Assumption 2, any depth and any widths with , there is an output matrix such that
The data and the activation are given; only is chosen. The widths other than are unrestricted, so the statement covers arbitrarily wide and deep networks.
Milestones
- Lemma 1 (p. 20): given linearly independent and , there are linearly independent vectors orthogonal to and with positive inner product with .
- The output data (§6.1, eqs. (25)–(29)): under Assumption 1, can be chosen with rows orthogonal to , to every row of and to every mixed product , and with (conditions (23)).
- The constant-output configuration (eqs. (19)–(21)): with , , positive deeper weights, biases and , every hidden neuron outputs and .
- Lemma 2 (p. 21): for each hidden layer there is such a configuration near which the change of caused by perturbing everything except is positively correlated with each row of , quadratically in its size (32), and some nearby configuration makes itself positively correlated (33).
- Local minimality (§6.2, eqs. (34)–(42)): the configuration of Lemma 2 at is a local minimum of .
- Sub-optimality (§6.3, eqs. (43)–(50)): and some has a strictly smaller loss.
Significance
The theorem is a counter-example: for generic data and almost every smooth activation, increasing width or depth does not by itself remove bad local minima of the squared loss. It refutes Theorem 3 of Yu and Chen (1995) (Corollary 2 of the paper) and the conjecture of Nguyen, Mukkamala and Hein for this setting. Positive landscape results therefore need further assumptions on the data, the targets or the architecture, and the construction indicates which assumptions cannot be dropped.
The result is proved in the paper; no machine-checked proof of it is known. A formalization would verify a full counter-example for arbitrary depth, which involves a layer-by-layer second-order expansion whose bookkeeping across layers is where informal arguments of this type are most error-prone. It also provides a reusable Lean model of a deep fully connected network with biases and an arbitrary activation, which other landscape results can build on.
Difficulty
The output of the constructed configuration does not depend on the input, so a perturbation of changes in ways that need not be first-order: the obvious approach of checking a vanishing gradient and a positive semidefinite Hessian is inconclusive, because the loss is flat along several directions at the constructed point (for instance, changes of that keep its row sums), and the decisive effect of perturbing appears only at second order in the hidden layers. Local minimality must be shown for all nearby configurations at once, by controlling the second-order term of every layer uniformly in a neighbourhood and propagating the sign through the layers. The correlation condition must hold for every row of and every neuron simultaneously.
Formalization scope
- Model. Matrices are functions
Fin a → Fin b → ℝ; indices of neurons and samples are 0-based. The parameter space is the full product of all weights and all hidden biases , with its usual sup-norm metric; local minimality is Mathlib'sIsLocalMin. The forward pass is defined by recursion over the layers, feeding the post-activation of layer to layer (the printed eq. (2b) writes the pre-activation, a slip). - Assumptions. Assumption 1(a) is written over ℕ as ; the products in 1(b) are indexed by unordered pairs , so that the family has members, as the paper counts. Assumption 2(a) is read as on the interval, matching the paper's gloss; is arbitrary outside it.
- Added hypotheses. and positive widths . With output width zero the loss is identically zero, and with a zero-width hidden layer the output does not depend on ; the paper's setting presupposes positive widths.
- Lemma 2 is stated with hypothesis (23) on instead of (28) on (milestone 2 shows (28) implies (23)), with the printed index range read as , and with sup-norm balls around the full parameter in place of Euclidean balls around its first layers.
- Ruled out. Fixing in the parameter type, choosing or existentially, or indexing the Hadamard products by ordered pairs (which makes Assumption 1 unsatisfiable for and the theorem vacuous) would each trivialize the goal; none is done.
- Not included. Theorems 2–5 and Corollaries 1–2 of the paper (piecewise linear activations, positive-measure sets of targets for sigmoid networks, and the cases ).
Contributions welcome: proofs of the milestones, in particular a Lean development of the layer-wise Taylor expansion behind Lemma 2.
Selected references
- T. Ding, D. Li, R. Sun, Sub-Optimal Local Minima Exist for Neural Networks with Almost All Non-Linear Activations, arXiv:1911.01413v3, 2020; published as Suboptimal local minima exist for wide neural networks with smooth activations, Mathematics of Operations Research 47(4), 2022. https://arxiv.org/abs/1911.01413
- X.-H. Yu, G.-A. Chen, On the local minima free condition of backpropagation learning, IEEE Transactions on Neural Networks 6(5):1300–1303, 1995. https://doi.org/10.1109/72.410380
- Q. Nguyen, M. C. Mukkamala, M. Hein, On the loss landscape of a class of deep neural networks with no bad local valleys, ICLR 2019. https://arxiv.org/abs/1809.10749
- K. Kawaguchi, Deep learning without poor local minima, NeurIPS 2016. https://arxiv.org/abs/1605.07110
- I. Safran, O. Shamir, Spurious local minima are common in two-layer ReLU neural networks, ICML 2018. https://arxiv.org/abs/1712.08968
- C. Yun, S. Sra, A. Jadbabaie, Small nonlinearities in activation functions create bad local minima in neural networks, ICLR 2019. https://arxiv.org/abs/1802.03487