Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper analyzes the gradient flow dynamics of two-layer ReLU networks trained on orthogonal data from small initialization. It proves that in the limit of small initialization, the training process converges to a saddle-to-saddle jump process, revealing an incremental learning phenomenon where neurons activate sequentially at each saddle point. The analysis recovers the known width requirement for interpolation and derives a novel implicit bias result: the learned interpolator's squared $\ell_2$-norm scales as $\sqrt{n}$, close to the minimal norm interpolator.
ReLU networks trained with gradient flow on orthogonal data exhibit an incremental learning process, activating neurons one-by-one at saddle points, and converging to interpolators with near-minimal $\ell_2$ norm.
The successful training of neural networks hinges on the use of first order optimization methods, yet the theoretical characterization of these methods remains incomplete. This is especially true in settings with mild overparameterization. In this work, we study the gradient flow dynamics of two-layer ReLU networks from small initialization with orthogonal training data. We prove the limiting flow converges to a saddle-to-saddle jump process as the initialization scale tends to zero, revealing an incremental learning phenomenon in which a new neuron activates at each saddle. This analysis recovers the known result of Dana et al. (2025, arXiv:2502.16977) that the network interpolates the training data with high probability as soon as $m \gtrsim \log(n)$, where $m$ is the network width and $n$ is the number of training samples. This incremental process characterization also allows us to derive a novel implicit bias result: the learned interpolator has a squared $\ell_2$-norm scaling as $\sqrt{n}$, which is within a constant factor of the minimal $\ell_2$-norm interpolator. More broadly, our work provides the first rigorous proof of an incremental learning process for ReLU networks, whilst suggesting mildly overparameterized networks can converge to interpolating solutions whose complexity is of the same order as that of the optimal interpolator.