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

Abstract

Infection segmentation on chest CT plays an important role in the quantitative analysis of COVID-19. Developing automatic segmentation tools in a short period with limited labelled images has become an urgent need. Pseudo label-based semi-supervised method is a promising way to leverage unlabelled data to improve segmentation performance. Existing methods usually obtain pseudo labels by first training a network with limited labelled images and then inferring unlabelled images. However, these methods may generate obviously inaccurate labels and degrade the subsequent training process. To address these challenges, in this paper, an active contour regularized semi-supervised learning framework was proposed to automatically segment infections with few labelled images. The active contour regularization was realized by the region-scalable fitting (RSF) model which is embedded to the loss function of the network to regularize and refine the pseudo labels of the unlabelled images. We further designed a splitting method to separately optimize the RSF regularization term and the segmentation loss term with iterative convolution-thresholding method and stochastic gradient descent, respectively, which enable fast optimization of each term. Furthermore, we built a statistical atlas to show the infection spatial distribution. Extensive experiments on a small public dataset and a large scale dataset showed that the proposed method outperforms state-of-the-art methods with up to 5% in dice similarity coefficient and normalized surface dice, 10% in relative absolute volume difference and 8 mm in 95% Hausdorff distance. Moreover, we observed that the infections tend to occur at the dorsal subpleural lung and posterior basal segments that are not mentioned in current radiology reports and are meaningful to advance our understanding of COVID-19.

Depth-wise dense neural network for automatic COVID19 infection detection and diagnosis

“…Afshar et al ( 2020 ) applied capsule network for diagnosis of COVID patient in X-ray images. Ma et al ( 2020 ) presented an active framework contour regularized semi-supervised learning for segmentation of COVID infection on small labelled dataset. Region-scalable fitting model is embedded into the loss function for active contour regularization and for refinement of pseudo labels of unlabeled data.…”

Multilevel depth-wise context attention network with atrous mechanism for segmentation of COVID19 affected regions

“…Afshar et al applied capsule network for diagnosis of COVID patient in X-ray images [ 1 ]. Ma et al presented a framework active contour regularized semi-supervised learning for segmentation of COVID infection on small labeled dataset [ 16 ]. Region-scalable fitting model is embedded into the loss function for active contour regularization and for refinement of pseudo labels of unlabeled data.…”

Semi-supervised Learning for COVID-19 Image Classification via ResNet

“…The rationale of these semi-supervised learning models is to enrich the supervision signals by exploiting the knowledge learned on unlabeled data [41], or regularize the network by enforcing smooth and consistent classification boundaries [40]. Regarding COVID-19 research such as COVID-19 image classification and image segmentation, semi-supervised learning is employed to resolve the lacking of labeled data [42][43][44][45][46][47]. However, for COVID-19 image classification, these studies [42][43][44] have not comprehensively examined the model performance on a large-scale of X-ray image dataset such as COVIDx [20] by comparing with the state-of-the-art, especially for the case of very few labeled data such as less than 10% labeled data.…”