PRDNet: Medical image segmentation based on parallel residual and dilated network

Abstract This paper presents an improved semantic segmentation model for medical images named PRDNet. Common semantic segmentation algorithms only feed the features of the last layer of convolutional neural network into the decoder, which results in the position information of the first several layers of convolutional neural network is not utilized. While in our work, ResNet and dilated convolution are simultaneously used to extract multi-layer features of medical images in parallel. In the decoding stage, the multi-layer features are fused according to the structure of feature pyramid network. Moreover, several classic semantic segmentation algorithms were selected for comparison, such as UNet, Attention UNet, FPN, Deeplab v3, SENet and DANet. After a series of experiments on CHAOS and ISIC2017 datasets, the algorithm proposed by us has a 1%–4% improvement in different evaluation metrics compared with other algorithms.

[1]  Wei Sun,et al.  Efficient skin lesion segmentation using separable-Unet with stochastic weight averaging , 2019, Comput. Methods Programs Biomed..

[2]  Muhammad Haroon Yousaf,et al.  Melanoma lesion detection and segmentation using deep region based convolutional neural network and fuzzy C-means clustering , 2019, Int. J. Medical Informatics.

[3]  George Papandreou,et al.  Rethinking Atrous Convolution for Semantic Image Segmentation , 2017, ArXiv.

[4]  Bernt Schiele,et al.  Parameter-Free Spatial Attention Network for Person Re-Identification , 2018, ArXiv.

[5]  Xiaochun Cao,et al.  Survey of recent progress in semantic image segmentation with CNNs , 2017, Science China Information Sciences.

[6]  Trevor Darrell,et al.  Fully Convolutional Networks for Semantic Segmentation , 2017, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[7]  Jie Yu,et al.  DSM: A Deep Supervised Multi-Scale Network Learning for Skin Cancer Segmentation , 2019, IEEE Access.

[8]  Yong Xia,et al.  D-UNet: A Dimension-Fusion U Shape Network for Chronic Stroke Lesion Segmentation , 2019, IEEE/ACM Transactions on Computational Biology and Bioinformatics.

[9]  Iasonas Kokkinos,et al.  DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs , 2016, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[10]  Pheng-Ann Heng,et al.  Channel-Unet: A Spatial Channel-Wise Convolutional Neural Network for Liver and Tumors Segmentation , 2019, Front. Genet..

[11]  Wei Sun,et al.  Methods and datasets on semantic segmentation: A review , 2018, Neurocomputing.

[12]  Petia Radeva,et al.  SLSDeep: Skin Lesion Segmentation Based on Dilated Residual and Pyramid Pooling Networks , 2018, MICCAI.

[13]  Yaxing Wang,et al.  S-UNet: A Bridge-Style U-Net Framework With a Saliency Mechanism for Retinal Vessel Segmentation , 2019, IEEE Access.

[14]  Noel C. F. Codella,et al.  Skin lesion analysis toward melanoma detection: A challenge at the 2017 International symposium on biomedical imaging (ISBI), hosted by the international skin imaging collaboration (ISIC) , 2016, 2018 IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018).

[15]  Loïc Le Folgoc,et al.  Attention U-Net: Learning Where to Look for the Pancreas , 2018, ArXiv.

[16]  M. Alper Selver,et al.  Segmentation of abdominal organs from CT using a multi-level, hierarchical neural network strategy , 2014, Comput. Methods Programs Biomed..

[17]  Roberto Cipolla,et al.  SegNet: A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation , 2015, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[18]  Jerry L Prince,et al.  Current methods in medical image segmentation. , 2000, Annual review of biomedical engineering.

[19]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[20]  Sergey Ioffe,et al.  Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift , 2015, ICML.

[21]  Jose Dolz,et al.  Multi-Scale Self-Guided Attention for Medical Image Segmentation , 2021, IEEE Journal of Biomedical and Health Informatics.

[22]  Janez Demsar,et al.  Statistical Comparisons of Classifiers over Multiple Data Sets , 2006, J. Mach. Learn. Res..

[23]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.