用nnU-Net自动分割MRI中的前列腺,准确率高达96%。
Automated Prostate Gland Segmentation in MRI Using nnU-Net
- 基于nnU-Net框架,融合T2、DWI和ADC多模态MRI数据
- 交叉验证Dice达0.96,外部验证仍保持0.82
- 适合临床研究使用,已开源部署工具
多参数MRI中前列腺的精确分割是图像配准、体积估算和放射组学分析等临床与科研应用的基础。然而,人工勾画耗时且存在观察者差异,通用分割工具对前列腺任务精度不足。本文提出一种基于nnU-Net v2的专用深度学习方法,利用T2加权成像、扩散加权成像(DWI)和表观扩散系数(ADC)图等多模态mpMRI数据,挖掘组织互补信息。在PI-CAI数据集的981例全腺体标注数据上训练,并通过5折交叉验证及来自医院La Fe的54例独立队列进行外部验证。模型在交叉验证中平均Dice得分为0.96±0.00,在外部测试集上为0.82,展现出强泛化能力。相比之下,通用工具TotalSegmentator的Dice仅为0.15,主要因腺体欠分割。结果表明,任务专用的多模态策略至关重要,所提方法具备可靠集成至临床研究流程的潜力。为促进可复现性与部署,模型已完全容器化,提供即用型推理工具。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。