A Novel Lightweight Swin-Unet Network for Semantic Segmentation of COVID-19 Lesion in CT Images

Abstract

The Corona Virus Disease 2019 (COVID-19) is highly infectious, has been spread worldwide, caused a global pandemic, and seriously endangered human health and life. The most effective methods for halting and stopping the transmission of the Corona Virus include early detection, quarantine, and successful treatment. Because it exhibits significant imaging characteristics for COVID-19 lesions in chest computed tomography (CT), it can be used to diagnose COVID-19. Aiming at the inaccuracies of uneven gray distribution, irregular regions, multi-scale, and multi-region segmentation in COVID-19 CT images. This paper proposed a novel Swin-Unet network to improve the accuracy of multi-scale lesion segmentation in COVID-19 CT images. First, in the double-layer Swin Transformer blocks of the Swin-Unet, a residual multi-layer perceptron (ResMLP) module was introduced and replaced the multi-layer perceptron (MLP) module to reduce the loss of features during the transmission process, thereby improving the segmentation precision of multi-scale lesion areas. Second, the uncertain region inpainting module (URIM) was added after Linear Projection, which can refine the uncertain regions in the segmentation features map, thereby improving the segmentation accuracy of different lesion regions. Third, a new loss function DF was designed. It can effectively improve the small target segmentation effect and thus improve the multi-scale segmentation result. Finally, the proposed method was compared to other methods on the public dataset. The Dice, Precision, Recall, and IOU of the proposed method are 0.812, 0.780, 0.848, and 0.683, respectively, which are better than the other models. Moreover, our model has fewer parameters and faster reasoning speed. The proposed method achieves excellent segmentation results for multi-scale and multi-region lesions, and it will be more beneficial in aiding COVID-19 diagnosis and treatment.

Keywords

• COVID-19
• CT image
• semantic segmentation
• Swin-Unet
• ResMLP