Gradient flow in parameter space is equivalent to linear interpolation in output space
Machine Learning
2026-02-02 v3 Artificial Intelligence
Mathematical Physics
math.MP
Optimization and Control
Machine Learning
Abstract
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 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.
Keywords
Cite
@article{arxiv.2408.01517,
title = {Gradient flow in parameter space is equivalent to linear interpolation in output space},
author = {Thomas Chen and Patrícia Muñoz Ewald},
journal= {arXiv preprint arXiv:2408.01517},
year = {2026}
}
Comments
To appear in Journal of Geometry and Physics