Asynchronous Multi-Server Federated Learning for Geo-Distributed Clients

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zuo, Yuncong, Cox, Bart, Chen, Lydia Y., Decouchant, Jérémie
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917699294068736
author Zuo, Yuncong
Cox, Bart
Chen, Lydia Y.
Decouchant, Jérémie
author_facet Zuo, Yuncong
Cox, Bart
Chen, Lydia Y.
Decouchant, Jérémie
contents Federated learning (FL) systems enable multiple clients to train a machine learning model iteratively through synchronously exchanging the intermediate model weights with a single server. The scalability of such FL systems can be limited by two factors: server idle time due to synchronous communication and the risk of a single server becoming the bottleneck. In this paper, we propose a new FL architecture, to our knowledge, the first multi-server FL system that is entirely asynchronous, and therefore addresses these two limitations simultaneously. Our solution keeps both servers and clients continuously active. As in previous multi-server methods, clients interact solely with their nearest server, ensuring efficient update integration into the model. Differently, however, servers also periodically update each other asynchronously, and never postpone interactions with clients. We compare our solution to three representative baselines - FedAvg, FedAsync and HierFAVG - on the MNIST and CIFAR-10 image classification datasets and on the WikiText-2 language modeling dataset. Our solution converges to similar or higher accuracy levels than previous baselines and requires 61% less time to do so in geo-distributed settings.
format Preprint
id arxiv_https___arxiv_org_abs_2406_01439
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Asynchronous Multi-Server Federated Learning for Geo-Distributed Clients
Zuo, Yuncong
Cox, Bart
Chen, Lydia Y.
Decouchant, Jérémie
Machine Learning
Distributed, Parallel, and Cluster Computing
I.2.11
Federated learning (FL) systems enable multiple clients to train a machine learning model iteratively through synchronously exchanging the intermediate model weights with a single server. The scalability of such FL systems can be limited by two factors: server idle time due to synchronous communication and the risk of a single server becoming the bottleneck. In this paper, we propose a new FL architecture, to our knowledge, the first multi-server FL system that is entirely asynchronous, and therefore addresses these two limitations simultaneously. Our solution keeps both servers and clients continuously active. As in previous multi-server methods, clients interact solely with their nearest server, ensuring efficient update integration into the model. Differently, however, servers also periodically update each other asynchronously, and never postpone interactions with clients. We compare our solution to three representative baselines - FedAvg, FedAsync and HierFAVG - on the MNIST and CIFAR-10 image classification datasets and on the WikiText-2 language modeling dataset. Our solution converges to similar or higher accuracy levels than previous baselines and requires 61% less time to do so in geo-distributed settings.
title Asynchronous Multi-Server Federated Learning for Geo-Distributed Clients
topic Machine Learning
Distributed, Parallel, and Cluster Computing
I.2.11
url https://arxiv.org/abs/2406.01439