Dynamic Scheduling for Over-the-Air Federated Edge Learning with Energy Constraints

05/31/2021
by   Yuxuan Sun, et al.
22

Machine learning and wireless communication technologies are jointly facilitating an intelligent edge, where federated edge learning (FEEL) is a promising training framework. As wireless devices involved in FEEL are resource limited in terms of communication bandwidth, computing power and battery capacity, it is important to carefully schedule them to optimize the training performance. In this work, we consider an over-the-air FEEL system with analog gradient aggregation, and propose an energy-aware dynamic device scheduling algorithm to optimize the training performance under energy constraints of devices, where both communication energy for gradient aggregation and computation energy for local training are included. The consideration of computation energy makes dynamic scheduling challenging, as devices are scheduled before local training, but the communication energy for over-the-air aggregation depends on the l2-norm of local gradient, which is known after local training. We thus incorporate estimation methods into scheduling to predict the gradient norm. Taking the estimation error into account, we characterize the performance gap between the proposed algorithm and its offline counterpart. Experimental results show that, under a highly unbalanced local data distribution, the proposed algorithm can increase the accuracy by 4.9 CIFAR-10 dataset compared with the myopic benchmark, while satisfying the energy constraints.

READ FULL TEXT
research
02/17/2022

Time-Correlated Sparsification for Efficient Over-the-Air Model Aggregation in Wireless Federated Learning

Federated edge learning (FEEL) is a promising distributed machine learni...
research
05/02/2023

Dynamic Scheduling for Federated Edge Learning with Streaming Data

In this work, we consider a Federated Edge Learning (FEEL) system where ...
research
10/01/2018

Chasing Similarity: Distribution-aware Aggregation Scheduling (Extended Version)

Parallel aggregation is a ubiquitous operation in data analytics that is...
research
02/24/2021

Wirelessly Powered Federated Edge Learning: Optimal Tradeoffs Between Convergence and Power Transfer

Federated edge learning (FEEL) is a widely adopted framework for trainin...
research
10/31/2022

Blind Asynchronous Over-the-Air Federated Edge Learning

Federated Edge Learning (FEEL) is a distributed machine learning techniq...
research
06/02/2021

Communication-Efficient Split Learning Based on Analog Communication and Over the Air Aggregation

Split-learning (SL) has recently gained popularity due to its inherent p...
research
06/17/2021

Optimized Power Control Design for Over-the-Air Federated Edge Learning

This paper investigates the transmission power control in over-the-air f...

Please sign up or login with your details

Forgot password? Click here to reset