Client Selection for Federated Learning Using Combinatorial Multi-Armed Bandit under Long-Term Energy Constraint

Konglin Zhu,Fuchun Zhang,Lei Jiao,Bowei Xue,Lin Zhang
DOI: https://doi.org/10.1016/j.comnet.2024.110512
IF: 5.493
2024-01-01
Computer Networks
Abstract:Due to the submodularity of outcome in FL system, the more clients involve, the less increment of outcome returns. Therefore, it is essential to design a client selection strategy to choose an appropriate subset of all clients to participate in model training. However, the client selection is not easy due to the heterogeneity of clients and the long-term energy budget constraints. Moreover, independent long-term energy budgets of clients intertwined with short-term selection often make the problem become NP-hard. In this paper, we propose an online strategy Energy Aware Client Selection for FL (EACS-FL) to address above mentioned problem. The problem is formulated with the combined energy and delay optimization target, and the Combinatorial Multi-Arm Bandit (CMAB) is introduced to solve the problem in the online manner. We take advantage of Lyapunov optimization to manage energy consumption of clients which enable us to deal with independent energy budgets through minimizing virtual energy deficit queues. Theoretical analysis shows that EACS-FL achieves sublinear regret and keeps all queues stable. Experiment results show that the proposed approach outperforms the existing works and achieves close-to-optimal delay and energy consumption performance.
What problem does this paper attempt to address?