Compressed and distributed least-squares regression: convergence rates with applications to Federated Learning - Archive ouverte HAL Accéder directement au contenu
Pré-Publication, Document De Travail Année : 2024

Compressed and distributed least-squares regression: convergence rates with applications to Federated Learning

Résumé

In this paper, we investigate the impact of compression on stochastic gradient algorithms for machine learning, a technique widely used in distributed and federated learning. We underline differences in terms of convergence rates between several unbiased compression operators, that all satisfy the same condition on their variance, thus going beyond the classical worst-case analysis. To do so, we focus on the case of least-squares regression (LSR) and analyze a general stochastic approximation algorithm for minimizing quadratic functions relying on a random field. We consider weak assumptions on the random field, tailored to the analysis (specifically, expected H\"older regularity), and on the noise covariance, enabling the analysis of various randomizing mechanisms, including compression. We then extend our results to the case of federated learning. More formally, we highlight the impact on the convergence of the covariance of the additive noise induced by the algorithm. We demonstrate despite the non-regularity of the stochastic field, that the limit variance term scales with (where is the Hessian of the optimization problem and the number of iterations) generalizing the rate for the vanilla LSR case where it is (Bach and Moulines, 2013). Then, we analyze the dependency of on the compression strategy and ultimately its impact on convergence, first in the centralized case, then in two heterogeneous FL frameworks.
Fichier non déposé

Dates et versions

hal-04504217 , version 1 (14-03-2024)

Identifiants

  • HAL Id : hal-04504217 , version 1

Citer

Aymeric Dieuleveut, Constantin Philippenko. Compressed and distributed least-squares regression: convergence rates with applications to Federated Learning. 2024. ⟨hal-04504217⟩
4 Consultations
0 Téléchargements

Partager

Gmail Facebook X LinkedIn More