Adaptive Transmission Scheduling in Wireless Networks for Asynchronous Federated Learning

03/02/2021
by   Hyun-Suk Lee, et al.
0

In this paper, we study asynchronous federated learning (FL) in a wireless distributed learning network (WDLN). To allow each edge device to use its local data more efficiently via asynchronous FL, transmission scheduling in the WDLN for asynchronous FL should be carefully determined considering system uncertainties, such as time-varying channel and stochastic data arrivals, and the scarce radio resources in the WDLN. To address this, we propose a metric, called an effectivity score, which represents the amount of learning from asynchronous FL. We then formulate an Asynchronous Learning-aware transmission Scheduling (ALS) problem to maximize the effectivity score and develop three ALS algorithms, called ALSA-PI, BALSA, and BALSA-PO, to solve it. If the statistical information about the uncertainties is known, the problem can be optimally and efficiently solved by ALSA-PI. Even if not, it can be still optimally solved by BALSA that learns the uncertainties based on a Bayesian approach using the state information reported from devices. BALSA-PO suboptimally solves the problem, but it addresses a more restrictive WDLN in practice, where the AP can observe a limited state information compared with the information used in BALSA. We show via simulations that the models trained by our ALS algorithms achieve performances close to that by an ideal benchmark and outperform those by other state-of-the-art baseline scheduling algorithms in terms of model accuracy, training loss, learning speed, and robustness of learning. These results demonstrate that the adaptive scheduling strategy in our ALS algorithms is effective to asynchronous FL.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
12/14/2022

Scheduling and Aggregation Design for Asynchronous Federated Learning over Wireless Networks

Federated Learning (FL) is a collaborative machine learning (ML) framewo...
research
07/23/2021

Device Scheduling and Update Aggregation Policies for Asynchronous Federated Learning

Federated Learning (FL) is a newly emerged decentralized machine learnin...
research
05/22/2023

Asynchronous Multi-Model Federated Learning over Wireless Networks: Theory, Modeling, and Optimization

Federated learning (FL) has emerged as a key technique for distributed m...
research
10/28/2022

Efficient and Light-Weight Federated Learning via Asynchronous Distributed Dropout

Asynchronous learning protocols have regained attention lately, especial...
research
01/27/2023

Uplink Scheduling in Federated Learning: an Importance-Aware Approach via Graph Representation Learning

Federated Learning (FL) has emerged as a promising framework for distrib...
research
06/12/2021

Joint Client Scheduling and Resource Allocation under Channel Uncertainty in Federated Learning

The performance of federated learning (FL) over wireless networks depend...
research
07/14/2018

Stochastic Stability in Schelling's Segregation Model with Markovian Asynchronous Update

We investigate the dependence of steady-state properties of Schelling's ...

Please sign up or login with your details

Forgot password? Click here to reset