Reliability of uncertainty quantification methods for deep learning auto-segmentation in head and neck organs at risk

<i>Objective.</i>Deep learning auto-segmentation has greatly advanced contouring in radiotherapy. However, quality assurance remains necessary due to performance fluctuation among individual patients. This manual process reintroduces variability and partially reduces time-saving benefits. As a solution, uncertainty quantification (UQ) is increasingly explored for its ability to estimate output confidence. While numerous methods to quantify uncertainty exist, their comparative reliability remains underexplored. This study compares the reliability of commonly used UQ approaches for auto-segmentation in radiotherapy.<i>Approach.</i>We evaluated the reliability of three popular uncertainty methods (Monte Carlo dropout, deep ensemble modelling and test-time augmentation) and uncertainty metrics (predictive entropy, mutual information and variance). We trained a 3D U-Net within the nnU-Net framework to segment 19 organs at risk (OAR) for head and neck cancer patients. We evaluated the reliability of the UQ methods and metrics on a set of 10 patients using segmentation model accuracy (surface Dice similarity coefficient), confidence calibration (expected calibration error (label)), and error localisation ability (uncertainty-error (U-E) overlap). Both multi-class and class-specific uncertainty maps were assessed.<i>Main results.</i>Segmentation accuracy remained stable without significant deviations across all UQ methods with respect to the baseline model without UQ. The reliability of different UQ methods in terms of confidence calibration and error localisation was also comparable. In contrast, the choice of UQ metric significantly influenced reliability. Multi-class predictive entropy (ECE-label: 0.06-0.06, U-E overlap: 0.45-0.46) consistently outperformed variance (ECE-label: 0.13-0.14, U-E overlap: 0.32-0.40,<i>p</i>< 0.001) and mutual information (ECE-label: 0.13-0.14, U-E overlap: 0.35-0.40,<i>p</i>< 0.001). Predictive entropy also demonstrated superior reliability as a class-specific UQ metric, though variability was observed between OARs.<i>Significance.</i>This study demonstrates that the choice of UQ approach substantially impacts the reliability of uncertainty maps. While different UQ methods performed comparably, the specific UQ metric chosen significantly affected reliability. These findings underscore the importance of careful metric selection and evaluation prior to application.

Authors

Institutions

Publication Details

Journal
Physics in Medicine and Biology
Published
2025-10-08
DOI
https://doi.org/10.1088/1361-6560/ae110c
Primary Topic
Advanced Radiotherapy Techniques
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Reliability of uncertainty quantification methods for deep learning auto-segmentation in head and neck organs at risk

Peter M. A. van Ooijen, Tomas Janssen, Jelmer M. Wolterink, Charlotte L. Brouwer et al.
Physics in Medicine and Biology
Advanced Radiotherapy Techniques
article

Reliability of uncertainty quantification methods for deep learning auto-segmentation in head and neck organs at risk

Peter M. A. van Ooijen, Tomas Janssen, Jelmer M. Wolterink, Charlotte L. Brouwer, Federica Carmen Maruccio, Rita Simões, Joëlle E. van Aalst
article en

Abstract

<i>Objective.</i>Deep learning auto-segmentation has greatly advanced contouring in radiotherapy. However, quality assurance remains necessary due to performance fluctuation among individual patients. This manual process reintroduces variability and partially reduces time-saving benefits. As a solution, uncertainty quantification (UQ) is increasingly explored for its ability to estimate output confidence. While numerous methods to quantify uncertainty exist, their comparative reliability remains underexplored. This study compares the reliability of commonly used UQ approaches for auto-segmentation in radiotherapy.<i>Approach.</i>We evaluated the reliability of three popular uncertainty methods (Monte Carlo dropout, deep ensemble modelling and test-time augmentation) and uncertainty metrics (predictive entropy, mutual information and variance). We trained a 3D U-Net within the nnU-Net framework to segment 19 organs at risk (OAR) for head and neck cancer patients. We evaluated the reliability of the UQ methods and metrics on a set of 10 patients using segmentation model accuracy (surface Dice similarity coefficient), confidence calibration (expected calibration error (label)), and error localisation ability (uncertainty-error (U-E) overlap). Both multi-class and class-specific uncertainty maps were assessed.<i>Main results.</i>Segmentation accuracy remained stable without significant deviations across all UQ methods with respect to the baseline model without UQ. The reliability of different UQ methods in terms of confidence calibration and error localisation was also comparable. In contrast, the choice of UQ metric significantly influenced reliability. Multi-class predictive entropy (ECE-label: 0.06-0.06, U-E overlap: 0.45-0.46) consistently outperformed variance (ECE-label: 0.13-0.14, U-E overlap: 0.32-0.40,<i>p</i>< 0.001) and mutual information (ECE-label: 0.13-0.14, U-E overlap: 0.35-0.40,<i>p</i>< 0.001). Predictive entropy also demonstrated superior reliability as a class-specific UQ metric, though variability was observed between OARs.<i>Significance.</i>This study demonstrates that the choice of UQ approach substantially impacts the reliability of uncertainty maps. While different UQ methods performed comparably, the specific UQ metric chosen significantly affected reliability. These findings underscore the importance of careful metric selection and evaluation prior to application.

Physics in Medicine and BiologyVol. 70(20)
University Medical Center Groningen (NL), Hanze University of Applied Sciences (NL), Amsterdam UMC Location University of Amsterdam (NL), The Netherlands Cancer Institute (NL), Dialyse Centrum Groningen (NL), Oncode Institute (NL), Netherlands Cancer Institute-Antoni van Leeuwenhoek Hospital (NL), University of Twente (NL)
Openalex Percentile: Top 62%
Advanced Radiotherapy Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.