Abstract:Coronavirus disease (COVID-19) broke out at the end of 2019, and has resulted in an ongoing global pandemic. Segmentation of pneumonia infections from chest computed tomography (CT) scans of COVID-19 patients is significant for accurate diagnosis and quantitative analysis. Deep learning-based methods can be developed for automatic segmentation and offer a great potential to strengthen timely quarantine and medical treatment. Unfortunately, due to the urgent nature of the COVID-19 pandemic, a systematic collection of CT data sets for deep neural network training is quite difficult, especially high-quality annotations of multi-category infections are limited. In addition, it is still a challenge to segment the infected areas from CT slices because of the irregular shapes and fuzzy boundaries. To solve these issues, we propose a novel COVID-19 pneumonia lesion segmentation network, called Spatial Self-Attention network (SSA-Net), to identify infected regions from chest CT images automatically. In our SSA-Net, a self-attention mechanism is utilized to expand the receptive field and enhance the representation learning by distilling useful contextual information from deeper layers without extra training time, and spatial convolution is introduced to strengthen the network and accelerate the training convergence. Furthermore, to alleviate the insufficiency of labeled multi-class data and the long-tailed distribution of training data, we present a semi-supervised few-shot iterative segmentation framework based on re-weighting the loss and selecting prediction values with high confidence, which can accurately classify different kinds of infections with a small number of labeled image data. Experimental results show that SSA-Net outperforms state-of-the-art medical image segmentation networks and provides clinically interpretable saliency maps, which are useful for COVID-19 diagnosis and patient triage. Meanwhile, our semi-supervised iterative segmentation model can improve the learning ability in small and unbalanced training set and can achieve higher performance.

Self-Ensembling Co-Training Framework for Semi-Supervised COVID-19 CT Segmentation

Pixel-Level and Affinity-Level Knowledge Distillation for Unsupervised Segmentation of Covid-19 Lesions

C3D-UNET: A Comprehensive 3D Unet for Covid-19 Segmentation with Intact Encoding and Local Attention

Deep Co-supervision and Attention Fusion Strategy for Automatic COVID-19 Lung Infection Segmentation on CT Images

Weakly supervised segmentation of COVID-19 infection with local lesion coherence on CT images

Federated semi-supervised learning for COVID region segmentation in chest CT using multi-national data from China, Italy, Japan

Unpaired Cross-Modal Interaction Learning for COVID-19 Segmentation on Limited CT Images

Exploiting Shared Knowledge From Non-COVID Lesions for Annotation-Efficient COVID-19 CT Lung Infection Segmentation

SSA-Net: Spatial Self-Attention Network for COVID-19 Pneumonia Infection Segmentation with Semi-supervised Few-shot Learning

Unsupervised domain adaptation based COVID-19 CT infection segmentation network

A Weakly Supervised Region-Based Active Learning Method for COVID-19 Segmentation in CT Images

Active contour regularized semi-supervised learning for COVID-19 CT infection segmentation with limited annotations

Weakly Supervised Segmentation of COVID19 Infection with Scribble Annotation on CT Images

An efficient deep neural network framework for COVID-19 lung infection segmentation

A Noise-Robust Framework for Automatic Segmentation of COVID-19 Pneumonia Lesions From CT Images

Dual Multiscale Mean Teacher Network for Semi-Supervised Infection Segmentation in Chest CT Volume for COVID-19

Synergistic learning of lung lobe segmentation and hierarchical multi-instance classification for automated severity assessment of COVID-19 in CT images

Novel coronavirus pneumonia lesion segmentation in CT images

A Teacher-Student Framework with Fourier Augmentation for COVID-19 Infection Segmentation in CT Images

COVID-19 lung infection segmentation with a novel two-stage cross-domain transfer learning framework

A Deep Ensemble Learning Approach to Lung CT Segmentation for COVID-19 Severity Assessment