Stability and Generalization of the Decentralized Stochastic Gradient Descent Ascent Algorithm
Miaoxi Zhu, Li Shen, Bo Du, Dacheng Tao
Abstract
The growing size of available data has attracted increasing interest in solving minimax problems in a decentralized manner for various machine learning tasks. Previous theoretical research has primarily focused on the convergence rate and communication complexity of decentralized minimax algorithms, with little attention given to their generalization. In this paper, we investigate the primal-dual generalization bound of the decentralized stochastic gradient descent ascent (D-SGDA) algorithm using the approach of algorithmic stability under both convex-concave and nonconvex-nonconcave settings. Our theory refines the algorithmic stability in a decentralized manner and demonstrates that the decentralized structure does not destroy the stability and generalization of D-SGDA, implying that it can generalize as well as the vanilla SGDA in certain situations. Our results analyze the impact of different topologies on the generalization bound of the D-SGDA algorithm beyond trivial factors such as sample sizes, learning rates, and iterations. We also evaluate the optimization error and balance it with the generalization gap to obtain the optimal population risk of D-SGDA in the convex-concave setting. Additionally, we perform several numerical experiments which validate our theoretical findings.
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 7b682e95-ea16-4af9-acee-3b330c31cfe2Cited by top-tier papers5
- Generalization Analysis of Stochastic Weight Averaging with General SamplingPeng Wang, Li Shen, Zerui Tao, Shuaida He et al.ICML 2024 · 6 citations
- On The Surprising Effectiveness of a Single Global Merging in Decentralized LearningTongtian Zhu, Tianyu Zhang, Mingze Wang, Zhanpeng Zhou et al.ICLR 2026 · 2 citations
- Understanding the Stability-based Generalization of Personalized Federated LearningYingqi Liu, Qinglun Li, Jie Tang, Yifan Shi et al.ICLR 2025
- Scaling Law Analysis in Federated Learning: How to Select the Optimal Model Size?Xuanyu Chen, Nan Yang, Shuai Wang, Dong YuanAAAI 2026
- DICE: Data Influence Cascade in Decentralized LearningTongtian Zhu, Wenhao Li, Can Wang, Fengxiang HeICLR 2025
Builds on20
- On Gradient Descent Ascent for Nonconvex-Concave Minimax ProblemsTianyi Lin, Chi Jin, Michael I. JordanICML 2020 · 587 citations
- Decentralized Deep Learning with Arbitrary Communication CompressionAnastasia Koloskova, Tao Lin, Sebastian U. Stich, Martin JaggiICLR 2020 · 263 citations
- Exponential Graph is Provably Efficient for Decentralized Deep TrainingBicheng Ying, Kun Yuan, Yiming Chen, Hanbin Hu et al.NeurIPS 2021 · 123 citations
- Decentralized Policy Gradient Descent Ascent for Safe Multi-Agent Reinforcement LearningSongtao Lu, Kaiqing Zhang, Tianyi Chen, Tamer Basar et al.AAAI 2021 · 93 citations
- On the Algorithmic Stability of Adversarial TrainingYue Xing, Qifan Song, Guang ChengNeurIPS 2021 · 74 citations
Related papers
- Stability-Based Generalization Analysis of the Asynchronous Decentralized SGDXiaoge Deng, Tao Sun, Shengwei Li, Dongsheng LiAAAI 2023 · 26 citations
- Stability and Generalization of Stochastic Gradient Methods for Minimax ProblemsYunwen Lei, Zhenhuan Yang, Tianbao Yang, Yiming YingICML 2021 · 57 citations
- Improved Stability and Generalization Guarantees of the Decentralized SGD AlgorithmBatiste Le Bars, Aurélien Bellet, Marc Tommasi, Kevin Scaman et al.ICML 2024 · 13 citations
- What is a Good Metric to Study Generalization of Minimax Learners?Asuman E. Ozdaglar, Sarath Pattathil, Jiawei Zhang, Kaiqing ZhangNeurIPS 2022 · 23 citations
- Stability and Generalization of Decentralized Stochastic Gradient DescentTao Sun, Dongsheng Li, Bao WangAAAI 2021 · 44 citations
