1. A Sharp Estimate on the Transient Time of Distributed Stochastic Gradient Descent
- Author
-
Alex Olshevsky, Shi Pu, and Ioannis Ch. Paschalidis
- Subjects
FOS: Computer and information sciences ,0209 industrial biotechnology ,Computer Science - Machine Learning ,02 engineering and technology ,Machine Learning (cs.LG) ,Matrix (mathematics) ,020901 industrial engineering & automation ,Mixing (mathematics) ,Convergence (routing) ,FOS: Mathematics ,Applied mathematics ,Computer Science - Multiagent Systems ,Electrical and Electronic Engineering ,Mathematics - Optimization and Control ,Mathematics ,Computer Science Applications ,Stochastic gradient descent ,Rate of convergence ,Computer Science - Distributed, Parallel, and Cluster Computing ,Control and Systems Engineering ,Optimization and Control (math.OC) ,Bounded function ,Spectral gap ,Distributed, Parallel, and Cluster Computing (cs.DC) ,Convex function ,Multiagent Systems (cs.MA) - Abstract
This paper is concerned with minimizing the average of $n$ cost functions over a network in which agents may communicate and exchange information with each other. We consider the setting where only noisy gradient information is available. To solve the problem, we study the distributed stochastic gradient descent (DSGD) method and perform a non-asymptotic convergence analysis. For strongly convex and smooth objective functions, DSGD asymptotically achieves the optimal network independent convergence rate compared to centralized stochastic gradient descent (SGD). Our main contribution is to characterize the transient time needed for DSGD to approach the asymptotic convergence rate, which we show behaves as $K_T=\mathcal{O}\left(\frac{n}{(1-\rho_w)^2}\right)$, where $1-\rho_w$ denotes the spectral gap of the mixing matrix. Moreover, we construct a "hard" optimization problem for which we show the transient time needed for DSGD to approach the asymptotic convergence rate is lower bounded by $\Omega \left(\frac{n}{(1-\rho_w)^2} \right)$, implying the sharpness of the obtained result. Numerical experiments demonstrate the tightness of the theoretical results.
- Published
- 2019