Decentralized Sporadic Federated Learning: A Unified Algorithmic Framework with Convergence Guarantees

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zehtabi, Shahryar, Han, Dong-Jun, Parasnis, Rohit, Hosseinalipour, Seyyedali, Brinton, Christopher G.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913720739823616
author Zehtabi, Shahryar
Han, Dong-Jun
Parasnis, Rohit
Hosseinalipour, Seyyedali
Brinton, Christopher G.
author_facet Zehtabi, Shahryar
Han, Dong-Jun
Parasnis, Rohit
Hosseinalipour, Seyyedali
Brinton, Christopher G.
contents Decentralized federated learning (DFL) captures FL settings where both (i) model updates and (ii) model aggregations are exclusively carried out by the clients without a central server. Existing DFL works have mostly focused on settings where clients conduct a fixed number of local updates between local model exchanges, overlooking heterogeneity and dynamics in communication and computation capabilities. In this work, we propose Decentralized Sporadic Federated Learning ($\texttt{DSpodFL}$), a DFL methodology built on a generalized notion of $\textit{sporadicity}$ in both local gradient and aggregation processes. $\texttt{DSpodFL}$ subsumes many existing decentralized optimization methods under a unified algorithmic framework by modeling the per-iteration (i) occurrence of gradient descent at each client and (ii) exchange of models between client pairs as arbitrary indicator random variables, thus capturing $\textit{heterogeneous and time-varying}$ computation/communication scenarios. We analytically characterize the convergence behavior of $\texttt{DSpodFL}$ for both convex and non-convex models and for both constant and diminishing learning rates, under mild assumptions on the communication graph connectivity, data heterogeneity across clients, and gradient noises. We show how our bounds recover existing results from decentralized gradient descent as special cases. Experiments demonstrate that $\texttt{DSpodFL}$ consistently achieves improved training speeds compared with baselines under various system settings.
format Preprint
id arxiv_https___arxiv_org_abs_2402_03448
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Decentralized Sporadic Federated Learning: A Unified Algorithmic Framework with Convergence Guarantees
Zehtabi, Shahryar
Han, Dong-Jun
Parasnis, Rohit
Hosseinalipour, Seyyedali
Brinton, Christopher G.
Machine Learning
Distributed, Parallel, and Cluster Computing
Decentralized federated learning (DFL) captures FL settings where both (i) model updates and (ii) model aggregations are exclusively carried out by the clients without a central server. Existing DFL works have mostly focused on settings where clients conduct a fixed number of local updates between local model exchanges, overlooking heterogeneity and dynamics in communication and computation capabilities. In this work, we propose Decentralized Sporadic Federated Learning ($\texttt{DSpodFL}$), a DFL methodology built on a generalized notion of $\textit{sporadicity}$ in both local gradient and aggregation processes. $\texttt{DSpodFL}$ subsumes many existing decentralized optimization methods under a unified algorithmic framework by modeling the per-iteration (i) occurrence of gradient descent at each client and (ii) exchange of models between client pairs as arbitrary indicator random variables, thus capturing $\textit{heterogeneous and time-varying}$ computation/communication scenarios. We analytically characterize the convergence behavior of $\texttt{DSpodFL}$ for both convex and non-convex models and for both constant and diminishing learning rates, under mild assumptions on the communication graph connectivity, data heterogeneity across clients, and gradient noises. We show how our bounds recover existing results from decentralized gradient descent as special cases. Experiments demonstrate that $\texttt{DSpodFL}$ consistently achieves improved training speeds compared with baselines under various system settings.
title Decentralized Sporadic Federated Learning: A Unified Algorithmic Framework with Convergence Guarantees
topic Machine Learning
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2402.03448