Online Client Scheduling for Fast Federated Learning

Bo Xu, Wenchao Xia, Jun Zhang, Tony Q.S. Quek, Hongbo Zhu

Research output: Contribution to journalArticlepeer-review

Abstract

Federated learning (FL) enables clients to collaboratively learn a shared task while keeping data privacy, which can be adopted at the edge of wireless networks to improve edge intelligence. In this letter, we aim to minimize the training latency of a wireless FL system for a given training loss by client scheduling. Instead of assuming that the prior information about wireless channel state and local computing power of the clients is available, we consider a more practical scenario without knowing the prior information. We first reformulate the client scheduling problem as a multi-armed bandit program and then propose an online scheduling scheme based on ϵ -greedy algorithm to achieve a tradeoff between exploration and exploitation. In addition, the proposed client scheduling scheme reduces the number of training rounds and the time interval per round simultaneously by jointly considering the significance of local updates and delay issue of each client. Simulation results show that in the case of non-independent and identically distributed data, the proposed scheme can save half the training time compared to the scheme which only considers the significance of local updates, and can improve more than 20% test accuracy compared to the scheme which only considers the time consumption per round of each client.

Original languageEnglish
Article number9390199
Pages (from-to)1434-1438
Number of pages5
JournalIEEE Wireless Communications Letters
Volume10
Issue number7
DOIs
Publication statusPublished - 2021 Jul

All Science Journal Classification (ASJC) codes

  • Control and Systems Engineering
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Online Client Scheduling for Fast Federated Learning'. Together they form a unique fingerprint.

Cite this