arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2206.03413physics.med-phcs.CV
收藏

基于深度学习的可变形图像配准辅助的锥束直接分割基于CT的自适应放射治疗自动分割

Deep Learning based Direct Segmentation Assisted by Deformable Image Registration for Cone-Beam CT based Auto-Segmentation for Adaptive Radiotherapy

University of Texas Southwestern Medical Center(德克萨斯大学西南医学中心)

Xiao Liang, Howard Morgan, Ti Bai, Michael Dohopolski, Dan Nguyen, Steve Jiang

摘要:

基于锥形束CT(CBCT)的在线自适应放射治疗需要精确的自动分割,以减少医生编辑轮廓的时间成本。然而,基于深度学习(DL)的CBCT图像直接分割是一项具有挑战性的任务,主要是由于图像质量差和缺乏标记良好的大型训练数据集。可变形图像配准(Deformable Image Registration,简称DIF)通常用于将同一患者的计划CT(planning CT,简称pCT)上的手动轮廓传播到CBCT。在这项工作中,我们将在联合国的协助下解决上述问题。我们的方法包括三个主要组成部分。首先,我们使用变形的pCT轮廓从pCT和CBCT之间的多个拼接方法作为伪标签的DL为基础的直接分割模型的初始训练。其次,我们使用变形的pCT轮廓从另一个pCT算法作为影响体积定义的DL为基础的直接分割的兴趣区域。第三,使用较小的真实标签集进一步微调初始训练的DL模型。我们发现,与基于DIR的分割相比,在使用伪标签训练并且没有影响者体积的CBCT上基于DL的直接分割显示出较差的性能。然而,在直接分割网络中添加变形的pCT轮廓作为影响体积显著提高了分割性能,达到了基于DIR的分割的准确度水平。具有影响者数量的DL模型可以通过使用较小的真实标签集进行微调来进一步改进。实验表明,19个结构中有7个结构与基于DIR的分割相比具有至少0.2的Dice相似系数增加。基于DL的直接CBCT分割模型可以通过使用变形的pCT轮廓作为用于初始训练的伪标签和影响体积,并且通过使用较小的真实标签集合进行模型微调来改进,以优于基于DIR的分割模型。

英文摘要:

Cone-beam CT (CBCT)-based online adaptive radiotherapy calls for accurate auto-segmentation to reduce the time cost for physicians to edit contours. However, deep learning (DL)-based direct segmentation of CBCT images is a challenging task, mainly due to the poor image quality and lack of well-labelled large training datasets. Deformable image registration (DIR) is often used to propagate the manual contours on the planning CT (pCT) of the same patient to CBCT. In this work, we undertake solving the problems mentioned above with the assistance of DIR. Our method consists of three main components. First, we use deformed pCT contours derived from multiple DIR methods between pCT and CBCT as pseudo labels for initial training of the DL-based direct segmentation model. Second, we use deformed pCT contours from another DIR algorithm as influencer volumes to define the region of interest for DL-based direct segmentation. Third, the initially trained DL model is further fine-tuned using a smaller set of true labels. We found that DL-based direct segmentation on CBCT trained with pseudo labels and without influencer volumes shows poor performance compared to DIR-based segmentation. However, adding deformed pCT contours as influencer volumes in the direct segmentation network dramatically improves segmentation performance, reaching the accuracy level of DIR-based segmentation. The DL model with influencer volumes can be further improved through fine-tuning using a smaller set of true labels. Experiments showed that 7 out of 19 structures have an at least 0.2 Dice similarity coefficient increase compared to DIR-based segmentation. A DL-based direct CBCT segmentation model can be improved to outperform DIR-based segmentation models by using deformed pCT contours as pseudo labels and influencer volumes for initial training, and by using a smaller set of true labels for model fine tuning.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑