Abstract:Prostate gland segmentation is the primary step to estimate gland volume, which aids in the prostate disease management. In this study, we present a 2D-3D convolutional neural network (CNN) ensemble that automatically segments the whole prostate gland along with the peripheral zone (PZ) (PPZ-SegNet) using a T2-weighted sequence (T2W) of Magnetic Resonance Imaging (MRI). The study used 4 different public data sets organized as Train #1 and Test #1 (independently derived from the same cohort), Test #2, Test #3 and Test #4. The prostate gland and the peripheral zone (PZ) anatomy were manually delineated with consensus read by a radiologist, except for Test #4 cohorts that had pre-marked glandular anatomy. A Bayesian hyperparameter optimization method was applied to construct the network model (PPZ-SegNet) with a training cohort (Train #1, n = 150) using a five-fold cross validation. The model evaluation was performed on an independent cohort of 283 T2W MRI prostate cases (Test #1 to #4) without any additional tuning. The data cohorts were derived from The Cancer Imaging Archives (TCIA): PROSTATEx Challenge, Prostatectomy, Repeatability studies and PROMISE12-Challenge. The segmentation performance was evaluated by computing the Dice similarity coefficient and Hausdorff distance between the estimated-deep-network identified regions and the radiologist-drawn annotations. The deep network architecture was able to segment the prostate gland anatomy with an average Dice score of 0.86 in Test #1 (n = 192), 0.79 in Test #2 (n = 26), 0.81 in Test #3 (n = 15), and 0.62 in Test #4 (n = 50). We also found the Dice coefficient improved with larger prostate volumes in 3 of the 4 test cohorts. The variation of the Dice scores from different cohorts of test images suggests the necessity of more diverse models that are inclusive of dependencies such as the gland sizes and others, which will enable us to develop a universal network for prostate and PZ segmentation. Our training and evaluation code can be accessed through the link: https://github.com/mariabaldeon/PPZ-SegNet.git.

HD-Net: Hybrid Discriminative Network for Prostate Segmentation in MR Images

3D Multi-Scale Discriminative Network with Multi-Directional Edge Loss for Prostate Zonal Segmentation in Bi-Parametric MR Images

Learning Multi-Scale Synergic Discriminative Features for Prostate Image Segmentation.

Multi-scale discriminative network for prostate cancer lesion segmentation in multiparametric MR images

3D APA-Net: 3D Adversarial Pyramid Anisotropic Convolutional Network for Prostate Segmentation in MR Images

Encoder-decoder with dense dilated spatial pyramid pooling for prostate MR images segmentation.

ProNet—Professional Prostate Segmentation Network of Transrectal Ultrasound Images Based on Deep Convolutional Neural Networks

Volumetric Densely Dilated Spatial Pooling ConNets for Prostate Segmentation.

ProSegNet: A New Network of Prostate Segmentation Based on MR Images

A Novel Residual and Gated Network for Prostate Segmentation on MR Images.

A Deep Learning Health Data Analysis Approach: Automatic 3d Prostate Mr Segmentation With Densely-Connected Volumetric Convnets

Automatic Segmentation of Prostate Magnetic Resonance Imaging Using Generative Adversarial Networks

Densely Dilated Spatial Pooling Convolutional Network Using Benign Loss Functions for Imbalanced Volumetric Prostate Segmentation

Prostate Lesion Segmentation Based on a 3D End-to-end Convolution Neural Network with Deep Multi-Scale Attention.

3D Global Convolutional Adversarial Network\\ for Prostate MR Volume Segmentation

Automatic Segmentation of Prostate MRI Based on 3D Pyramid Pooling Unet

Automatic Segmentation of the Prostate on MR Images Based on Anatomy and Deep Learning.

A multi-object deep neural network architecture to detect prostate anatomy in T2-weighted MRI: Performance evaluation

Deep Convolutional Neural Network for Prostate Mr Segmentation

Prostate Segmentation in MR Images Using Ensemble Deep Convolutional Neural Networks