Journal
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS
Volume 26, Issue 11, Pages 5596-5607Publisher
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/JBHI.2022.3198440
Keywords
Adaptation models; Data models; Feature extraction; Analytical models; Task analysis; Biological system modeling; Training; Federated learning; personalization; cancer diagnosis; skin cancer diagnosis
Categories
Funding
- Hong Kong General Research Fund [16203319]
- National Natural Science Foundation of China [61872417, 62061160490]
Ask authors/readers for more resources
The performance of deep networks for medical image analysis is often limited by the scarcity of medical data and privacy concerns. To address this issue, we propose CusFL, a customized federated learning approach that enables each client to train a client-specific model based on a federated global model. By using a federated feature extractor for guidance, CusFL allows clients to selectively learn useful knowledge from the federated model and improve their personalized models.
The performance of deep networks for medical image analysis is often constrained by limited medical data, which is privacy-sensitive. Federated learning (FL) alleviates the constraint by allowing different institutions to collaboratively train a federated model without sharing data. However, the federated model is often suboptimal with respect to the characteristics of each client's local data. Instead of training a single global model, we propose Customized FL (CusFL), for which each client iteratively trains a client-specific/private model based on a federated global model aggregated from all private models trained in the immediate previous iteration. Two overarching strategies employed by CusFL lead to its superior performance: 1) the federated model is mainly for feature alignment and thus only consists of feature extraction layers; 2) the federated feature extractor is used to guide the training of each private model. In that way, CusFL allows each client to selectively learn useful knowledge from the federated model to improve its personalized model. We evaluated CusFL on multi-source medical image datasets for the identification of clinically significant prostate cancer and the classification of skin lesions.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available