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

Liangkun Yu, Rana Albelaihi, Xiang Sun, Nirwan Ansari, Michael Devetsikiotis

Research output: Contribution to journalArticlepeer-review

96 Scopus citations

Abstract

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.

Original languageEnglish (US)
Pages (from-to)4385-4395
Number of pages11
JournalIEEE Internet of Things Journal
Volume9
Issue number6
DOIs
StatePublished - Mar 15 2022

All Science Journal Classification (ASJC) codes

  • Signal Processing
  • Information Systems
  • Hardware and Architecture
  • Computer Science Applications
  • Computer Networks and Communications

Keywords

  • Client selection
  • energy consumption
  • federated learning (FL)
  • waiting time

Fingerprint

Dive into the research topics of 'Jointly Optimizing Client Selection and Resource Management in Wireless Federated Learning for Internet of Things'. Together they form a unique fingerprint.

Cite this