Automatic magnetic resonance prostate segmentation by deep learning with holistically nested networks
Journal of Medical Imaging
deep learning; holistically nested edge detection; holistically nested networks; magnetic resonance images; prostate; segmentation
© 2017 Society of Photo-Optical Instrumentation Engineers (SPIE). Accurate automatic segmentation of the prostate in magnetic resonance images (MRI) is a challenging task due to the high variability of prostate anatomic structure. Artifacts such as noise and similar signal intensity of tissues around the prostate boundary inhibit traditional segmentation methods from achieving high accuracy. We investigate both patch-based and holistic (image-to-image) deep-learning methods for segmentation of the prostate. First, we introduce a patch-based convolutional network that aims to refine the prostate contour which provides an initialization. Second, we propose a method for end-to-end prostate segmentation by integrating holistically nested edge detection with fully convolutional networks. Holistically nested networks (HNN) automatically learn a hierarchical representation that can improve prostate boundary detection. Quantitative evaluation is performed on the MRI scans of 250 patients in fivefold cross-validation. The proposed enhanced HNN model achieves a mean ± standard deviation. A Dice similarity coefficient (DSC) of 89.77%±3.29% and a mean Jaccard similarity coefficient (IoU) of 81.59%±5.18% are used to calculate without trimming any end slices. The proposed holistic model significantly (p<0.001) outperforms a patch-based AlexNet model by 9% in DSC and 13% in IoU. Overall, the method achieves state-of-the-art performance as compared with other MRI prostate segmentation methods in the literature.
Cheng, R., Roth, H., Lay, N., Lu, L., Turkbey, B., Gandler, W., McCreedy, E., Pohida, T., Pinto, P., Choyke, P., McAuliffe, M., & Summers, R. (2017). Automatic magnetic resonance prostate segmentation by deep learning with holistically nested networks. Journal of Medical Imaging, 4 (4). http://dx.doi.org/10.1117/1.JMI.4.4.041302