Superpixel-based deep convolutional neural networks and active contour model for automatic prostate segmentation on 3D MRI scans

Med Biol Eng Comput. 2020 Sep;58(9):1947-1964. doi: 10.1007/s11517-020-02199-5. Epub 2020 Jun 21.

Abstract

Automatic and reliable prostate segmentation is an essential prerequisite for assisting the diagnosis and treatment, such as guiding biopsy procedure and radiation therapy. Nonetheless, automatic segmentation is challenging due to the lack of clear prostate boundaries owing to the similar appearance of prostate and surrounding tissues and the wide variation in size and shape among different patients ascribed to pathological changes or different resolutions of images. In this regard, the state-of-the-art includes methods based on a probabilistic atlas, active contour models, and deep learning techniques. However, these techniques have limitations that need to be addressed, such as MRI scans with the same spatial resolution, initialization of the prostate region with well-defined contours and a set of hyperparameters of deep learning techniques determined manually, respectively. Therefore, this paper proposes an automatic and novel coarse-to-fine segmentation method for prostate 3D MRI scans. The coarse segmentation step combines local texture and spatial information using the Intrinsic Manifold Simple Linear Iterative Clustering algorithm and probabilistic atlas in a deep convolutional neural networks model jointly with the particle swarm optimization algorithm to classify prostate and non-prostate tissues. Then, the fine segmentation uses the 3D Chan-Vese active contour model to obtain the final prostate surface. The proposed method has been evaluated on the Prostate 3T and PROMISE12 databases presenting a dice similarity coefficient of 84.86%, relative volume difference of 14.53%, sensitivity of 90.73%, specificity of 99.46%, and accuracy of 99.11%. Experimental results demonstrate the high performance potential of the proposed method compared to those previously published.

Keywords: Deep convolutional neural networks; Medical images; Particle swarm optimization; Probabilistic atlas; Prostate segmentation; Superpixels.

Publication types

  • Evaluation Study

MeSH terms

  • Algorithms
  • Databases, Factual
  • Deep Learning
  • Humans
  • Image Interpretation, Computer-Assisted / statistics & numerical data*
  • Imaging, Three-Dimensional / statistics & numerical data*
  • Latent Class Analysis
  • Magnetic Resonance Imaging / statistics & numerical data*
  • Male
  • Models, Statistical
  • Neural Networks, Computer*
  • Prostatic Neoplasms / diagnostic imaging*