Asyn2F: An Asynchronous Federated Learning Framework with Bidirectional Model Aggregation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Cao, Tien-Dung, Vuong, Nguyen T., Le, Thai Q., Dao, Hoang V. N., Truong-Huu, Tram
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914094288732160
author Cao, Tien-Dung
Vuong, Nguyen T.
Le, Thai Q.
Dao, Hoang V. N.
Truong-Huu, Tram
author_facet Cao, Tien-Dung
Vuong, Nguyen T.
Le, Thai Q.
Dao, Hoang V. N.
Truong-Huu, Tram
contents In federated learning, the models can be trained synchronously or asynchronously. Many research works have focused on developing an aggregation method for the server to aggregate multiple local models into the global model with improved performance. They ignore the heterogeneity of the training workers, which causes the delay in the training of the local models, leading to the obsolete information issue. In this paper, we design and develop Asyn2F, an Asynchronous Federated learning Framework with bidirectional model aggregation. By bidirectional model aggregation, Asyn2F, on one hand, allows the server to asynchronously aggregate multiple local models and results in a new global model. On the other hand, it allows the training workers to aggregate the new version of the global model into the local model, which is being trained even in the middle of a training epoch. We develop Asyn2F considering the practical implementation requirements such as using cloud services for model storage and message queuing protocols for communications. Extensive experiments with different datasets show that the models trained by Asyn2F achieve higher performance compared to the state-of-the-art techniques. The experiments also demonstrate the effectiveness, practicality, and scalability of Asyn2F, making it ready for deployment in real scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2403_01417
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Asyn2F: An Asynchronous Federated Learning Framework with Bidirectional Model Aggregation
Cao, Tien-Dung
Vuong, Nguyen T.
Le, Thai Q.
Dao, Hoang V. N.
Truong-Huu, Tram
Machine Learning
Distributed, Parallel, and Cluster Computing
In federated learning, the models can be trained synchronously or asynchronously. Many research works have focused on developing an aggregation method for the server to aggregate multiple local models into the global model with improved performance. They ignore the heterogeneity of the training workers, which causes the delay in the training of the local models, leading to the obsolete information issue. In this paper, we design and develop Asyn2F, an Asynchronous Federated learning Framework with bidirectional model aggregation. By bidirectional model aggregation, Asyn2F, on one hand, allows the server to asynchronously aggregate multiple local models and results in a new global model. On the other hand, it allows the training workers to aggregate the new version of the global model into the local model, which is being trained even in the middle of a training epoch. We develop Asyn2F considering the practical implementation requirements such as using cloud services for model storage and message queuing protocols for communications. Extensive experiments with different datasets show that the models trained by Asyn2F achieve higher performance compared to the state-of-the-art techniques. The experiments also demonstrate the effectiveness, practicality, and scalability of Asyn2F, making it ready for deployment in real scenarios.
title Asyn2F: An Asynchronous Federated Learning Framework with Bidirectional Model Aggregation
topic Machine Learning
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2403.01417