Jointly Optimizing Client Selection and Resource Management in Wireless Federated Learning for Internet of Things

Federated learning (FL) has been proposed to efficiently and privacy-preserving distributed machine learning architecture for the Internet of Things (IoT). In a wireless FL system, clients in IoT devices train their local models over the local data sets. The derived local models are uploaded to an FL server to generate a global model, broadcasted to the clients in the next global iteration for further training. Owing to the heterogeneous feature of the clients, client selection is critical to determine the overall training time. Traditionally, the objective of client selection is to select the maximum number of clients who can derive and upload their local models before the deadline in each global iteration. However, selecting more clients increases the energy consumption of the clients. Moreover, selecting the maximum number of clients is unnecessary as having fewer clients in early global iterations and more clients in later global iterations have been proved to achieve higher model accuracy. Hence, this article proposes to dynamically adjust and optimize the tradeoff between maximizing the number of selected clients and minimizing the total energy consumption of the clients by selecting suitable clients and allocating appropriate resources in terms of CPU frequency and transmission power. We formulate the joint client selection and resource management problem and design the energy and latency-aware resource management and client selection (ELASTIC) algorithm to efficiently solve the problem. Extensive simulations are conducted to demonstrate the performance of ELASTIC.