arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.14255eess.IVcs.CV

AttResDU-Net:使用基于注意力的残差双U-Net进行医学图像分割

AttResDU-Net: Medical Image Segmentation Using Attention-based Residual Double U-Net

Islamic University of Technology(伊斯兰理工大学)

Akib Mohammed Khan, Alif Ashrafee, Fahim Shahriar Khan, Md. Bakhtiar Hasan, Md. Hasanul Kabir

AI总结:

本文提出了一种基于注意力的残差双U-Net架构,通过引入注意力门和残差连接提升医学图像分割性能,在三个数据集上取得高分,证明其作为可靠自动分割方法的实用性。

摘要:

手动检查结肠镜检查中的息肉以诊断结直肠癌,或对皮肤病变进行活检以诊断皮肤癌,是耗时、费力且复杂的过程。自动医学图像分割旨在加速这一诊断过程。然而,由于目标的外观和尺寸存在显著差异且没有明显的边界,因此存在许多挑战。本文提出了一种基于注意力的残差 Double U-Net 架构(AttResDU-Net),对现有的医学图像分割网络进行了改进。受 Double U-Net 启发,该架构在跳跃连接上加入了注意力门,并在卷积块中加入了残差连接。注意力门通过抑制下采样路径中不相关的特征表示,使模型能够保留更多相关的空间信息,从而让模型学会关注不同形状和大小的目标区域。此外,残差连接通过确保更好的梯度流动,有助于训练更深的模型。我们在三个数据集上进行了实验:CVC Clinic-DB、ISIC 2018 和 2018 Data Science Bowl 数据集,分别取得了 94.35%、91.68% 和 92.45% 的 Dice Coefficient 得分。我们的结果表明,AttResDU-Net 可以在实践中作为一种可靠的自动医学图像分割方法。

英文摘要:

Manually inspecting polyps from a colonoscopy for colorectal cancer or performing a biopsy on skin lesions for skin cancer are time-consuming, laborious, and complex procedures. Automatic medical image segmentation aims to expedite this diagnosis process. However, numerous challenges exist due to significant variations in the appearance and sizes of objects with no distinct boundaries. This paper proposes an attention-based residual Double U-Net architecture (AttResDU-Net) that improves on the existing medical image segmentation networks. Inspired by the Double U-Net, this architecture incorporates attention gates on the skip connections and residual connections in the convolutional blocks. The attention gates allow the model to retain more relevant spatial information by suppressing irrelevant feature representation from the down-sampling path for which the model learns to focus on target regions of varying shapes and sizes. Moreover, the residual connections help to train deeper models by ensuring better gradient flow. We conducted experiments on three datasets: CVC Clinic-DB, ISIC 2018, and the 2018 Data Science Bowl datasets and achieved Dice Coefficient scores of 94.35%, 91.68% and 92.45% respectively. Our results suggest that AttResDU-Net can be facilitated as a reliable method for automatic medical image segmentation in practice.

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

↑