Finding Local Minima Efficiently in Decentralized Optimization
Wenhan Xian, Heng Huang
Abstract
In this paper we study the second-order optimality of decentralized stochastic algorithm that escapes saddle point efficiently for nonconvex optimization problems. We propose a new pure gradient-based decentralized stochastic algorithm PEDESTAL with a novel convergence analysis framework to address the technical challenges unique to the decentralized stochastic setting. Our method is the first decentralized stochastic algorithm to achieve second-order optimality with non-asymptotic analysis. We provide theoretical guarantees with the gradient complexity of ˜ O ( ϵ − 3 ) to find O ( ϵ, √ ϵ ) -second-order stationary point, which matches state-of-the-art results of centralized counterparts or decentralized methods to find first-order stationary point. We also conduct two decentralized tasks in our experiments, a matrix sensing task with synthetic data and a matrix factorization task with a real-world dataset to validate the performance of our method.
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 4d58c4e0-b3a4-4f64-85f5-36e656710c23Builds on5
- On the Almost Sure Convergence of Stochastic Gradient Descent in Non-Convex ProblemsPanayotis Mertikopoulos, Nadav Hallak, Ali Kavis, Volkan CevherNeurIPS 2020 · 120 citations
- Improving the Sample and Communication Complexity for Decentralized Non-Convex Optimization: Joint Gradient Estimation and TrackingHaoran Sun, Songtao Lu, Mingyi HongICML 2020 · 57 citations
- A Hybrid Variance-Reduced Method for Decentralized Stochastic Non-Convex OptimizationRan Xin, Usman A. Khan, Soummya KarICML 2021 · 51 citations
- D-SPIDER-SFO: A Decentralized Optimization Algorithm with Faster Convergence Rate for Nonconvex ProblemsTaoxing Pan, Jun Liu, Jie WangAAAI 2020 · 19 citations
- Second Order Optimality in Decentralized Non-Convex Optimization via Perturbed Gradient TrackingIsidoros Tziotis, Constantine Caramanis, Aryan MokhtariNeurIPS 2020 · 9 citations
Related papers
- Escape saddle points by a simple gradient-descent based algorithmChenyi Zhang, Tongyang LiNeurIPS 2021 · 19 citations
- Faster Gradient-Free Methods for Escaping Saddle PointsHualin Zhang, Bin GuICLR 2023
- Escaping Saddle Points with Bias-Variance Reduced Local Perturbed SGD for Communication Efficient Nonconvex Distributed LearningTomoya Murata, Taiji SuzukiNeurIPS 2022 · 4 citations
- Communication Efficient Distributed Newton Method with Fast Convergence RatesChengchang Liu, Lesi Chen, Luo Luo, John C. S. LuiKDD 2023 · 4 citations
- A Faster Decentralized Algorithm for Nonconvex Minimax ProblemsWenhan Xian, Feihu Huang, Yanfu Zhang, Heng HuangNeurIPS 2021 · 72 citations
