Debiasing Federated Learning with Correlated Client Participation
Zhenyu Sun, Ziyang Zhang, Zheng Xu, Gauri Joshi, Pranay Sharma, Ermin Wei
Abstract
In cross-device federated learning (FL) with millions of mobile clients, only a small subset of clients participate in training in every communication round, and Federated Averaging (FedAvg) is the most popular algorithm in practice. Existing analyses of FedAvg usually assume the participating clients are independently sampled in each round from a uniform distribution, which does not reflect real-world scenarios. This paper introduces a theoretical framework that models client participation in FL as a Markov chain to study optimization convergence when clients have nonuniform and correlated participation across rounds. We apply this framework to analyze a more general and practical pattern: every client must wait a minimum number of R rounds (minimum separation) before re-participating. We theoretically prove and empirically observe that increasing minimum separation reduces the bias induced by intrinsic non-uniformity of client availability in cross-device FL systems. Furthermore, we develop an effective debiasing algorithm for FedAvg that provably converges to the unbiased optimal solution under arbitrary minimum separation and unknown client availability distribution.
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.
Cited by top-tier papers4
- Learning from A Single Markovian Trajectory: Optimality and Variance ReductionZhenyu Sun, Ermin WeiNeurIPS 2025 · 2 citations
- Streaming Federated Learning with Markovian DataKhiem Huynh, Malcolm Egan, Giovanni Neglia, Jean-Marie GorceNeurIPS 2025 · 2 citations
- Improved Lower Bounds for First-order Stochastic Non-convex Optimization under Markov SamplingZhenyu Sun, Ermin WeiICML 2025
- A Unified Analysis of Stochastic Gradient Descent with Arbitrary Data Permutations and BeyondYipeng Li, Xinchen Lyu, Zhenyu LiuNeurIPS 2025
Builds on13
- On the Convergence of FedAvg on Non-IID DataXiang Li, Kaixuan Huang, Wenhao Yang, Shusen Wang et al.ICLR 2020 · 2,930 citations
- Tackling the Objective Inconsistency Problem in Heterogeneous Federated OptimizationJianyu Wang, Qinghua Liu, Hao Liang, Gauri Joshi et al.NeurIPS 2020 · 2,231 citations
- Clustered Sampling: Low-Variance and Improved Representativity for Clients Selection in Federated LearningYann Fraboni, Richard Vidal, Laetitia Kameni, Marco LorenziICML 2021 · 249 citations
- Practical and Private (Deep) Learning Without Sampling or ShufflingPeter Kairouz, Brendan McMahan, Shuang Song, Om Thakkar et al.ICML 2021 · 239 citations
- Minibatch vs Local SGD for Heterogeneous Distributed LearningBlake E. Woodworth, Kumar Kshitij Patel, Nati SrebroNeurIPS 2020 · 231 citations
Related papers
- On the Convergence of Federated Averaging with Cyclic Client ParticipationYae Jee Cho, Pranay Sharma, Gauri Joshi, Zheng Xu et al.ICML 2023 · 47 citations
- Federated Learning under Heterogeneous and Correlated Client AvailabilityAngelo Rodio, Francescomaria Faticanti, Othmane Marfoq, Giovanni Neglia et al.INFOCOM 2023 · 27 citations
- A Unified Analysis of Federated Learning with Arbitrary Client ParticipationShiqiang Wang, Mingyue JiNeurIPS 2022 · 85 citations
- Optimizing Federated Learning on Non-IID Data with Reinforcement LearningHao Wang, Zakhary Kaplan, Di Niu, Baochun LiINFOCOM 2020 · 1,002 citations
- Fast Federated Learning in the Presence of Arbitrary Device UnavailabilityXinran Gu, Kaixuan Huang, Jingzhao Zhang, Longbo HuangNeurIPS 2021 · 142 citations
