Three-Dimensional Convolutional Neural Network for Prostate MRI Segmentation and Comparison of Prostate Volume Measurements by Use of Artificial Neural Network and Ellipsoid Formula

AJR Am J Roentgenol. 2020 Jun;214(6):1229-1238. doi: 10.2214/AJR.19.22254. Epub 2020 Mar 24.

Abstract

OBJECTIVE. The purposes of this study were to assess the performance of a 3D convolutional neural network (CNN) for automatic segmentation of prostates on MR images and to compare the volume estimates from the 3D CNN with those of the ellipsoid formula. MATERIALS AND METHODS. The study included 330 MR image sets that were divided into 260 training sets and 70 test sets for automated segmentation of the entire prostate. Among these, 162 training sets and 50 test sets were used for transition zone segmentation. Assisted by manual segmentation by two radiologists, the following values were obtained: estimates of ground-truth volume (VGT), software-derived volume (VSW), mean of VGT and VSW (VAV), and automatically generated volume from the 3D CNN (VNET). These values were compared with the volume calculated with the ellipsoid formula (VEL). RESULTS. The Dice similarity coefficient for the entire prostate was 87.12% and for the transition zone was 76.48%. There was no significant difference between VNET and VAV (p = 0.689) in the test sets of the entire prostate, whereas a significant difference was found between VEL and VAV (p < 0.001). No significant difference was found among the volume estimates in the test sets of the transition zone. Overall intraclass correlation coefficients between the volume estimates were excellent (0.887-0.995). In the test sets of entire prostate, the mean error between VGT and VNET (2.5) was smaller than that between VGT and VEL (3.3). CONCLUSION. The fully automated network studied provides reliable volume estimates of the entire prostate compared with those obtained with the ellipsoid formula. Fast and accurate volume measurement by use of the 3D CNN may help clinicians evaluate prostate disease.

Keywords: MRI; convolutional neural network; ellipsoid formula; prostate segmentation; prostate volume.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Humans
  • Image Interpretation, Computer-Assisted / methods*
  • Imaging, Three-Dimensional / methods*
  • Magnetic Resonance Imaging / methods*
  • Male
  • Neural Networks, Computer*
  • Prostatic Neoplasms / diagnostic imaging*
  • Prostatic Neoplasms / pathology
  • Retrospective Studies