COVID-19 Image Segmentation Algorithms Based on Conditional Generative Adversarial Network

ABSTRACT

The size and location of the lesions in CT images of novel corona virus pneumonia (COVID-19) change all the time, and the lesion areas have low contrast and blurred boundaries, resulting in difficult segmentation. To solve this problem, a COVID-19 image segmentation algorithm based on conditional generative adversarial network (CGAN) is proposed. Uses the improved DeeplabV3+ network as a generator, which enhances the extraction of multi-scale contextual features, reduces the number of network parameters and improves the training speed. A Markov discriminator with 6 fully convolutional layers is proposed instead of a common discriminator, with the aim of focusing more on the local features of the CT image. By continuously adversarial training between the generator and the discriminator, the network weights are optimised so that the final segmented image generated by the generator is infinitely close to the ground truth. On the COVID-19 CT public dataset, the area under the curve of ROC, F1-Score and dice similarity coefficient achieved 96.64%, 84.15% and 86.14% respectively. The experimental results show that the proposed algorithm is accurate and robust, and it has the possibility of becoming a safe, inexpensive, and time-saving medical assistant tool in clinical diagnosis, which provides a reference for computer-aided diagnosis. © 2022 ACM.