In deep learning, local training methods often lead to overfitting and data silos due to privacy concerns and regulatory challenges. Federated learning addresses these issues by keeping data local and aggregating local models to update a global model. However, challenges like data heterogeneity and malicious client behaviors can degrade model performance, and fairness is often overlooked. This paper proposes a federated learning aggregation algorithm based on Shapley values, ensuring fairness by adjusting client weights and sampling strategies based on their marginal contributions. Using Shapley values and Monte Carlo sampling, the algorithm optimizes client weights, improving overall model performance while mitigating the impact of malicious clients and data heterogeneity. Experimental results on datasets like CIFAR-10 and Fashion MNIST demonstrate that the proposed algorithm balances fairness and effectiveness, outperforming traditional frameworks like Federated Averaging, Federated Proximal, and Collaborative Fairness in Federated Learning. This study provides an effective solution for federated learning in heterogeneous data environments, addressing fairness and robustness issues.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Research on a Federated Learning Algorithm Based on Shapley Value Aggregation

  • Dashan Zhang,
  • Hong Zhang

摘要

In deep learning, local training methods often lead to overfitting and data silos due to privacy concerns and regulatory challenges. Federated learning addresses these issues by keeping data local and aggregating local models to update a global model. However, challenges like data heterogeneity and malicious client behaviors can degrade model performance, and fairness is often overlooked. This paper proposes a federated learning aggregation algorithm based on Shapley values, ensuring fairness by adjusting client weights and sampling strategies based on their marginal contributions. Using Shapley values and Monte Carlo sampling, the algorithm optimizes client weights, improving overall model performance while mitigating the impact of malicious clients and data heterogeneity. Experimental results on datasets like CIFAR-10 and Fashion MNIST demonstrate that the proposed algorithm balances fairness and effectiveness, outperforming traditional frameworks like Federated Averaging, Federated Proximal, and Collaborative Fairness in Federated Learning. This study provides an effective solution for federated learning in heterogeneous data environments, addressing fairness and robustness issues.