Two-Stage Client Selection for Federated Learning Against Free-riding Attack: A Multi-Armed Bandits and Auction-Based Approach
Renhao Lu,Weizhe Zhang,Hui He,Qiong Li,Xiaoxiong Zhong,Hongwei Yang,Desheng Wang,Lu Shi,Yuelin Guo,Zejun Wang
DOI: https://doi.org/10.1109/jiot.2024.3431555
2024-01-01
Abstract:Utilizing the federated learning (FL) technique, data owners can collaboratively train artificial intelligence models, retaining all training data on their premises to minimize the potential for personal data breaches. However, self-interested users (e.g., free riders) bring new challenges that hinder the development of FL techniques. To this end, we propose a two-stage client selection scheme comprising a multiarmed bandit (MAB)-based candidate client selection method and an auction-based training client selection method. Specifically, our client selection scheme initially formulates the FL system into an MAB system, where clients are the arms and the server is the player. Then, we quantify the similarity between a local model and the server side, which is the designed metric for model aggregation and reward computation updating based on the fuzzy mathematical strategy. Next, based on the Thompson Sampling strategy, the server can intelligently determine the reward of each client, and clients with more significant rewards have the chance for local model training. With an auction method, the server can determine the training clients to reduce the training cost while maximizing each client's revenue. Extensive experiments on real-world data sets demonstrate that the proposed scheme outperforms representative FL schemes (i.e., FedAvg, FedProx, FedMax, and MFL) regarding the model's convergence rate and cost in FL systems with free riders.