Alom Md Zahangir, Yakopcic Chris, Hasan Mahmudul, Taha Tarek M, Asari Vijayan K
University of Dayton, Department of Electrical and Computer Engineering, Dayton, Ohio, United States.
Comcast Labs, Washington, DC, United States.
J Med Imaging (Bellingham). 2019 Jan;6(1):014006. doi: 10.1117/1.JMI.6.1.014006. Epub 2019 Mar 27.
Deep learning (DL)-based semantic segmentation methods have been providing state-of-the-art performance in the past few years. More specifically, these techniques have been successfully applied in medical image classification, segmentation, and detection tasks. One DL technique, U-Net, has become one of the most popular for these applications. We propose a recurrent U-Net model and a recurrent residual U-Net model, which are named RU-Net and R2U-Net, respectively. The proposed models utilize the power of U-Net, residual networks, and recurrent convolutional neural networks. There are several advantages to using these proposed architectures for segmentation tasks. First, a residual unit helps when training deep architectures. Second, feature accumulation with recurrent residual convolutional layers ensures better feature representation for segmentation tasks. Third, it allows us to design better U-Net architectures with the same number of network parameters with better performance for medical image segmentation. The proposed models are tested on three benchmark datasets, such as blood vessel segmentation in retinal images, skin cancer segmentation, and lung lesion segmentation. The experimental results show superior performance on segmentation tasks compared to equivalent models, including a variant of a fully connected convolutional neural network called SegNet, U-Net, and residual U-Net.
在过去几年中,基于深度学习(DL)的语义分割方法一直保持着领先的性能。具体而言,这些技术已成功应用于医学图像分类、分割和检测任务。一种深度学习技术——U-Net,已成为这些应用中最受欢迎的技术之一。我们提出了一种循环U-Net模型和一种循环残差U-Net模型,分别命名为RU-Net和R2U-Net。所提出的模型利用了U-Net、残差网络和循环卷积神经网络的优势。使用这些提出的架构进行分割任务有几个优点。首先,残差单元有助于深度架构的训练。其次,循环残差卷积层的特征积累确保了分割任务的更好特征表示。第三,它使我们能够设计出具有相同网络参数数量的更好的U-Net架构,以实现更好的医学图像分割性能。所提出的模型在三个基准数据集上进行了测试,如视网膜图像中的血管分割、皮肤癌分割和肺部病变分割。实验结果表明,与等效模型相比,所提出的模型在分割任务上具有卓越的性能,这些等效模型包括一种名为SegNet的全连接卷积神经网络变体、U-Net和残差U-Net。