Affiliation:
1. School of Information Engineer, East China Jiaotong University, Nanchang 330013, China
2. School of Science, East China Jiaotong University, Nanchang 330013, China
Abstract
Federated learning (FL) has emerged as a promising framework for collaborative machine learning, allowing the training of machine learning models on distributed devices without centralizing sensitive data. However, FL falls short in terms of fairness, as each client receives the same model regardless of their individual contributions. This unfairness discourages active client participation in FL. To address this challenge, we propose a contribution-based differentiated global model mechanism. Specifically, we introduce the contribution score as a metric to assess client contributions in FL and utilize deep Q-networks (DQN) to dynamically update the contribution scores. Subsequently, we allocate clients to different clusters based on their contributions by using a clustering algorithm, where each cluster is associated with a distinct global model. This mechanism encourages clients to make greater contributions for improved global models. Experimental results confirm the effectiveness of our approach in enhancing fairness in FL.
Funder
National Natural Science Foundation of China
Subject
Artificial Intelligence,Human-Computer Interaction,Theoretical Computer Science,Software