UM  > Faculty of Science and Technology
Residential Collegefalse
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 PublicationNeural Networks
ISSN0893-6080
Volume178Pages: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.

KeywordCatastrophic Forgetting Federated Learning Invalid Aggregation Multi-center Disease Diagnosis Non-iid Data
DOI10.1016/j.neunet.2024.106409
URLView the original
Indexed BySCIE
Language英語English
WOS Research AreaComputer Science ; Neurosciences & Neurology
WOS SubjectComputer Science, Artificial Intelligence ; Neurosciences
WOS IDWOS:001248529600003
PublisherPERGAMON-ELSEVIER SCIENCE LTD, THE BOULEVARD, LANGFORD LANE, KIDLINGTON, OXFORD OX5 1GB, ENGLAND
Scopus ID2-s2.0-85194970800
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionFaculty of Science and Technology
DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Corresponding AuthorLei, Baiying; Wang, Tianfu
Affiliation1.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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Du, Jie]'s Articles
[Li, Wei]'s Articles
[Liu, Peng]'s Articles
Baidu academic
Similar articles in Baidu academic
[Du, Jie]'s Articles
[Li, Wei]'s Articles
[Liu, Peng]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Du, Jie]'s Articles
[Li, Wei]'s Articles
[Liu, Peng]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.