SGDA with shuffling: faster convergence for nonconvex-PŁ minimax optimization
Hanseul Cho, Chulhee Yun
摘要
Stochastic gradient descent-ascent (SGDA) is one of the main workhorses for solving finite-sum minimax optimization problems. Most practical implementations of SGDA randomly reshuffle components and sequentially use them (i.e., without-replacement sampling); however, there are few theoretical results on this approach for minimax algorithms, especially outside the easier-to-analyze (strongly-)monotone setups. To narrow this gap, we study the convergence bounds of SGDA with random reshuffling (SGDA-RR) for smooth nonconvex-nonconcave objectives with Polyak-Łojasiewicz (PŁ) geometry. We analyze both simultaneous and alternating SGDA-RR for nonconvex-PŁ and primal-PŁ-PŁ objectives, and obtain convergence rates faster than with-replacement SGDA. Our rates extend to mini-batch SGDA-RR, recovering known rates for full-batch gradient descent-ascent (GDA). Lastly, we present a comprehensive lower bound for GDA with an arbitrary step-size ratio, which matches the full-batch upper bound for the primal-PŁ-PŁ case.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper7
- Tighter Lower Bounds for Shuffling SGD: Random Permutations and BeyondJaeyoung Cha, Jaewook Lee, Chulhee YunICML 2023 · 被引用 26 次
- On the Last-Iterate Convergence of Shuffling Gradient MethodsZijian Liu, Zhengyuan ZhouICML 2024 · 被引用 11 次
- On the Training Instability of Shuffling SGD with Batch NormalizationDavid Xing Wu, Chulhee Yun, Suvrit SraICML 2023 · 被引用 6 次
- Stochastic Extragradient with Flip-Flop Shuffling & Anchoring: Provable ImprovementsJiseok Chae, Chulhee Yun, Donghwan KimNeurIPS 2024 · 被引用 2 次
- Provable Benefit of Random Permutations over Uniform Sampling in Stochastic Coordinate DescentDonghwa Kim, Jaewook Lee, Chulhee YunICML 2025
它引用的顶会 Paper13
- On Gradient Descent Ascent for Nonconvex-Concave Minimax ProblemsTianyi Lin, Chi Jin, Michael I. JordanICML 2020 · 被引用 587 次
- What is Local Optimality in Nonconvex-Nonconcave Minimax Optimization?Chi Jin, Praneeth Netrapalli, Michael I. JordanICML 2020 · 被引用 381 次
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 被引用 172 次
- Large-scale Robust Deep AUC Maximization: A New Surrogate Loss and Empirical Studies on Medical Image ClassificationZhuoning Yuan, Yan Yan, Milan Sonka, Tianbao YangICCV 2021 · 被引用 147 次
- Global Convergence and Variance Reduction for a Class of Nonconvex-Nonconcave Minimax ProblemsJunchi Yang, Negar Kiyavash, Niao HeNeurIPS 2020 · 被引用 136 次
相关 Paper
- Sampling without Replacement Leads to Faster Rates in Finite-Sum Minimax OptimizationAniket Das, Bernhard Schölkopf, Michael MuehlebachNeurIPS 2022 · 被引用 11 次
- SGD with shuffling: optimal rates without component convexity and large epoch requirementsKwangjun Ahn, Chulhee Yun, Suvrit SraNeurIPS 2020 · 被引用 83 次
- Universal Gradient Descent Ascent Method for Nonconvex-Nonconcave Minimax OptimizationTaoli Zheng, Linglingzhi Zhu, Anthony Man-Cho So, Jose H. Blanchet 等NeurIPS 2023 · 被引用 33 次
- PAGE: A Simple and Optimal Probabilistic Gradient Estimator for Nonconvex OptimizationZhize Li, Hongyan Bao, Xiangliang Zhang, Peter RichtárikICML 2021 · 被引用 164 次
- Minibatch vs Local SGD with Shuffling: Tight Convergence Bounds and BeyondChulhee Yun, Shashank Rajput, Suvrit SraICLR 2022 · 被引用 47 次
