Gradient descent on overparameterized two-layer linear networks converges linearly for any smooth strongly convex loss under mild initialization and step-size conditions, at a rate at most a condition-number factor worse than the non-overparameterized model.
Convergence of gradient descent for learning linear neural networks
1 Pith paper cite this work. Polarity classification is still indexing.
abstract
We study the convergence properties of gradient descent for training deep linear neural networks, i.e., deep matrix factorizations, by extending a previous analysis for the related gradient flow. We show that under suitable conditions on the step sizes gradient descent converges to a critical point of the loss function, i.e., the square loss in this article. Furthermore, we demonstrate that for almost all initializations gradient descent converges to a global minimum in the case of two layers. In the case of three or more layers we show that gradient descent converges to a global minimum on the manifold matrices of some fixed rank, where the rank cannot be determined a priori.
fields
cs.LG 1years
2025 1verdicts
ACCEPT 1representative citing papers
citing papers explorer
-
A Local Polyak-Lojasiewicz and Descent Lemma of Gradient Descent For Overparametrized Linear Models
Gradient descent on overparameterized two-layer linear networks converges linearly for any smooth strongly convex loss under mild initialization and step-size conditions, at a rate at most a condition-number factor worse than the non-overparameterized model.