Stability and Generalization of the Decentralized Stochastic Gradient Descent Ascent Algorithm
Miaoxi Zhu, Li Shen, Bo Du, Dacheng Tao
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了最后一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper5
- Generalization Analysis of Stochastic Weight Averaging with General SamplingPeng Wang, Li Shen, Zerui Tao, Shuaida He 等ICML 2024 · 被引用 6 次
- On The Surprising Effectiveness of a Single Global Merging in Decentralized LearningTongtian Zhu, Tianyu Zhang, Mingze Wang, Zhanpeng Zhou 等ICLR 2026 · 被引用 2 次
- Understanding the Stability-based Generalization of Personalized Federated LearningYingqi Liu, Qinglun Li, Jie Tang, Yifan Shi 等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
它引用的顶会 Paper20
- On Gradient Descent Ascent for Nonconvex-Concave Minimax ProblemsTianyi Lin, Chi Jin, Michael I. JordanICML 2020 · 被引用 587 次
- Decentralized Deep Learning with Arbitrary Communication CompressionAnastasia Koloskova, Tao Lin, Sebastian U. Stich, Martin JaggiICLR 2020 · 被引用 263 次
- Exponential Graph is Provably Efficient for Decentralized Deep TrainingBicheng Ying, Kun Yuan, Yiming Chen, Hanbin Hu 等NeurIPS 2021 · 被引用 123 次
- Decentralized Policy Gradient Descent Ascent for Safe Multi-Agent Reinforcement LearningSongtao Lu, Kaiqing Zhang, Tianyi Chen, Tamer Basar 等AAAI 2021 · 被引用 93 次
- On the Algorithmic Stability of Adversarial TrainingYue Xing, Qifan Song, Guang ChengNeurIPS 2021 · 被引用 74 次
相关 Paper
- Stability-Based Generalization Analysis of the Asynchronous Decentralized SGDXiaoge Deng, Tao Sun, Shengwei Li, Dongsheng LiAAAI 2023 · 被引用 26 次
- Stability and Generalization of Stochastic Gradient Methods for Minimax ProblemsYunwen Lei, Zhenhuan Yang, Tianbao Yang, Yiming YingICML 2021 · 被引用 57 次
- Improved Stability and Generalization Guarantees of the Decentralized SGD AlgorithmBatiste Le Bars, Aurélien Bellet, Marc Tommasi, Kevin Scaman 等ICML 2024 · 被引用 13 次
- What is a Good Metric to Study Generalization of Minimax Learners?Asuman E. Ozdaglar, Sarath Pattathil, Jiawei Zhang, Kaiqing ZhangNeurIPS 2022 · 被引用 23 次
- Stability and Generalization of Decentralized Stochastic Gradient DescentTao Sun, Dongsheng Li, Bao WangAAAI 2021 · 被引用 44 次
