Global Convergence of Gradient Descent for Deep Linear Residual Networks
Machine Learning
2019-11-05 v1 Machine Learning
Abstract
We analyze the global convergence of gradient descent for deep linear residual networks by proposing a new initialization: zero-asymmetric (ZAS) initialization. It is motivated by avoiding stable manifolds of saddle points. We prove that under the ZAS initialization, for an arbitrary target matrix, gradient descent converges to an -optimal point in iterations, which scales polynomially with the network depth . Our result and the convergence time for the standard initialization (Xavier or near-identity) [Shamir, 2018] together demonstrate the importance of the residual structure and the initialization in the optimization for deep linear neural networks, especially when is large.
Cite
@article{arxiv.1911.00645,
title = {Global Convergence of Gradient Descent for Deep Linear Residual Networks},
author = {Lei Wu and Qingcan Wang and Chao Ma},
journal= {arXiv preprint arXiv:1911.00645},
year = {2019}
}