Resource Allocation for Hierarchical Federated Learning in Edge Networks

Tangju Yuan,Lele Wang,Yu Gu,Wenbin Gong,Yutao Jiang,Liwan Chen
DOI: https://doi.org/10.1109/eei63073.2024.10696072
2024-01-01
Abstract:Numerous internet of things(IoT) devices generate massive amounts of data, and transmitting the data to cloud servers for model training incurs significant communication overhead and training delays. Federated learning protects user privacy and reduces communication overheads through a unique approach to model training. Furthermore, the diversity of IoT devices results in data and resource heterogeneity, severely impeding the efficiency of federated learning training. To address these issues, this article proposes a device selection strategy within a hierarchical federated learning framework. This strategy improves training efficiency by comprehensively considering device data along with computational and communication resources. Simulation experiments demonstrate that our proposed framework and algorithm achieve higher learning accuracy and faster execution times compared to existing solutions.
What problem does this paper attempt to address?