Automatic Prostate Segmentation on 3D MRI Scans Using Convolutional Neural Networks with Residual Connections and Superpixels

Automatic and reliable prostate segmentation is an essential prerequisite for assisting the diagnosis and treatment, such as guiding biopsy procedure and radiation therapy. Notwithstanding, automatic segmentation is challenging due to the lack of clear prostate boundaries owing to similar appearance of prostate and surrounding tissues and the wide variation in size and shape among different patients. Therefore, this paper proposes an automatic method for prostate segmentation on 3D MRI scans using a content-sensitive superpixels technique, deep convolutional neural networks with residual connections (ResCNN), and the particle swarm optimization (PSO) algorithm. The proposed method has been evaluated on the Prostate 3T and PROMISE12 databases, presenting a Dice similarity coefficient of 86.68 %, Jaccard index of 76.58%, relative volume difference of 3.92%, volumetric similarity of 96.69 %, sensitivity of 88.36 %, specificity of 93.43%, accuracy of 91.97% and an area under ROC curve of 90.90 %. Experimental results demonstrate the high performance-potential of the proposed method comparable to those previously published.

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

[2]  Christopher J Kane,et al.  Prostate Cancer, Version 1.2016. , 2016, Journal of the National Comprehensive Cancer Network : JNCCN.

[3]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[4]  Russell C. Eberhart,et al.  A new optimizer using particle swarm theory , 1995, MHS'95. Proceedings of the Sixth International Symposium on Micro Machine and Human Science.

[5]  Ang Li,et al.  Automatic prostate segmentation on MR images with deep network and graph model , 2016, 2016 38th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[6]  Allan Hanbury,et al.  Metrics for evaluating 3D medical image segmentation: analysis, selection, and tool , 2015, BMC Medical Imaging.

[7]  Anselmo Cardoso de Paiva,et al.  Convolutional neural network-based PSO for lung nodule false positive reduction on CT images , 2018, Comput. Methods Programs Biomed..

[8]  J. C. Arroyo Kuribreña,et al.  Cáncer de próstata , 2004 .

[9]  Anant Madabhushi,et al.  A Deep Convolutional Neural Network for segmenting and classifying epithelial and stromal regions in histopathological images , 2016, Neurocomputing.

[10]  Christopher Joseph Pal,et al.  Brain tumor segmentation with Deep Neural Networks , 2015, Medical Image Anal..

[11]  Anselmo Cardoso de Paiva,et al.  Lung nodules diagnosis based on evolutionary convolutional neural network , 2017, Multimedia Tools and Applications.

[12]  Hao Chen,et al.  Volumetric ConvNets with Mixed Residual Connections for Automated Prostate Segmentation from 3D MR Images , 2017, AAAI.

[13]  Nathan Lay,et al.  Automatic magnetic resonance prostate segmentation by deep learning with holistically nested networks , 2017, Journal of medical imaging.

[14]  Yong-Jin Liu,et al.  Intrinsic Manifold SLIC: A Simple and Efficient Method for Computing Content-Sensitive Superpixels , 2018, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[15]  David Dagan Feng,et al.  Atlas registration and ensemble deep convolutional neural network-based prostate segmentation using magnetic resonance imaging , 2018, Neurocomputing.

[16]  David Dagan Feng,et al.  Prostate segmentation in MR images using ensemble deep convolutional neural networks , 2017, 2017 IEEE 14th International Symposium on Biomedical Imaging (ISBI 2017).

[17]  Yann LeCun,et al.  Convolutional networks and applications in vision , 2010, Proceedings of 2010 IEEE International Symposium on Circuits and Systems.

[18]  Desire Sidibé,et al.  A survey of prostate segmentation methodologies in ultrasound, magnetic resonance and computed tomography images , 2012, Comput. Methods Programs Biomed..

[19]  A. Jemal,et al.  Cancer statistics, 2019 , 2019, CA: a cancer journal for clinicians.

[20]  R. Eberhart,et al.  Comparing inertia weights and constriction factors in particle swarm optimization , 2000, Proceedings of the 2000 Congress on Evolutionary Computation. CEC00 (Cat. No.00TH8512).

[21]  Florian Jung,et al.  Evaluation of prostate segmentation algorithms for MRI: The PROMISE12 challenge , 2014, Medical Image Anal..

[22]  Yaozong Gao,et al.  Deformable MR Prostate Segmentation via Deep Feature Learning and Sparse Patch Matching , 2016, IEEE Transactions on Medical Imaging.

[23]  Tian Liu,et al.  3D transrectal ultrasound (TRUS) prostate segmentation based on optimal feature learning framework , 2016, SPIE Medical Imaging.