Proceedings

Stability and Generalization for Decentralized Markov SGD

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. Main Track. Pages 4922-4930.

https://doi.org/10.24963/ijcai.2026/548

Abstract

Stochastic gradient methods are central to large-scale learning, yet their generalization theory typically relies on independent sampling assumptions. In many practical applications, data are generated by Markov chains and learning is performed in a decentralized manner, which introduces significant analytical challenges. In this work, we investigate the stability and generalization of decentralized stochastic gradient descent (SGD) and stochastic gradient descent ascent (SGDA) under Markov chain sampling. Leveraging a stability-based framework, we characterize how Markovian dependence and decentralized communication jointly influence generalization behavior. Our analysis captures the effects of network topology, Markov chain mixing properties, and primal–dual dynamics. We establish non-asymptotic generalization bounds for both algorithms, extending existing results on Markov stochastic gradient methods to decentralized and minimax settings.

  1. [Bars et al., 2024] Batiste Le Bars, Aurélien Bellet, and Marc Tommasi. Improved stability and generalization analysis of the decentralized sgd algorithm. In Interna- tional Conference on Machine Learning (ICML), 2024.
  2. [Bassily et al., 2020] Raef Bassily, Vitaly Feldman, Cristóbal Guzmán, and Kunal Talwar. Stability of stochastic gradient descent on nonsmooth convex losses. In Advances in Neural Information Processing Systems (NeurIPS), pages 4381-4391, 2020.
  3. [Chen et al., 2014] Jianshu Chen, Zaid J Towfic, and Ali H Sayed. Dictionary learning over distributed models. IEEE Transactions on Signal Processing, 63(4):1001- 1016, 2014.
  4. [Deng et al., 2023] Xiaoge Deng, Tao Sun, Shengwei Li, and Dongsheng Li. Stability-based generalization analysis of the asynchronous decentralized sgd. In AAAI Conference on Artificial Intelligence, pages 7340-7348, 2023.
  5. [Doan et al., 2020] Thinh T Doan, Lam M Nguyen, Nhan H Pham, and Justin Romberg. Convergence rates of accel- erated markov gradient descent with applications in rein- forcement learning, 2020.
  6. [Doan, 2022] Thinh T Doan. Finite-time analysis of markov gradient descent. IEEE Transactions on Automatic Con- trol, 68(4):2140-2153, 2022.
  7. [Duchi et al., 2011] John C Duchi, Alekh Agarwal, and Mar- tin J Wainwright. Dual averaging for distributed optimiza- tion: Convergence analysis and network scaling. IEEE Transactions on Automatic control, 57(3):592-606, 2011.
  8. [Duchi et al., 2012] John C Duchi, Alekh Agarwal, Mikael Johansson, and Michael I Jordan. Ergodic mirror descent. SIAM Journal on Optimization, 22(4):1549-1578, 2012.
  9. [Farnia and Ozdaglar, 2021] Farzan Farnia and Asuman Ozdaglar. Train simultaneously, generalize better: Stabil- ity of gradient-based minimax learners. In International Conference on Machine Learning (ICML), pages 3174- 3185. PMLR, 2021.
  10. [Goodfellow et al., 2014] Ian J Goodfellow, Jean Pouget- Abadie, Mehdi Mirza, Bing Xu, David Warde-Farley, Sherjil Ozair, Aaron Courville, and Yoshua Bengio. Gen- erative adversarial nets. In Advances in Neural Informa- tion Processing Systems, volume 27, 2014.
  11. [Hardt et al., 2016] Moritz Hardt, Ben Recht, and Yoram Singer. Train faster, generalize better: Stability of stochas- tic gradient descent. In International Conference on Ma- chine Learning (ICML), pages 1225-1234, 2016.
  12. [Hu et al., 2025] Xiaolin Hu, Zixuan Gong, Gengze Xu, Wei Liu, Jian Luan, Bin Wang, and Yong Liu. Stability and generalization of zeroth-order decentralized stochas- tic gradient descent with changing topology. In Proceed- ings of the AAAI Conference on Artificial Intelligence, vol- ume 39, pages 17342-17350, 2025.
  13. [Koloskova et al., 2020] Anastasia Koloskova, Nicolas Loizou, Sadra Boreiri, Martin Jaggi, and Sebastian Stich. A unified theory of decentralized sgd with changing topology and local updates. In International Conference on Machine Learning (ICML), pages 5381-5393.
  14. PMLR, 2020.
  15. [Le Bars et al., 2023] Batiste Le Bars, Aurélien Bellet, Marc Tommasi, Erick Lavoie, and Anne-Marie Kermarrec. Re- fined convergence and topology learning for decentralized sgd with heterogeneous data. In International Conference on Artificial Intelligence and Statistics, pages 1672-1702. PMLR, 2023.
  16. [Lei and Ying, 2020] Yunwen Lei and Yiming Ying. Fine- grained analysis of stability and generalization for stochas- tic gradient descent. In International Conference on Ma- chine Learning (ICML), pages 5809-5819, 2020.
  17. [Lei et al., 2020] Yunwen Lei, Antoine Ledent, and Marius Kloft. Sharper generalization bounds for pairwise learn- ing. In Advances in Neural Information Processing Sys- tems (NeurIPS), pages 21236-21246, 2020.
  18. [Lei et al., 2021a] Yunwen Lei, Mingrui Liu, and Yiming Ying. Generalization guarantee of sgd for pairwise learn- ing. In Advances in Neural Information Processing Sys- tems (NeurIPS), pages 21216-21228, 2021.
  19. [Lei et al., 2021b] Yunwen Lei, Zhenhuan Yang, Tianbao Yang, and Yiming Ying. Stability and generalization of stochastic gradient methods for minimax problems. In International Conference on Machine Learning (ICML), pages 6175-6186. PMLR, 2021.
  20. [Li et al., 2010] Lihong Li, Wei Chu, John Langford, and Robert E Schapire. A contextual-bandit approach to per- sonalized news article recommendation. In International Conference on World Wide Web (WWW), pages 661-670, 2010.
  21. [Li et al., 2014] Mu Li, Tong Zhang, Yuqiang Chen, and Alexander J Smola. Efficient mini-batch training for stochastic optimization. In International Conference on Knowledge Discovery and Data Mining, pages 661-670, 2014.
  22. [Lian et al., 2017] Xiangru Lian, Ce Zhang, Huan Zhang, Cho-Jui Hsieh, Wei Zhang, and Ji Liu. Can decentral- ized algorithms outperform centralized algorithms? a case study for decentralized parallel stochastic gradient de- scent. In Advances in Neural Information Processing Sys- tems (NeurIPS), 2017.
  23. [Lian et al., 2018] Xiangru Lian, Wei Zhang, Ce Zhang, and Ji Liu. Asynchronous decentralized parallel stochastic gra- dient descent. In International Conference on Machine Learning (ICML), pages 3043-3052, 2018.
  24. [Lin et al., 2018] Tao Lin, Sebastian U Stich, Kumar Kshitij Patel, and Martin Jaggi. Don't use large mini-batches, use local sgd. arXiv preprint arXiv:1808.07217, 2018.
  25. [Lopes and Sayed, 2007] Cassio G Lopes and Ali H Sayed. Incremental adaptive strategies over distributed networks. IEEE Transactions on Signal Processing, 55(8):4064- 4077, 2007.
  26. [Madry et al., 2017] Aleksander Madry, Aleksandar Makelov, Ludwig Schmidt, Dimitris Tsipras, and Adrian Vladu. Towards deep learning models resistant to adversarial attacks, 2017.
  27. [Mao et al., 2020] Xianghui Mao, Kun Yuan, Yubin Hu, Yuantao Gu, Ali H Sayed, and Wotao Yin. Walkman: A communication-efficient random-walk algorithm for de- centralized optimization. IEEE Transactions on Signal Processing, 68:2513-2528, 2020.
  28. [Nedic and Ozdaglar, 2009] Angelia Nedic and Asuman Ozdaglar. Distributed subgradient methods for multi-agent optimization. IEEE Transactions on Automatic Control, 54:48-61, 2009.
  29. [Ozdaglar et al., 2022] Asuman Ozdaglar, Sarath Pattathil, Jiawei Zhang, and Kaiqing Zhang. What is a good metric to study generalization of minimax learners? In Advances in Neural Information Processing Systems (NeuIPS), vol- ume 35, pages 38190-38203, 2022.
  30. [Ram et al., 2009] S Sundhar Ram, A Nedić, and Venu- gopal V Veeravalli. Incremental stochastic subgradient al- gorithms for convex optimization. SIAM Journal on Opti- mization, 20(2):691-717, 2009.
  31. [Richards and Rebeschini, 2020] Dominic Richards and Patrick Rebeschini. Graph-dependent implicit regular- isation for distributed stochastic subgradient descent. Journal of Machine Learning Research, 21:1-44, 2020.
  32. [Shah and Avrachenkov, 2018] Suhail M Shah and Kon- stantin E Avrachenkov. Linearly convergent asynchronous distributed admm via markov sampling, 2018.
  33. [Shi et al., 2015] Wei Shi, Qing Ling, Gang Wu, and Wotao Yin. Extra: An exact first-order algorithm for decentral- ized consensus optimization. SIAM Journal on Optimiza- tion, 25(2):944-966, 2015.
  34. [Sun et al., 2018] Tao Sun, Yuejiao Sun, and Wotao Yin. On markov chain gradient descent. In Advances in Neural In- formation Processing Systems (NeurIPS), 2018.
  35. [Sun et al., 2021] Tao Sun, Dongsheng Li, and Bao Wang. Stability and generalization of decentralized stochastic gradient descent. In AAAI Conference on Artificial Intelli- gence, pages 9756-9764, 2021.
  36. [Sun et al., 2022] Tao Sun, Dongsheng Li, and Bao Wang. Adaptive random walk gradient descent for decentralized optimization. In International Conference on Machine Learning, pages 20790-20809. PMLR, 2022.
  37. [Sun et al., 2023] Tao Sun, Dongsheng Li, and Bao Wang. On the decentralized stochastic gradient descent with markov chain sampling. IEEE Transactions on Signal Pro- cessing, pages 1-14, 2023.
  38. [Sundhar Ram et al., 2010] S Sundhar Ram, Angelia Nedić, and Venugopal V Veeravalli. Distributed stochastic subgradient projection algorithms for convex optimiza- tion. Journal of Optimization Theory and Applications, 147:516-545, 2010.
  39. [Tadić and Doucet, 2011] Vladislav B Tadić and Arnaud Doucet. Asymptotic bias of stochastic gradient search. In IEEE Conference on Decision and Control and European Control Conference, pages 722-727. IEEE, 2011.
  40. [Wai et al., 2018] Hoi-To Wai, Zhuoran Yang, Zhaoran Wang, and Mingyi Hong. Multi-agent reinforcement learning via double averaging primal-dual optimization. In Advances in Neural Information Processing Systems (NeurIPS), volume 31, 2018.
  41. [Wang and Chen, 2024] Jiahuan Wang and Hong Chen. To- wards stability and generalization bounds in decentralized minibatch stochastic gradient descent. In Proceedings of the AAAI Conference on Artificial Intelligence, volume 38, pages 15511-15519, 2024.
  42. [Wang et al., 2022] Puyu Wang, Yuewen Lei, Yiming Ying, and Ding-Xuan Zhou. Stability and generalization for markov chain stochastic gradient methods. In Advances in Neural Information Processing Systems (NeurIPS), pages 37735-37748, 2022.
  43. [Yang et al., 2021] Zhenhuan Yang, Yunwen Lei, Puyu Wang, Tianbao Yang, and Yiming Ying. Simple stochastic and online gradient descent algorithms for pairwise learn- ing. In Advances in Neural Information Processing Sys- tems (NeurIPS), volume 34, pages 20160-20171, 2021.
  44. [Ye et al., 2025] Haoxiang Ye, Tao Sun, and Qing Ling. Generalization error analysis for attack-free and byzantine-resilient decentralized learning with data heterogeneity, 2025.
  45. [Yuan et al., 2016] Kun Yuan, Qing Ling, and Wotao Yin. On the convergence of decentralized gradient descent. SIAM Journal on Optimization, 26(3):1835-1854, 2016.
  46. [Yuan et al., 2023] Kun Yuan, Sulaiman A Alghunaim, and Xinmeng Huang. Removing data heterogeneity influence enhances network topology dependence of decentralized sgd. Journal of Machine Learning Research, 24(280):1- 53, 2023.
  47. [Zeng and Lei, 2025] Shuang Zeng and Yunwen Lei. Sta- bility and generalization analysis of decentralized sgd: sharper bounds beyond lipschitzness and smoothness. In International Conference on Machine Learning (ICML). PMLR, 2025.
  48. [Zhu et al., 2022] Tongtian Zhu, Fengxiang He, Lan Zhang, Zhengyang Niu, Mingli Song, and Dacheng Tao. Topology-aware generalization of decentralized sgd. In International Conference on Machine Learning (ICML), pages 27479-27503, 2022.
  49. [Zhu et al., 2023] Miaoxi Zhu, Li Shen, Bo Du, and Dacheng Tao. Stability and generalization of the de- centralized stochastic gradient descent ascent algorithm. In Advances in Neural Information Processing Systems (NeuIPS), volume 36, 2023.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards