Scale Federated Learning for Label Set Mismatch in Medical Image Classification
Federated learning (FL) has been introduced to the healthcare domain as a decentralized learning paradigm that allows multiple parties to train a model collaboratively without privacy leakage. However, most previous studies have assumed that every client holds an identical label set. In reality, medical specialists tend to annotate only diseases within their area of expertise or interest. This implies that label sets in each client can be different and even disjoint. In this paper, we propose the framework FedLSM to solve the problem of Label Set Mismatch. FedLSM adopts different training strategies on data with different uncertainty levels to efficiently utilize unlabeled or partially labeled data as well as class-wise adaptive aggregation in the classification layer to avoid inaccurate aggregation when clients have missing labels. We evaluated FedLSM on two public real-world medical image datasets, including chest X-ray (CXR) diagnosis with 112,120 CXR images and skin lesion diagnosis with 10,015 dermoscopy images, and showed that it significantly outperformed other state-of-the-art FL algorithms. The code can be found at https://github.com/dzp2095/FedLSM.
Code (1)
Tasks
Federated Learningimage-classificationImage ClassificationMedical Image ClassificationMissing LabelsSimilar Papers 제목 키워드 기반
Learning Underrepresented Classes from Decentralized Partially Labeled Medical Images
Using decentralized data for federated training is one promising emerging research direction for alleviating data scarcity in the medical domain. However, in contrast to large-scale fully labeled data commonly seen in ge…
Federated LearningObject RecognitionSelf-Supervised LearningThoracic Disease ClassificationFederated Semi-supervised Medical Image Classification via Inter-client Relation Matching
Federated learning (FL) has emerged with increasing popularity to collaborate distributed medical institutions for training deep networks. However, despite existing FL algorithms only allow the supervised training settin…
Federated Learningimage-classificationImage ClassificationMedical Image Classification+2Federated Contrastive Learning for Decentralized Unlabeled Medical Images
A label-efficient paradigm in computer vision is based on self-supervised contrastive pre-training on unlabeled data followed by fine-tuning with a small number of labels. Making practical use of a federated computing en…
Contrastive LearningData AugmentationFederated LearningFederated Medical Image Segmentation under Real-World Label Noise: A Benchmark Suite for Noisy Label Learning Method Selection
While federated learning (FL) enables collaborative medical image segmentation without centralizing sensitive data, real-world deployment is frequently complicated by cross-site label imperfections such as contour disagr…
Medical Image SegmentationFederated LearningLabel-Efficient Self-Supervised Federated Learning for Tackling Data Heterogeneity in Medical Imaging
The collection and curation of large-scale medical datasets from multiple institutions is essential for training accurate deep learning models, but privacy concerns often hinder data sharing. Federated learning (FL) is a…
Federated LearningMedical Image AnalysisPrivacy PreservingRepresentation Learning+2