Nonlinear Dynamics In Optimization Landscape of Shallow Neural Networks with Tunable Leaky ReLU

arXiv:2510.25060 · math.OC, cs.LG, math.DS · Submitted 2025-10-29 · Read on arXiv

math.OC, cs.LG, math.DS

Submitted: 2025-10-29

Updated: 2026-09-01

License: http://creativecommons.org/licenses/by/4.0/

The gist: In this work, we study the nonlinear dynamics of a shallow neural network trained with mean-squared loss and leaky ReLU activation.

Terminology

Abstract

In this work, we study the nonlinear dynamics of a shallow neural network trained with mean-squared loss and leaky ReLU activation. Under Gaussian inputs and equal layer width k, (1) we establish, based on the equivariant gradient degree, a theoretical framework, applicable to any number of neurons k>= 4, to detect bifurcation of critical points with associated symmetries from global minimum as leaky parameter α varies. Typically, our analysis reveals that a multi-mode degeneracy consistently occurs at the critical number 0, independent of k. (2) As a by-product, we further show that such bifurcations are width-independent, arise only for nonnegative α and that the global minimum undergoes no further symmetry-breaking instability throughout the engineering regime α in range (0,1). An explicit example with k=5 is presented to illustrate the framework and exhibit the resulting bifurcation together with their symmetries.

Sources

Related papers