IEEE Trans Med Imaging. 2022 Mar;41(3):531-542. doi: 10.1109/TMI.2021.3117564. Epub 2022 Mar 2.
Computed Tomography (CT) plays an important role in monitoring radiation-induced Pulmonary Fibrosis (PF), where accurate segmentation of the PF lesions is highly desired for diagnosis and treatment follow-up. However, the task is challenged by ambiguous boundary, irregular shape, various position and size of the lesions, as well as the difficulty in acquiring a large set of annotated volumetric images for training. To overcome these problems, we propose a novel convolutional neural network called PF-Net and incorporate it into a semi-supervised learning framework based on Iterative Confidence-based Refinement And Weighting of pseudo Labels (I-CRAWL). Our PF-Net combines 2D and 3D convolutions to deal with CT volumes with large inter-slice spacing, and uses multi-scale guided dense attention to segment complex PF lesions. For semi-supervised learning, our I-CRAWL employs pixel-level uncertainty-based confidence-aware refinement to improve the accuracy of pseudo labels of unannotated images, and uses image-level uncertainty for confidence-based image weighting to suppress low-quality pseudo labels in an iterative training process. Extensive experiments with CT scans of Rhesus Macaques with radiation-induced PF showed that: 1) PF-Net achieved higher segmentation accuracy than existing 2D, 3D and 2.5D neural networks, and 2) I-CRAWL outperformed state-of-the-art semi-supervised learning methods for the PF lesion segmentation task. Our method has a potential to improve the diagnosis of PF and clinical assessment of side effects of radiotherapy for lung cancers.
计算机断层扫描(CT)在监测放射性肺纤维化(PF)中起着重要作用,其中准确分割 PF 病变对于诊断和治疗随访非常重要。然而,由于病变边界模糊、形状不规则、位置和大小多样,以及难以获取大量标注的容积图像进行训练,因此该任务具有挑战性。为了解决这些问题,我们提出了一种名为 PF-Net 的新型卷积神经网络,并将其纳入基于迭代置信度的伪标签细化和加权(I-CRAWL)的半监督学习框架中。我们的 PF-Net 结合了 2D 和 3D 卷积,以处理切片间隔较大的 CT 容积,并使用多尺度引导密集注意力来分割复杂的 PF 病变。对于半监督学习,我们的 I-CRAWL 使用基于像素级不确定性的置信度感知细化来提高未标注图像的伪标签的准确性,并使用基于图像级不确定性的置信度图像加权来在迭代训练过程中抑制低质量的伪标签。对具有放射性 PF 的恒河猴的 CT 扫描进行了广泛的实验,结果表明:1)PF-Net 的分割准确性高于现有的 2D、3D 和 2.5D 神经网络,2)I-CRAWL 在 PF 病变分割任务中优于最先进的半监督学习方法。我们的方法有可能改善 PF 的诊断和对肺癌放射治疗副作用的临床评估。