Affiliation:
1. Department of Information and Computer Engineering, Chung Yuan Christian University, Taoyuan 320, Taiwan
Abstract
Considering the sensitivity of data in medical scenarios, federated learning (FL) is suitable for applications that require data privacy. Medical personnel can use the FL framework for machine learning to assist in analyzing large-scale data that are protected within the institution. However, not all clients have the same distribution of datasets, so data imbalance problems occur among clients. The main challenge is to overcome the performance degradation caused by low accuracy and the inability to converge the model. This paper proposes a FedISM method to enhance performance in the case of Non-Independent Identically Distribution (Non-IID). FedISM exploits a shared model trained on a candidate dataset before performing FL among clients. The Candidate Selection Mechanism (CSM) was proposed to effectively select the most suitable candidate among clients for training the shared model. Based on the proposed approaches, FedISM not only trains the shared model without sharing any raw data, but it also provides an optimal solution through the selection of the best shared model. To evaluate performance, the proposed FedISM was applied to classify coronavirus disease (COVID), pneumonia, normal, and viral pneumonia in the experiments. The Dirichlet process was also used to simulate a variety of imbalanced data distributions. Experimental results show that FedISM improves accuracy by up to 25% when privacy concerns regarding patient data are rising among medical institutions.
Funder
National Science and Technology Council (NSTC) of Taiwan
Subject
General Mathematics,Engineering (miscellaneous),Computer Science (miscellaneous)
Reference52 articles.
1. COVID-Net: A Tailored Deep Convolutional Neural Network Design for Detection of COVID-19 Cases from Chest X-ray Images;Wang;Sci. Rep.,2020
2. Deep Learning for COVID-19 Detection Based on CT Images;Zhao;Sci. Rep.,2021
3. Imagenet Classification with Deep Convolutional Neural Networks;Krizhevsky;Commun. ACM,2017
4. McMahan, B., Moore, E., Ramage, D., Hampson, S., and y Arcas, B.A. (2017, January 20–27). Communication-Efficient Learning of Deep Networks from Decentralized Data. Proceedings of the Artificial Intelligence and Statistics, Fort Lauderdale, FL, USA.
5. Sikandar, H.S., Waheed, H., Tahir, S., Malik, S.U., and Rafique, W. (2023). A Detailed Survey on Federated Learning Attacks and Defenses. Electronics, 12.