An Incentive Mechanism Based on AoU, Data Quality, and Data Quantity for Federated Learning

Zhaohua Zheng,Yiming Hong,Keqiu Li,Qiquan Chen
DOI: https://doi.org/10.1109/cscwd61410.2024.10580290
2024-01-01
Abstract:Federated Learning (FL) is a new distributed machine learning that allows end devices to collaboratively train a global model using local data to protect data privacy. However, the existing FL incentive mechanisms only consider a single round’s data contribution, which needs improvement. For non-IID datasets, continuous selection of any end device will cause weights to diverge in a specific direction. Therefore, a new metric is needed to avoid continuously selecting a particular end device to ensure the overall performance of the incentive mechanism. We have introduced a metric to describe the importance of updates: Age of Update (AoU), which can help select end devices not selected in the previous round to promote faster model convergence. We proposed an incentive mechanism for a Two-stage Stackelberg Game with Multidimensional Individual Attributes (TSGMIA). We have derived the optimal equilibrium solution for the game in two stages. Based on this solution, we can ensure maximum server utility while incentivizing clients to participate actively in FL tasks and providing high-quality data to train the FL model. Finally, we conducted extensive simulation experiments to demonstrate that our proposed mechanism can effectively improve the performance of the FL model. Specifically, when the model accuracy reaches 0.9, compared to InFEDge, the communication round of TSGMIA will be reduced by 9%. In addition, compared with the fixed selection scheme, random selection scheme, and InFEDge, after 100 rounds of training, TSGMIA improved the accuracy of the global model by 30%, 10%, and 1.7%, respectively.
What problem does this paper attempt to address?