Title
Dynamic Client Association for Energy-Aware Hierarchical Federated Learning
Abstract
Federated learning (FL) has become a promising solution to train a shared model without exchanging local training samples. However, in the traditional cloud-based FL framework, clients suffer from limited energy budget and generate excessive communication overhead on the backbone network. These drawbacks motivate us to propose an energy-aware hierarchical federated learning framework in which the edge servers assist the cloud server to migrate the local models from the clients. Then a joint local computing power control and client association problem is formulated in order to minimize the training loss and the training latency simultaneously under the long-term energy constraints. To solve the problem, we recast it based on the general Lyapunov optimization framework with the instantaneous energy budget. We then propose a heuristic algorithm, which takes the importance of local updates into account, to achieve a suboptimal solution in polynomial time. Numerical results demonstrate that the proposed algorithm can reduce the training latency compared to the scheme with greedy client association and myopic energy control, and improve the learning performance compared to the scheme in which the associated clients transmit their local models with the maximal power.
Year
DOI
Venue
2021
10.1109/WCNC49053.2021.9417267
2021 IEEE WIRELESS COMMUNICATIONS AND NETWORKING CONFERENCE (WCNC)
DocType
ISSN
Citations 
Conference
1525-3511
0
PageRank 
References 
Authors
0.34
0
5
Name
Order
Citations
PageRank
Bo Xu110.70
Wenchao Xia201.01
Jun Zhang3748.17
Xinghua Sun49011.38
Hongbo Zhu536785.53