FedNMUT -- Federated Noisy Model Update Tracking Convergence Analysis

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chellapandi, Vishnu Pandi, Upadhyay, Antesh, Hashemi, Abolfazl, Żak, Stanislaw H.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909149108895744
author Chellapandi, Vishnu Pandi
Upadhyay, Antesh
Hashemi, Abolfazl
Żak, Stanislaw H.
author_facet Chellapandi, Vishnu Pandi
Upadhyay, Antesh
Hashemi, Abolfazl
Żak, Stanislaw H.
contents A novel Decentralized Noisy Model Update Tracking Federated Learning algorithm (FedNMUT) is proposed that is tailored to function efficiently in the presence of noisy communication channels that reflect imperfect information exchange. This algorithm uses gradient tracking to minimize the impact of data heterogeneity while minimizing communication overhead. The proposed algorithm incorporates noise into its parameters to mimic the conditions of noisy communication channels, thereby enabling consensus among clients through a communication graph topology in such challenging environments. FedNMUT prioritizes parameter sharing and noise incorporation to increase the resilience of decentralized learning systems against noisy communications. Theoretical results for the smooth non-convex objective function are provided by us, and it is shown that the $ε-$stationary solution is achieved by our algorithm at the rate of $\mathcal{O}\left(\frac{1}{\sqrt{T}}\right)$, where $T$ is the total number of communication rounds. Additionally, via empirical validation, we demonstrated that the performance of FedNMUT is superior to the existing state-of-the-art methods and conventional parameter-mixing approaches in dealing with imperfect information sharing. This proves the capability of the proposed algorithm to counteract the negative effects of communication noise in a decentralized learning framework.
format Preprint
id arxiv_https___arxiv_org_abs_2403_13247
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FedNMUT -- Federated Noisy Model Update Tracking Convergence Analysis
Chellapandi, Vishnu Pandi
Upadhyay, Antesh
Hashemi, Abolfazl
Żak, Stanislaw H.
Machine Learning
Distributed, Parallel, and Cluster Computing
A novel Decentralized Noisy Model Update Tracking Federated Learning algorithm (FedNMUT) is proposed that is tailored to function efficiently in the presence of noisy communication channels that reflect imperfect information exchange. This algorithm uses gradient tracking to minimize the impact of data heterogeneity while minimizing communication overhead. The proposed algorithm incorporates noise into its parameters to mimic the conditions of noisy communication channels, thereby enabling consensus among clients through a communication graph topology in such challenging environments. FedNMUT prioritizes parameter sharing and noise incorporation to increase the resilience of decentralized learning systems against noisy communications. Theoretical results for the smooth non-convex objective function are provided by us, and it is shown that the $ε-$stationary solution is achieved by our algorithm at the rate of $\mathcal{O}\left(\frac{1}{\sqrt{T}}\right)$, where $T$ is the total number of communication rounds. Additionally, via empirical validation, we demonstrated that the performance of FedNMUT is superior to the existing state-of-the-art methods and conventional parameter-mixing approaches in dealing with imperfect information sharing. This proves the capability of the proposed algorithm to counteract the negative effects of communication noise in a decentralized learning framework.
title FedNMUT -- Federated Noisy Model Update Tracking Convergence Analysis
topic Machine Learning
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2403.13247