Sampling without Replacement Leads to Faster Rates in Finite-Sum Minimax Optimization
Aniket Das, Bernhard Schölkopf, Michael Muehlebach
摘要
We analyze the convergence rates of stochastic gradient algorithms for smooth finite-sum minimax optimization and show that, for many such algorithms, sampling the data points without replacement leads to faster convergence compared to sampling with replacement. For the smooth and strongly convex-strongly concave setting, we consider gradient descent ascent and the proximal point method, and present a unified analysis of two popular without-replacement sampling strategies, namely Random Reshuffling (RR), which shuffles the data every epoch, and Single Shuffling or Shuffle Once (SO), which shuffles only at the beginning. We obtain tight convergence rates for RR and SO and demonstrate that these strategies lead to faster convergence than uniform sampling. Moving beyond convexity, we obtain similar results for smooth nonconvex-nonconcave objectives satisfying a two-sided Polyak-ojasiewicz inequality. Finally, we demonstrate that our techniques are general enough to analyze the effect of data-ordering attacks, where an adversary manipulates the order in which data points are supplied to the optimizer. Our analysis also recovers tight rates for the incremental gradient method, where the data points are not shuffled at all.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper10
- On the Last-Iterate Convergence of Shuffling Gradient MethodsZijian Liu, Zhengyuan ZhouICML 2024 · 被引用 11 次
- Provably Fast Finite Particle Variants of SVGD via Virtual Particle Stochastic ApproximationAniket Das, Dheeraj NagarajNeurIPS 2023 · 被引用 10 次
- Generalization Analysis of Stochastic Weight Averaging with General SamplingPeng Wang, Li Shen, Zerui Tao, Shuaida He 等ICML 2024 · 被引用 6 次
- Stochastic Extragradient with Flip-Flop Shuffling & Anchoring: Provable ImprovementsJiseok Chae, Chulhee Yun, Donghwan KimNeurIPS 2024 · 被引用 2 次
- Provably Faster Algorithms for Bilevel Optimization via Without-Replacement SamplingJunyi Li, Heng HuangNeurIPS 2024 · 被引用 1 次
它引用的顶会 Paper17
- On Gradient Descent Ascent for Nonconvex-Concave Minimax ProblemsTianyi Lin, Chi Jin, Michael I. JordanICML 2020 · 被引用 587 次
- Random Reshuffling: Simple Analysis with Vast ImprovementsKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikNeurIPS 2020 · 被引用 172 次
- A Game Theoretic Framework for Model Based Reinforcement LearningAravind Rajeswaran, Igor Mordatch, Vikash KumarICML 2020 · 被引用 137 次
- Global Convergence and Variance Reduction for a Class of Nonconvex-Nonconcave Minimax ProblemsJunchi Yang, Negar Kiyavash, Niao HeNeurIPS 2020 · 被引用 136 次
- Manipulating SGD with Data Ordering AttacksIlia Shumailov, Zakhar Shumaylov, Dmitry Kazhdan, Yiren Zhao 等NeurIPS 2021 · 被引用 125 次
相关 Paper
- SGDA with shuffling: faster convergence for nonconvex-PŁ minimax optimizationHanseul Cho, Chulhee YunICLR 2023
- SGD with shuffling: optimal rates without component convexity and large epoch requirementsKwangjun Ahn, Chulhee Yun, Suvrit SraNeurIPS 2020 · 被引用 83 次
- An Improved Analysis and Rates for Variance Reduction under Without-replacement Sampling OrdersXinmeng Huang, Kun Yuan, Xianghui Mao, Wotao YinNeurIPS 2021 · 被引用 1 次
- Proximal and Federated Random ReshufflingKonstantin Mishchenko, Ahmed Khaled, Peter RichtárikICML 2022 · 被引用 39 次
- A General Analysis of Example-Selection for Stochastic Gradient DescentYucheng Lu, Si Yi Meng, Christopher De SaICLR 2022 · 被引用 23 次
