Distributed Second Order Methods with Fast Rates and Compressed Communication
Rustem Islamov, Xun Qian, Peter Richtárik
Abstract
We develop several new communication-efficient second-order methods for distributed optimization. Our first method, NEWTON-STAR, is a variant of Newton's method from which it inherits its fast local quadratic rate. However, unlike Newton's method, NEWTON-STAR enjoys the same per iteration communication cost as gradient descent. While this method is impractical as it relies on the use of certain unknown parameters characterizing the Hessian of the objective function at the optimum, it serves as the starting point which enables us design practical variants thereof with strong theoretical guarantees. In particular, we design a stochastic sparsification strategy for learning the unknown parameters in an iterative fashion in a communication efficient manner. Applying this strategy to NEWTON-STAR leads to our next method, NEWTON-LEARN, for which we prove local linear and superlinear rates independent of the condition number. When applicable, this method can have dramatically superior convergence behavior when compared to state-of-the-art methods. Finally, we develop a globalization strategy using cubic regularization which leads to our next method, CUBIC-NEWTON-LEARN, for which we prove global sublinear and linear convergence rates, and a fast superlinear rate. Our results are supported with experimental results on real datasets, and show several orders of magnitude improvement on baseline and state-of-the-art methods in terms of communication complexity. Contents
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 cd2cfb96-eb06-41de-865b-9b72ab2ebf01Cited by top-tier papers17
- EF21: A New, Simpler, Theoretically Better, and Practically Faster Error FeedbackPeter Richtárik, Igor Sokolov, Ilyas FatkhullinNeurIPS 2021 · 219 citations
- What Do We Mean by Generalization in Federated Learning?Honglin Yuan, Warren Richard Morningstar, Lin Ning, Karan SinghalICLR 2022 · 98 citations
- FedNL: Making Newton-Type Methods Applicable to Federated LearningMher Safaryan, Rustem Islamov, Xun Qian, Peter RichtárikICML 2022 · 90 citations
- Improved Communication Efficiency in Federated Natural Policy Gradient via ADMM-based Gradient UpdatesGuangchen Lan, Han Wang, James Anderson, Christopher G. Brinton et al.NeurIPS 2023 · 32 citations
- A Stochastic Newton Algorithm for Distributed Convex OptimizationBrian Bullins, Kumar Kshitij Patel, Ohad Shamir, Nathan Srebro et al.NeurIPS 2021 · 20 citations
Builds on4
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 172 citations
- Adaptive Gradient Descent without DescentYura Malitsky, Konstantin MishchenkoICML 2020 · 171 citations
- Acceleration for Compressed Gradient Descent in Distributed and Federated OptimizationZhize Li, Dmitry Kovalev, Xun Qian, Peter RichtárikICML 2020 · 156 citations
- Stochastic Subspace Cubic Newton MethodFilip Hanzely, Nikita Doikov, Yurii E. Nesterov, Peter RichtárikICML 2020 · 62 citations
Related papers
- Communication Efficient Distributed Newton Method with Fast Convergence RatesChengchang Liu, Lesi Chen, Luo Luo, John C. S. LuiKDD 2023 · 4 citations
- Communication-Efficient Distributed Optimization with Quantized PreconditionersFoivos Alimisis, Peter Davies, Dan AlistarhICML 2021 · 17 citations
- Newton Method over Networks is Fast up to the Statistical PrecisionAmir Daneshmand, Gesualdo Scutari, Pavel E. Dvurechensky, Alexander V. GasnikovICML 2021 · 22 citations
- Second-Order Optimization with Lazy HessiansNikita Doikov, El Mahdi Chayti, Martin JaggiICML 2023 · 31 citations
- Communication Efficient Distributed Newton Method over Unreliable NetworksMing Wen, Chengchang Liu, Yuedong XuAAAI 2024 · 2 citations
