A Federated Learning Framework Via Decentralized Data Valuation for Chronic Disease Healthcare

Jinhua Li,Jiajin He,Ling Hu,Chao Wu
DOI: https://doi.org/10.1145/3669828.3669837
2024-01-01
Abstract:Medical data on chronic diseases are rife with the private information of patients. These data are often stored in many isolated institutions and cannot be shared due to privacy and security concerns. Federated learning provides a privacy protection solution for collaboration among multiple medical institutions by exchanging model parameters instead of private data. However, in real-world healthcare scenarios, the quality of medical data owned by different medical institutions is highly distinct, which poses a critical challenge for the application of federated learning. To tackle this problem, we propose a client level division based federated learning algorithm that estimates the quality of all data sources to fairly coordinate the training of the federated model among multiple participants. Specifically, we quantify the value of clients by testing their local models. Then we divide all clients into several levels based on their data quality and enable clients with high-quality data to have a higher priority in federated learning. Our method provides an insight into which participants are more valuable for the prediction task, thereby facilitating the model training process. Experiments on the CHRALS dataset demonstrate that our method accelerates the convergence of federated learning and improves the performance of the federated model.
What problem does this paper attempt to address?