Three-dimensional convolutional neural network for prostate MRI segmentation and comparison of prostate volume measurements by use of artificial neural network and ellipsoid formula

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 (V_GT), software-derived volume (V_SW), mean of V_GT and V_SW (V_AV), and automatica lly generated volume from t he 3D CN N (V_NET). These values were compared with the volume calculated with the ellipsoid formula (V_EL). 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 V_NET and V_AV (p = 0.689) in the test sets of the entire prostate, whereas a significant difference was found between V_EL and V_AV (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 V_GT and V_NET (2.5) was smaller than that between V_GT and V_EL (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.