2412.01389

Total: 1

#1 Refined Analysis of Federated Averaging's Bias and Federated Richardson-Romberg Extrapolation [PDF] [Copy] [Kimi] [REL]

Authors: Paul Mangold, Alain Durmus, Aymeric Dieuleveut, Sergey Samsonov, Eric Moulines

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 expansion of the bias 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.

Subjects: Machine Learning , Machine Learning , Optimization and Control

Publish: 2024-12-02 11:22:19 UTC