Federated Semi-Supervised Learning for COVID Region Segmentation in Chest CT using Multi-National Data from China, Italy, Japan
arXiv:2011.11750
Abstract
The recent outbreak of COVID-19 has led to urgent needs for reliable diagnosis and management of SARS-CoV-2 infection. As a complimentary tool, chest CT has been shown to be able to reveal visual patterns characteristic for COVID-19, which has definite value at several stages during the disease course. To facilitate CT analysis, recent efforts have focused on computer-aided characterization and diagnosis, which has shown promising results. However, domain shift of data across clinical data centers poses a serious challenge when deploying learning-based models. In this work, we attempt to find a solution for this challenge via federated and semi-supervised learning. A multi-national database consisting of 1704 scans from three countries is adopted to study the performance gap, when training a model with one dataset and applying it to another. Expert radiologists manually delineated 945 scans for COVID-19 findings. In handling the variability in both the data and annotations, a novel federated semi-supervised learning technique is proposed to fully utilize all available data (with or without annotations). Federated learning avoids the need for sensitive data-sharing, which makes it favorable for institutions and nations with strict regulatory policy on data privacy. Moreover, semi-supervision potentially reduces the annotation burden under a distributed setting. The proposed framework is shown to be effective compared to fully supervised scenarios with conventional data sharing instead of model weight sharing.
arXiv commentAccepted with minor revision to Medical Image Analysis
References in corpus (15)
- Semi-Supervised Classification with Graph Convolutional Networks
- Communication-Efficient Learning of Deep Networks from Decentralized Data
- Improved Regularization of Convolutional Neural Networks with Cutout
- FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidence
- Unsupervised Representation Learning by Predicting Image Rotations
- Interpolation Consistency Training for Semi-Supervised Learning
- Diagnosis of Coronavirus Disease 2019 (COVID-19) with Structured Latent Multi-View Representation Learning
- Self-training with Noisy Student improves ImageNet classification
- ReMixMatch: Semi-Supervised Learning with Distribution Alignment and Augmentation Anchoring
- One-Shot Federated Learning
- Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentation
- Federated Learning for Healthcare Informatics
- Privacy-preserving Federated Brain Tumour Segmentation
- Towards Utilizing Unlabeled Data in Federated Learning: A Survey and Prospective
- Multi-Institutional Deep Learning Modeling Without Sharing Patient Data: A Feasibility Study on Brain Tumor Segmentation