Passer à la navigation principale Passer à la recherche Passer au contenu principal

Refined Analysis of Constant Step Size Federated Averaging and Federated Richardson-Romberg Extrapolation

  • Ecole polytechnique
  • National Research University
  • Mohamed bin Zayed University of Artificial Intelligence (MBZUAI)

Résultats de recherche: Contribution à un journalArticle de conférenceRevue par des pairs

Résumé

In this paper, we present a novel analysis of FedAvg with constant step size, relying on the Markov property of the underlying process. We demonstrate that the global iterates of the algorithm converge to a stationary distribution and analyze its resulting bias and variance relative to the problem's solution. We provide a first-order bias expansion in both homogeneous and heterogeneous settings. Interestingly, this bias decomposes into two distinct components: one that depends solely on stochastic gradient noise and another on client heterogeneity. Finally, we introduce a new algorithm based on the Richardson-Romberg extrapolation technique to mitigate this bias.

langue originaleAnglais
Pages (de - à)5023-5031
Nombre de pages9
journalProceedings of Machine Learning Research
Volume258
étatPublié - 1 janv. 2025
Evénement28th International Conference on Artificial Intelligence and Statistics, AISTATS 2025 - Mai Khao, Thadlande
Durée: 3 mai 20255 mai 2025

Empreinte digitale

Examiner les sujets de recherche de « Refined Analysis of Constant Step Size Federated Averaging and Federated Richardson-Romberg Extrapolation ». Ensemble, ils forment une empreinte digitale unique.

Contient cette citation