论文标题
ARPM-NET:一种基于CNN的新型对抗方法,具有Markov随机场增强的前列腺和器官,处于骨盆CT图像中的风险分割
ARPM-net: A novel CNN-based adversarial method with Markov Random Field enhancement for prostate and organs at risk segmentation in pelvic CT images
论文作者
论文摘要
目的:研究是开发一种新型的基于CNN的对抗深度学习方法,以改善和加快CT图像的多器官语义分割,并在骨盆CT图像上生成准确的轮廓。方法:回顾性地选择了120例完整前列腺癌患者的CT和结构数据集,并进行了10倍的交叉验证。拟议的对抗性多分离多尺度合并马尔可夫随机场(MRF)增强网络(ARPM-NET)实现了对抗性训练方案。共同训练了分割网络和歧视网络,仅使用分割网络进行预测。分割网络将新设计的MRF块集成到多分离U-NET的变体中。鉴别器将原始CT的乘积和预测/地面真相视为输入,并将输入分为假/真实。分割网络和歧视网络可以整体共同训练,也可以在对分割网络进行精心培训之后使用歧视器进行微调。引入了多尺度的合并层,以保留在合并过程中使用较少的记忆力的空间分辨率,与极端的卷积层相比。提出了一种自适应损失功能,以增强小型或低对比器官的训练。使用骰子相似性系数(DSC),平均Hausdorff距离(AHD),平均表面Hausdorff距离(ASHD)和相对体积差(VD)测量建模轮廓的精度,使用临床轮廓作为对地面真实真实真实真真度的参考。将提出的ARPM-NET方法与几种先进的深度学习方法进行了比较。
Purpose: The research is to develop a novel CNN-based adversarial deep learning method to improve and expedite the multi-organ semantic segmentation of CT images, and to generate accurate contours on pelvic CT images. Methods: Planning CT and structure datasets for 120 patients with intact prostate cancer were retrospectively selected and divided for 10-fold cross-validation. The proposed adversarial multi-residual multi-scale pooling Markov Random Field (MRF) enhanced network (ARPM-net) implements an adversarial training scheme. A segmentation network and a discriminator network were trained jointly, and only the segmentation network was used for prediction. The segmentation network integrates a newly designed MRF block into a variation of multi-residual U-net. The discriminator takes the product of the original CT and the prediction/ground-truth as input and classifies the input into fake/real. The segmentation network and discriminator network can be trained jointly as a whole, or the discriminator can be used for fine-tuning after the segmentation network is coarsely trained. Multi-scale pooling layers were introduced to preserve spatial resolution during pooling using less memory compared to atrous convolution layers. An adaptive loss function was proposed to enhance the training on small or low contrast organs. The accuracy of modeled contours was measured with the Dice similarity coefficient (DSC), Average Hausdorff Distance (AHD), Average Surface Hausdorff Distance (ASHD), and relative Volume Difference (VD) using clinical contours as references to the ground-truth. The proposed ARPM-net method was compared to several stateof-the-art deep learning methods.