A Stochastic Newton Algorithm for Distributed Convex Optimization
Brian Bullins, Kumar Kshitij Patel, Ohad Shamir, Nathan Srebro, Blake E. Woodworth
Abstract
We propose and analyze a stochastic Newton algorithm for homogeneous distributed stochastic convex optimization, where each machine can calculate stochastic gradients of the same population objective, as well as stochastic Hessian-vector products (products of an independent unbiased estimator of the Hessian of the population objective with arbitrary vectors), with many such stochastic computations performed between rounds of communication. We show that our method can reduce the number, and frequency, of required communication rounds compared to existing methods without hurting performance, by proving convergence guarantees for quasi-self-concordant objectives (e.g., logistic regression), alongside empirical evidence.
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 55dd4443-2a50-4080-99f9-0072e60a513aCited by top-tier papers5
- Towards Optimal Communication Complexity in Distributed Non-Convex OptimizationKumar Kshitij Patel, Lingxiao Wang, Blake E. Woodworth, Brian Bullins et al.NeurIPS 2022 · 24 citations
- Federated Online and Bandit Convex OptimizationKumar Kshitij Patel, Lingxiao Wang, Aadirupa Saha, Nathan SrebroICML 2023 · 12 citations
- Local Composite Saddle Point OptimizationSite Bai, Brian BullinsICLR 2024 · 1 citation
- Revisiting Consensus Error: A Fine-grained Analysis of Local SGD under Second-order Data HeterogeneityKumar Kshitij Patel, Ali Zindari, Sebastian U. Stich, Lingxiao WangNeurIPS 2025 · 1 citation
- Distributionally Robust Linear Regression with Block Lewis WeightsNaren Sarayu Manoj, Kumar Kshitij PatelICLR 2026
Builds on6
- 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
- Is Local SGD Better than Minibatch SGD?Blake E. Woodworth, Kumar Kshitij Patel, Sebastian U. Stich, Zhen Dai et al.ICML 2020 · 277 citations
- Minibatch vs Local SGD for Heterogeneous Distributed LearningBlake E. Woodworth, Kumar Kshitij Patel, Nati SrebroNeurIPS 2020 · 231 citations
- Federated Accelerated Stochastic Gradient DescentHonglin Yuan, Tengyu MaNeurIPS 2020 · 217 citations
- Acceleration with a Ball Optimization OracleYair Carmon, Arun Jambulapati, Qijia Jiang, Yujia Jin et al.NeurIPS 2020 · 58 citations
Related papers
- Newton Method over Networks is Fast up to the Statistical PrecisionAmir Daneshmand, Gesualdo Scutari, Pavel E. Dvurechensky, Alexander V. GasnikovICML 2021 · 22 citations
- FedNS: A Fast Sketching Newton-Type Algorithm for Federated LearningJian Li, Yong Liu, Weiping WangAAAI 2024 · 7 citations
- Distributed Second Order Methods with Fast Rates and Compressed CommunicationRustem Islamov, Xun Qian, Peter RichtárikICML 2021 · 56 citations
- Communication-Efficient Distributed Optimization with Quantized PreconditionersFoivos Alimisis, Peter Davies, Dan AlistarhICML 2021 · 17 citations
- Jointly Improving the Sample and Communication Complexities in Decentralized Stochastic Minimax OptimizationXuan Zhang, Gabriel Mancino-Ball, Necdet Serhat Aybat, Yangyang XuAAAI 2024 · 14 citations
