Reputation based Adaptive Federated Learning in Medical Scenarios
Lianhai Wang, Tianrui Liu, Yuelu Wang, Xinlei Wang, Qi Li · 2024
Federated Learning (FL) is a distributed machine learning paradigm designed to address data silos and protect data privacy. However, in medical scenarios, the heterogeneity in data quality and the non-independent and identically distributed (Non-IID) nature of medical data from different hospitals can lead to degraded model performance in FL. To tackle this challenge, this paper proposes an adaptive federated learning framework based on reputation. By using a reputation evaluation model, the framework adaptively adjusts the aggregation weights of the models to enhance the performance of the global model. Additionally, our reputation evaluation model, integrated with blockchain technology, can rapidly identify malicious clients and effectively prevent poisoning attacks. Experimental results on two medical datasets demonstrate that this framework achieves higher model accuracy and faster convergence.