分割
人工智能
计算机科学
模式识别(心理学)
杠杆(统计)
豪斯多夫距离
阈值
活动轮廓模型
正规化(语言学)
Sørensen–骰子系数
随机梯度下降算法
掷骰子
图像分割
数学
图像(数学)
人工神经网络
统计
作者
Jun Ma,Ziwei Nie,Congcong Wang,Guang‐Hui Dong,Qiongjie Zhu,Jian He,Luying Gui,Xiaoping Yang
标识
DOI:10.1088/1361-6560/abc04e
摘要
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.
科研通智能强力驱动
Strongly Powered by AbleSci AI