cs.CVApr 2, 2026

Automated Prostate Gland Segmentation in MRI Using nnU-Net

Authors: Pablo Rodriguez-Belenguer, Gloria Ribas, Javier Aquerreta Escribano, Rafael Moreno-Calatayud, Leonor Cerda-Alberich, Luis Marti-Bonmati

Organizations: Biomedical Imaging Research Group (GIBI230), Instituto de Investigación Sanitaria La Fe, Valencia, Spain

Abstract

Accurate segmentation of the prostate gland in multiparametric MRI (mpMRI) is a fundamental step for a wide range of clinical and research applications, including image registration, volume estimation, and radiomic analysis. However, manual delineation is time-consuming and subject to inter-observer variability, while general-purpose segmentation tools often fail to provide sufficient accuracy for prostate-specific tasks. In this work, we propose a dedicated deep learning-based approach for automatic prostate gland segmentation using the nnU-Net v2 framework. The model leverages multimodal mpMRI data, including T2-weighted imaging, diffusion-weighted imaging (DWI), and apparent diffusion coefficient (ADC) maps, to exploit complementary tissue information. Training was performed on 981 cases from the PI-CAI dataset using whole-gland annotations, and model performance was assessed through 5-fold cross-validation and external validation on an independent cohort of 54 patients from Hospital La Fe. The proposed model achieved a mean Dice score of 0.96 +/- 0.00 in cross-validation and 0.82 on the external test set, demonstrating strong generalization despite domain shift. In comparison, a general-purpose approach (TotalSegmentator) showed substantially lower performance, with a Dice score of 0.15, primarily due to under-segmentation of the gland. These results highlight the importance of task-specific, multimodal segmentation strategies and demonstrate the potential of the proposed approach for reliable integration into clinical research workflows. To facilitate reproducibility and deployment, the model has been fully containerized and is available as a ready-to-use inference tool.

Explore similar work

CardsList
  1. Promptable segmentation with region exploration enables minimal-effort expert-level prostate cancer delineation

    Feb 19, 2026Junqing Yang, Natasha Thorley, Ahmed Nadeem Abbasi +4Medical Image SegmentationInteractive Medical Image Segmentation

  2. Align then Refine: Text-Guided 3D Prostate Lesion Segmentation

    Apr 20, 2026Cuiling Sun, Linkai Peng, Adam Murphy +9Cross-Modality Medical Image SegmentationText-Guided Image Segmentation

  3. ProFound: A moderate-sized vision foundation model for multi-task prostate imaging

    Mar 4, 2026Yipei Wang, Yinsong Xu, Weixi Yi +11Medical Imaging Foundation ModelsMedical Image Representation Learning