Federated aggregation method based on cosine similarity approximation Shapley value method contribution degree

Abstract

When calculating participants’ contribution to federated learning, addressing issues such as the inability to collect complete test data and the impact of malicious and dishonest participants on the global model is necessary. This article proposes a federated aggregation method based on cosine similarity approximation Shapley value method contribution degree. Firstly, a participant contribution calculation model combining cosine similarity and the approximate Shapley value method was designed to obtain the contribution values of the participants. Then, based on the calculation model of participant contribution, a federated aggregation algorithm is proposed, and the aggregation weights of each participant in the federated aggregation process are calculated by their contribution values. Finally, the gradient parameters of the global model were determined and propagated to all participants to update the local model. Experiments were conducted under different privacy protection parameters, data noise parameters, and the proportion of malicious participants. The results showed that the accuracy of the algorithm model can be maintained at 90% and 65% on the MNIST and CIFAR-10 datasets, respectively. This method can reasonably and accurately calculate the contribution of participants without a complete test dataset, reducing computational costs to a certain extent and can resist the influence of the aforementioned participants.