CoSinGAN: Learning COVID-19 Infection Segmentation from a Single Radiological Image
Open Access
- 3 November 2020
- journal article
- research article
- Published by MDPI AG in Diagnostics
- Vol. 10 (11), 901
- https://doi.org/10.3390/diagnostics10110901
Abstract
Computed tomography (CT) images are currently being adopted as the visual evidence for COVID-19 diagnosis in clinical practice. Automated detection of COVID-19 infection from CT images based on deep models is important for faster examination. Unfortunately, collecting large-scale training data systematically in the early stage is difficult. To address this problem, we explore the feasibility of learning deep models for lung and COVID-19 infection segmentation from a single radiological image by resorting to synthesizing diverse radiological images. Specifically, we propose a novel conditional generative model, called CoSinGAN, which can be learned from a single radiological image with a given condition, i.e., the annotation mask of the lungs and infected regions. Our CoSinGAN is able to capture the conditional distribution of the single radiological image, and further synthesize high-resolution (512 × 512) and diverse radiological images that match the input conditions precisely. We evaluate the efficacy of CoSinGAN in learning lung and infection segmentation from very few radiological images by performing 5-fold cross validation on COVID-19-CT-Seg dataset (20 CT cases) and an independent testing on the MosMed dataset (50 CT cases). Both 2D U-Net and 3D U-Net, learned from four CT slices by using our CoSinGAN, have achieved notable infection segmentation performance, surpassing the COVID-19-CT-Seg-Benchmark, i.e., the counterparts trained on an average of 704 CT slices, by a large margin. Such results strongly confirm that our method has the potential to learn COVID-19 infection segmentation from few radiological images in the early stage of COVID-19 pandemic.Keywords
This publication has 28 references indexed in Scilit:
- Chest CT Findings in 2019 Novel Coronavirus (2019-nCoV) Infections from Wuhan, China: Key Points for the RadiologistRadiology, 2020
- Nowcasting and forecasting the potential domestic and international spread of the 2019-nCoV outbreak originating in Wuhan, China: a modelling studyThe Lancet, 2020
- Semantic photo manipulation with a generative image priorACM Transactions on Graphics, 2019
- A Style-Based Generator Architecture for Generative Adversarial NetworksPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2019
- Articulating the Logic of China’s Postsocialist Society in the Era of Neo-Liberal TransformationPublished by Brill ,2018
- Non-stationary texture synthesis by adversarial expansionACM Transactions on Graphics, 2018
- High-Resolution Image Synthesis and Semantic Manipulation with Conditional GANsPublished by Institute of Electrical and Electronics Engineers (IEEE) ,2018
- Loss Functions for Image Restoration With Neural NetworksIEEE Transactions on Computational Imaging, 2016
- U-Net: Convolutional Networks for Biomedical Image SegmentationPublished by Springer Science and Business Media LLC ,2015
- Image Quality Assessment: From Error Visibility to Structural SimilarityIEEE Transactions on Image Processing, 2004