Convergence and Implicit Regularization Properties of Gradient Descent for Deep Residual Networks
Machine Learning
2023-01-26 v4 Optimization and Control
Abstract
We prove linear convergence of gradient descent to a global optimum for the training of deep residual networks with constant layer width and smooth activation function. We show that if the trained weights, as a function of the layer index, admit a scaling limit as the depth increases, then the limit has finite variation with . Proofs are based on non-asymptotic estimates for the loss function and for norms of the network weights along the gradient descent path. We illustrate the relevance of our theoretical results to practical settings using detailed numerical experiments on supervised learning problems.
Cite
@article{arxiv.2204.07261,
title = {Convergence and Implicit Regularization Properties of Gradient Descent for Deep Residual Networks},
author = {Rama Cont and Alain Rossier and RenYuan Xu},
journal= {arXiv preprint arXiv:2204.07261},
year = {2023}
}