Dynamic Clustering and Power Control for Two-Tier Wireless Federated Learning

05/19/2022
by   Wei Guo, et al.
0

Federated learning (FL) has been recognized as a promising distributed learning paradigm to support intelligent applications at the wireless edge, where a global model is trained iteratively through the collaboration of the edge devices without sharing their data. However, due to the relatively large communication cost between the devices and parameter server (PS), direct computing based on the information from the devices may not be resource efficient. This paper studies the joint communication and learning design for the over-the-air computation (AirComp)-based two-tier wireless FL scheme, where the lead devices first collect the local gradients from their nearby subordinate devices, and then send the merged results to the PS for the second round of aggregation. We establish a convergence result for the proposed scheme and derive the upper bound on the optimality gap between the expected and optimal global loss values. Next, based on the device distance and data importance, we propose a hierarchical clustering method to build the two-tier structure. Then, with only the instantaneous channel state information (CSI), we formulate the optimality gap minimization problem and solve it by using an efficient alternating minimization method. Numerical results show that the proposed scheme outperforms the baseline ones.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
05/19/2022

Joint Device Selection and Power Control for Wireless Federated Learning

This paper studies the joint device selection and power control scheme f...
research
08/05/2023

Private Federated Learning with Dynamic Power Control via Non-Coherent Over-the-Air Computation

To further preserve model weight privacy and improve model performance i...
research
12/27/2021

Over-the-Air Multi-Task Federated Learning Over MIMO Interference Channel

With the explosive growth of data and wireless devices, federated learni...
research
06/20/2022

Performance-Oriented Design for Intelligent Reflecting Surface Assisted Federated Learning

To efficiently exploit the massive raw data that is pervading generated ...
research
07/19/2022

Over-the-Air Federated Edge Learning with Hierarchical Clustering

We examine federated learning (FL) with over-the-air (OTA) aggregation, ...
research
03/30/2021

1-Bit Compressive Sensing for Efficient Federated Learning Over the Air

For distributed learning among collaborative users, this paper develops ...
research
05/17/2023

Over-the-Air Federated Learning in MIMO Cloud-RAN Systems

To address the limitations of traditional over-the-air federated learnin...

Please sign up or login with your details

Forgot password? Click here to reset