Residential College | false |
Status | 已發表Published |
Federated learning using model projection for multi-center disease diagnosis with non-IID data | |
Du, Jie1; Li, Wei1; Liu, Peng2; Vong, Chi Man3; You, Yongke4; Lei, Baiying1; Wang, Tianfu1 | |
2024-10-01 | |
Source Publication | Neural Networks |
ISSN | 0893-6080 |
Volume | 178Pages:106409 |
Abstract | Multi-center disease diagnosis aims to build a global model for all involved medical centers. Due to privacy concerns, it is infeasible to collect data from multiple centers for training (i.e., centralized learning). Federated Learning (FL) is a decentralized framework that enables multiple clients (e.g., medical centers) to collaboratively train a global model while retaining patient data locally for privacy. However, in practice, the data across medical centers are not independently and identically distributed (Non-IID), causing two challenging issues: (1) catastrophic forgetting at clients, i.e., the local model at clients will forget the knowledge received from the global model after local training, causing reduced performance; and (2) invalid aggregation at the server, i.e., the global model at the server may not be favorable to some clients after model aggregation, resulting in a slow convergence rate. To mitigate these issues, an innovative Federated learning using Model Projection (FedMoP) is proposed, which guarantees: (1) the loss of local model on global data does not increase after local training without accessing the global data so that the performance will not be degenerated; and (2) the loss of global model on local data does not increase after aggregation without accessing local data so that convergence rate can be improved. Extensive experimental results show that our FedMoP outperforms state-of-the-art FL methods in terms of accuracy, convergence rate and communication cost. In particular, our FedMoP also achieves comparable or even higher accuracy than centralized learning. Thus, our FedMoP can ensure privacy protection while outperforming centralized learning in accuracy and communication cost. |
Keyword | Catastrophic Forgetting Federated Learning Invalid Aggregation Multi-center Disease Diagnosis Non-iid Data |
DOI | 10.1016/j.neunet.2024.106409 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science ; Neurosciences & Neurology |
WOS Subject | Computer Science, Artificial Intelligence ; Neurosciences |
WOS ID | WOS:001248529600003 |
Publisher | PERGAMON-ELSEVIER SCIENCE LTD, THE BOULEVARD, LANGFORD LANE, KIDLINGTON, OXFORD OX5 1GB, ENGLAND |
Scopus ID | 2-s2.0-85194970800 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Lei, Baiying; Wang, Tianfu |
Affiliation | 1.National-Regional Key Technology Engineering Lab for Medical Ultrasound, Guangdong Key Lab for Biomedical Measurements and Ultrasound Imaging, School of Biomedical Engineering, Medical School, Shenzhen University 2.Artificial Intelligence Industrial Innovation Research Center, Shenzhen Institute for Advanced Study, University of Electronic Science and Technology of China, Shenzhen, 518110, China 3.Department of Computer and Information Science, University of Macau, Macau SAR, 999078, China 4.Department of nephrology, Shenzhen University General Hospital, Shenzhen University, Shenzhen 518060, Guangdong, China |
Recommended Citation GB/T 7714 | Du, Jie,Li, Wei,Liu, Peng,et al. Federated learning using model projection for multi-center disease diagnosis with non-IID data[J]. Neural Networks, 2024, 178, 106409. |
APA | Du, Jie., Li, Wei., Liu, Peng., Vong, Chi Man., You, Yongke., Lei, Baiying., & Wang, Tianfu (2024). Federated learning using model projection for multi-center disease diagnosis with non-IID data. Neural Networks, 178, 106409. |
MLA | Du, Jie,et al."Federated learning using model projection for multi-center disease diagnosis with non-IID data".Neural Networks 178(2024):106409. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment