arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.04847eess.IVcs.CV
收藏

编织注意力U网:一种新的混合CNN和基于注意力的头颈CT图像危险器官分割方法

Weaving Attention U-net: A Novel Hybrid CNN and Attention-based Method for Organs-at-risk Segmentation in Head and Neck CT Images

Washington University in St. Louis(圣路易斯华盛顿大学)

Zhuangzhuang Zhang, Tianyu Zhao, Hiram Gay, Weixiong Zhang, Baozhou Sun

摘要:

在放射治疗计划中,手动勾画轮廓是劳动密集型和耗时的。准确和强大的自动分割模型提高了效率和治疗效果。我们的目标是开发一种新的混合深度学习方法,将卷积神经网络(CNN)和自注意力机制相结合,用于头颈部计算机断层扫描(CT)图像的快速准确的多器官分割。回顾性收集并使用115例患者的头颈部CT图像和手动轮廓。我们将训练/验证/测试比率设置为81/9/25,并使用10倍交叉验证策略来选择最佳模型参数。所提出的混合模型为每种情况共分割了10个风险器官(OAR)。该模型的性能通过三个指标进行评估,即,Dice相似系数(DSC)、Hausdorff距离95%(HD 95)和平均表面距离(MSD)。我们还测试了该模型在头颈部2015挑战数据集上的性能,并将其与几种最先进的自动分割算法进行了比较。所提出的方法生成的轮廓,非常类似于地面实况的10个OAR。我们的结果表明,新的编织注意力U-网络的头部和颈部CT图像的分割表现出优越或相似的性能。

英文摘要:

In radiotherapy planning, manual contouring is labor-intensive and time-consuming. Accurate and robust automated segmentation models improve the efficiency and treatment outcome. We aim to develop a novel hybrid deep learning approach, combining convolutional neural networks (CNNs) and the self-attention mechanism, for rapid and accurate multi-organ segmentation on head and neck computed tomography (CT) images. Head and neck CT images with manual contours of 115 patients were retrospectively collected and used. We set the training/validation/testing ratio to 81/9/25 and used the 10-fold cross-validation strategy to select the best model parameters. The proposed hybrid model segmented ten organs-at-risk (OARs) altogether for each case. The performance of the model was evaluated by three metrics, i.e., the Dice Similarity Coefficient (DSC), Hausdorff distance 95% (HD95), and mean surface distance (MSD). We also tested the performance of the model on the Head and Neck 2015 challenge dataset and compared it against several state-of-the-art automated segmentation algorithms. The proposed method generated contours that closely resemble the ground truth for ten OARs. Our results of the new Weaving Attention U-net demonstrate superior or similar performance on the segmentation of head and neck CT images.

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

↑