Advancements in prostate cancer segmentation: Integrating prostate zonal information.
Authors
Affiliations (3)
Affiliations (3)
- Faculty of Mathematics and Informatics, Institute of Data Science and Digital Technologies, Vilnius University, Vilnius, Lithuania.
- National Cancer Institute, Vilnius, Lithuania.
- Faculty of Mathematics and Informatics, Institute of Applied Mathematics, Vilnius University, Vilnius, Lithuania.
Abstract
Prostate cancer is one of the most common cancers in men, and early diagnosis is critical. Segmentation of cancerous regions in multiparametric MRI is a key step. Deep neural networks, such as nnU-Net, perform well, and incorporating prostate zonal information may further improve accuracy. This study introduces four prostate cancer segmentation ensembles that integrate zonal data, compared with a baseline model, which uses zonal information as a separate input channel. Ensembles employ specific prostate zone cancer segmentation models trained with the nnU-Net method. To address variability in manual annotations, a new evaluation metric, the tolerant Dice Score Coefficient (<i>DSC</i> <sub><i>τ</i></sub> ), is proposed, accounting for ground truth inaccuracies. Ensemble 3 yields the best performance, with a 4.77% higher mean DSC and 6.17% higher mean <i>DSC</i> <sub><i>τ</i></sub> than the baseline. Although the metrics of Ensemble 4 are slightly lower, it reduces false positives by 7.79% and uses fewer models (2 vs. 3), making it more efficient. Furthermore, the application of the Conover post hoc test for unreplicated blocked data shows that there is no statistically significant difference in performance metrics between the results of two ensembles. Thus, Ensemble 4 is the preferred approach for prostate cancer segmentation. Additionally, all ensembles achieve 5.03% to 7.13% higher mean <i>DSC</i> <sub><i>τ</i></sub> values compared to the standard DSC, confirming the effectiveness of the new metric in handling segmentation uncertainties. The experiment results indicate that the proposed Ensemble 4 is the most suitable solution for the prostate cancer segmentation task. Moreover, the results also indicate that the proposed metric, <i>DSC</i> <sub><i>τ</i></sub> , accounts for ground truth segmentation errors.