Multimodal Learning 相关度: 7/10

Automated Prostate Gland Segmentation in MRI Using nnU-Net

Pablo Rodriguez-Belenguer, Gloria Ribas, Javier Aquerreta Escribano, Rafael Moreno-Calatayud, Leonor Cerda-Alberich, Luis Marti-Bonmati
arXiv: 2604.01964v1 发布: 2026-04-02 更新: 2026-04-02

AI 摘要

使用nnU-Net自动分割MRI前列腺,实现高精度和泛化性,优于通用分割方法。

主要贡献

  • 提出了一种基于nnU-Net v2的MRI前列腺自动分割方法
  • 利用多模态mpMRI数据提高了分割精度
  • 在PI-CAI数据集和独立队列上验证了模型的泛化能力

方法论

使用nnU-Net v2框架,在PI-CAI数据集上训练模型,利用多模态mpMRI数据,进行5折交叉验证和外部验证。

原文摘要

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.

标签

医学图像分割 深度学习 nnU-Net MRI 前列腺

arXiv 分类

cs.CV