Assessing the performance of deep learning-based models for prostate cancer segmentation using uncertainty scores

This study focuses on comparing deep learning methods for the segmentation and quantification of uncertainty in prostate segmentation from MRI images. The aim is to improve the workflow of prostate cancer detection and diagnosis. Seven different U-Net-based architectures, augmented with Monte-Carlo...

Full description

Saved in:
Bibliographic Details
Main Authors Quihui-Rubio, Pablo Cesar, Flores-Araiza, Daniel, Ochoa-Ruiz, Gilberto, Gonzalez-Mendoza, Miguel, Mata, Christian
Format Journal Article
LanguageEnglish
Published 08.08.2023
Subjects
Computer Science - Computer Vision and Pattern Recognition
Computer Science - Learning
Online AccessGet full text

Cover

More Information
Summary:This study focuses on comparing deep learning methods for the segmentation and quantification of uncertainty in prostate segmentation from MRI images. The aim is to improve the workflow of prostate cancer detection and diagnosis. Seven different U-Net-based architectures, augmented with Monte-Carlo dropout, are evaluated for automatic segmentation of the central zone, peripheral zone, transition zone, and tumor, with uncertainty estimation. The top-performing model in this study is the Attention R2U-Net, achieving a mean Intersection over Union (IoU) of 76.3% and Dice Similarity Coefficient (DSC) of 85% for segmenting all zones. Additionally, Attention R2U-Net exhibits the lowest uncertainty values, particularly in the boundaries of the transition zone and tumor, when compared to the other models.
DOI:10.48550/arxiv.2308.04653