Linearly Converging Error Compensated SGD
Eduard Gorbunov, Dmitry Kovalev, Dmitry Makarenko, Peter Richtárik
Abstract
In this paper, we propose a unified analysis of variants of distributed SGD with arbitrary compressions and delayed updates. Our framework is general enough to cover different variants of quantized SGD, Error-Compensated SGD (EC-SGD) and SGD with delayed updates (D-SGD). Via a single theorem, we derive the complexity results for all the methods that fit our framework. For the existing methods, this theorem gives the best-known complexity results. Moreover, using our general scheme, we develop new variants of SGD that combine variance reduction or arbitrary sampling with error feedback and quantization and derive the convergence rates for these methods beating the state-of-the-art results. In order to illustrate the strength of our framework, we develop 16 new methods that fit this. In particular, we propose the first method called EC-SGD-DIANA that is based on error-feedback for biased compression operator and quantization of gradient differences and prove the convergence guarantees showing that EC-SGD-DIANA converges to the exact optimum asymptotically in expectation with constant learning rate for both convex and strongly convex objectives when workers compute full gradients of their loss functions. Moreover, for the case when the loss function of the worker has the form of finite sum, we modified the method and got a new one called EC-LSVRG-DIANA which is the first distributed stochastic method with error feedback and variance reduction that converges to the exact optimum asymptotically in expectation with a constant learning rate. (2) Dataset D i may or may not be available to worker i in its entirety. Typically, we assume that worker i has only access to samples from D i . If the dataset is fully available, it is typically 34th Conference on Neural Information Processing Systems (NeurIPS 2020),
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 31aa68f1-dd07-424e-9c99-cb56321c6e62Cited by top-tier papers36
- EF21: A New, Simpler, Theoretically Better, and Practically Faster Error FeedbackPeter Richtárik, Igor Sokolov, Ilyas FatkhullinNeurIPS 2021 · 219 citations
- Linear Convergence in Federated Learning: Tackling Client Heterogeneity and Sparse GradientsAritra Mitra, Rayana H. Jaafar, George J. Pappas, Hamed HassaniNeurIPS 2021 · 193 citations
- MARINA: Faster Non-Convex Distributed Learning with CompressionEduard Gorbunov, Konstantin Burlachenko, Zhize Li, Peter RichtárikICML 2021 · 129 citations
- 1-bit Adam: Communication Efficient Large-Scale Training with Adam's Convergence SpeedHanlin Tang, Shaoduo Gan, Ammar Ahmad Awan, Samyam Rajbhandari et al.ICML 2021 · 106 citations
- What Do We Mean by Generalization in Federated Learning?Honglin Yuan, Warren Richard Morningstar, Lin Ning, Karan SinghalICLR 2022 · 98 citations
Builds on3
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- A Unified Theory of Decentralized SGD with Changing Topology and Local UpdatesAnastasia Koloskova, Nicolas Loizou, Sadra Boreiri, Martin Jaggi et al.ICML 2020 · 623 citations
- Decentralized Deep Learning with Arbitrary Communication CompressionAnastasia Koloskova, Tao Lin, Sebastian U. Stich, Martin JaggiICLR 2020 · 263 citations
Related papers
- EF-BV: A Unified Theory of Error Feedback and Variance Reduction Mechanisms for Biased and Unbiased Compression in Distributed OptimizationLaurent Condat, Kai Yi, Peter RichtárikNeurIPS 2022 · 30 citations
- Detached Error Feedback for Distributed SGD with Random SparsificationAn Xu, Heng HuangICML 2022 · 12 citations
- ErrorCompensatedX: error compensation for variance reduced algorithmsHanlin Tang, Yao Li, Ji Liu, Ming YanNeurIPS 2021 · 13 citations
- Quantized Compressive Sampling of Stochastic Gradients for Efficient Communication in Distributed Deep LearningAfshin Abdi, Faramarz FekriAAAI 2020 · 32 citations
- Acceleration for Compressed Gradient Descent in Distributed and Federated OptimizationZhize Li, Dmitry Kovalev, Xun Qian, Peter RichtárikICML 2020 · 156 citations
