DIAGNijmegen / AbdomenMRUS-prostate-segmentation

Grand Challenge wrapper for whole-gland prostate segmentation with nnUNet
Apache License 2.0
14 stars 5 forks source link

Prostate Segmentation in MRI

Managed By

Diagnostic Image Analysis Group, Radboud University Medical Center, Nijmegen, The Netherlands

Contact Information

Inference

This algorithm is hosted on Grand-Challenge.com. Alternatively, inference can be performed locally.

Summary

This algorithm segments the whole prostate gland in biparametric MRI (bpMRI). Development of this model was geared toward robust prostate segmentation, at the expense of fine-grained zonal segmentation. This algorithm was used to provide prostate segmentations for the PI-CAI challenge.

Mechanism

This algorithm is a deep learning-based model, which ensembles five independent nnU-Net models (using 5-fold cross-validation). To prioritize robust segmentation, we trained these models with Cross-Entropy + Focal loss. We trained these models with a total of 438 prostate biparametric MRI (bpMRI) scans paired with a manual prostate segmentation. These scans were sourced from two independent hospitals: 299 cases from Radboudumc (of which 248 part of ProstateX) and 139 cases from Prostate158.

We ensured there is no patient overlap between this algorithm's training dataset and the PI-CAI Hidden Validation and Tuning Cohort or Hidden Testing Cohort.

Validation and Performance

This algorithm is evaluated using 5-fold cross-validation using the dataset described in Mechanism. Segmentation performance and its standard deviation across cases are provided below.

Dice similarity coefficient Jaccard index
0.8968 ± 0.0547 0.8169 ± 0.0820

Training

Training steps are provided in here.

Uses and Directions

Warnings