Label-free coronavirus disease 2019 lesion segmentation based on synthetic healthy lung image subtraction

Coronavirus disease 2019 (COVID-19) has become a global pandemic and is still posing a severe health risk to the public. Accurate and efficient segmentation of pneumonia lesions in computed tomography (CT) scans is vital for treatment decision-making. We proposed a novel unsupervised approach using a cycle consistent generative adversarial network (cycle-GAN) which automates and accelerates the process of lesion delineation.

The workflow includes lung volume segmentation, healthy lung image synthesis, infected and healthy image subtraction, and binary lesion mask generation. The lung volume was first delineated using a pre-trained U-net and worked as the input for the following network. A cycle-GAN was trained to generate synthetic healthy lung CT images from infected lung images. After that, the pneumonia lesions were extracted by subtracting the synthetic healthy lung CT images from the infected lung CT images. A median filter and k-means clustering were then applied to contour the lesions. The auto segmentation approach was validated on three different datasets.

The average Dice coefficient reached 0.666 ± 0.178 on the three datasets. Especially, the dice reached 0.748 ± 0.121 and 0.730 ± 0.095, respectively, on two public datasets Coronacases and Radiopedia. Meanwhile, the average precision and sensitivity for lesion segmentation on the three datasets were 0.679 ± 0.244 and 0.756 ± 0.162. The performance is comparable to existing supervised segmentation networks and outperforms unsupervised ones.

The proposed label-free segmentation method achieved high accuracy and efficiency in automatic COVID-19 lesion delineation. The segmentation result can serve as a baseline for further manual modification and a quality assurance tool for lesion diagnosis. Furthermore, due to its unsupervised nature, the result is not influenced by physicians' experience which otherwise is crucial for supervised methods.