Gradient flow in parameter space is equivalent to linear interpolation in output space
Not on the map. Why?
Abstract (arXiv)
We prove that the standard gradient flow in parameter space that underlies many training algorithms in deep learning can be continuously deformed into an adapted gradient flow which yields (constrained) Euclidean gradient flow in output space. Moreover, for the L2 loss, if the Jacobian of the outputs with respect to the parameters is full rank (for fixed training data), then the time variable can be reparametrized so that the resulting flow is simply linear interpolation, and a global minimum can be achieved. For the cross-entropy loss, under the same rank condition and assuming the labels have positive components, we derive an explicit formula for the unique global minimum.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 15 of its 25 references that are in the map. First posted on arXiv 2 Aug 2024.
Builds on
15 of 25 references.
References · 15
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2018On Lazy Training in Differentiable Programmingcited by 671
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 302
- 2014Qualitatively characterizing neural network optimization problemscited by 186
- 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 156
- 2020On the emergence of simplex symmetry in the final and penultimate layers of neural network classifierscited by 51
- 2018The loss landscape of overparameterized neural networkscited by 38
- 2019Pure and Spurious Critical Points: a Geometric Study of Linear Networkscited by 33
- 2022Functional dimension of feedforward ReLU neural networkscited by 20
- 2018Natural gradient via optimal transportcited by 19
3 more
- 2023Mildly Overparameterized ReLU Networks Have a Favorable Loss Landscapecited by 16
- 2024Combining Wasserstein-1 and Wasserstein-2 proximals: robust manifold learning via well-posed generative flowscited by 4
- 2023Global L2 minimization at uniform exponential rate via geometrically adapted gradient descent in Deep Learningcited by 2
Built on by
Citing papers per year: 2025: 2.
2 papers.