An Efficiency-boosting Client Selection Scheme for Federated Learning with Fairness Guarantee

Abstract

The issue of potential privacy leakage during centralized AI's model training has drawn intensive concern from the public. A Parallel and Distributed Computing (or PDC) scheme, termed Federated Learning (FL), has emerged as a new paradigm to cope with the privacy issue by allowing clients to perform model training locally, without the necessity to upload their personal sensitive data. In FL, the number of clients could be sufficiently large, but the bandwidth available for model distribution and re-upload is quite limited, making it sensible to only involve part of the volunteers to participate in the training process. The client selection policy is critical to an FL process in terms of training efficiency, the final model's quality as well as fairness. In this paper, we will model the fairness guaranteed client selection as a Lyapunov optimization problem and then a C2MAB-based method is proposed for estimation of the model exchange time between each client and the server, based on which we design a fairness guaranteed algorithm termed RBCS-F for problem-solving. The regret of RBCS-F is strictly bounded by a finite constant, justifying its theoretical feasibility. Barring the theoretical results, more empirical data can be derived from our real training experiments on public datasets.

Accepted by IEEE TPDS. DOI: 10.1109/TPDS.2020.3040887

Cited by in corpus (6)

• Towards Fairness-Aware Federated Learning
• Towards Reliable Participation in UAV-Enabled Federated Edge Learning on Non-IID Data
• Gradient Sparsification for Efficient Wireless Federated Learning with Differential Privacy
• Fairness in Federated Learning: Trends, Challenges, and Opportunities
• Cyber Attacks Prevention Towards Prosumer-based EV Charging Stations: An Edge-assisted Federated Prototype Knowledge Distillation Approach
• From Learning to Analytics: Improving Model Efficacy with Goal-Directed Client Selection