Towards More Efficient Stochastic Decentralized Learning: Faster Convergence and Sparse Communication
arXiv:1805.09969
Abstract
Recently, the decentralized optimization problem is attracting growing attention. Most existing methods are deterministic with high per-iteration cost and have a convergence rate quadratically depending on the problem condition number. Besides, the dense communication is necessary to ensure the convergence even if the dataset is sparse. In this paper, we generalize the decentralized optimization problem to a monotone operator root finding problem, and propose a stochastic algorithm named DSBA that (i) converges geometrically with a rate linearly depending on the problem condition number, and (ii) can be implemented using sparse communication only. Additionally, DSBA handles learning problems like AUC-maximization which cannot be tackled efficiently in the decentralized setting. Experiments on convex minimization and AUC-maximization validate the efficiency of our method.
arXiv commentAccepted to ICML 2018
References in corpus (6)
- SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectives
- On the Linear Convergence of the ADMM in Decentralized Consensus Optimization
- Decentralized Quasi-Newton Methods
- DQM: Decentralized Quadratically Approximated Alternating Direction Method of Multipliers
- Gossip Dual Averaging for Decentralized Optimization of Pairwise Functions
- Decentralized Consensus Optimization with Asynchrony and Delays
Cited by in corpus (5)
- Stochastic Recursive Gradient Descent Ascent for Stochastic Nonconvex-Strongly-Concave Minimax Problems
- MATCHA: Speeding Up Decentralized SGD via Matching Decomposition Sampling
- Asynchronous Accelerated Proximal Stochastic Gradient for Strongly Convex Distributed Finite Sums
- Improving the Sample and Communication Complexity for Decentralized Non-Convex Optimization: A Joint Gradient Estimation and Tracking Approach
- Near Optimal Stochastic Algorithms for Finite-Sum Unbalanced Convex-Concave Minimax Optimization