《计算机应用》唯一官方网站

• •    下一篇

对抗样本嵌入注意力U型网络的3D医图分割

许志雄1,李波1,边小勇2   

  1. 1. 武汉科技大学
    2. 武汉科技大学 计算机科学与技术学院,武汉 430065
  • 收稿日期:2024-08-12 修回日期:2024-09-25 发布日期:2024-11-07 出版日期:2024-11-07
  • 通讯作者: 李波
  • 基金资助:
    国家自然科学基金项目

Adversarial sample embedded attention U-Net for 3D medical image segmentation

  • Received:2024-08-12 Revised:2024-09-25 Online:2024-11-07 Published:2024-11-07
  • Contact: LI Bo

摘要: CT图像和核磁共振图像(MRI)广泛应用于医学图像深度分割,但传统分割方法到肿瘤边界模糊及其结构复杂的影响,忽略了对抗样本对分割模型的区分能力,使得最新的分割效果依然面临挑战。针对以上问题,提出对抗样本嵌入注意力U型网络学习的3D医学图像分割模型。对抗样本嵌入的注意力U型网络通过样本变换构建对抗样本,提取医学图像的肿瘤特征信息;引入低维度特征筛选和高维度特征融合模块,提纯肿瘤可区分特征表示能力;使用基于交叉熵、Dice损失和对比损失的组合损失函数训练整个网络,得到富于判别的分割模型。所提方法在神经鞘膜瘤(NST)、自动心脏诊断挑战(ACDC)上的Dice相似性系数分别达到88.14%和91.75%,与nnU-Net方法相比,分别提高了1.26和2.48个百分点。实验结果表明,所提方法有效提高了在肿瘤边界模糊时的3D医学图像分割性能。

关键词: 医图分割, 深度学习, 注意力U型网络, 对比学习, 特征融合

Abstract: CT Image and Magnetic Resonance Image (MRI) were widely used in medical image depth segmentation. However, traditional segmentation methods were affected by tumor boundary blurring and structural complexity, and the impact of adversarial samples on the differentiation ability of the segmentation model was ignored, leading to challenges in achieving optimal segmentation results. To address these issues, a 3D medical image segmentation model with adversarial sample-embedded attention U-network learning was proposed. Adversarial samples were constructed through sample transformation, and tumor feature information from medical images was extracted. Low-dimensional feature screening and high-dimensional feature fusion modules were introduced to purify the tumor-distinguishable feature representation. The entire network was trained using a combined loss function based on cross-entropy, Dice loss, and comparison loss to obtain a segmentation model rich in discriminative features. The Dice similarity coefficients for the method on Nerve Sheath Tumor (NST) and Automated Cardiac Diagnosis Challenge (ACDC) reached 88.14% and 91.75%, respectively, which represented improvements of 1.26 percentage points and 2.48 percentage points compared to the nnU-Net method. Experimental results showed that the proposed method effectively improved the performance of 3D medical image segmentation when the tumor boundary was blurred.

Key words: Medical image segmentation, deep learning, attention U-Net, contrastive learning, feature fusion

中图分类号: