arrow
Return

Uncertainty-Aware Classification: A Human-Guided Bayesian Deep Learning Framework

delete2026-01-01
delete1
PRE
AI
K
Kampen, PeterJ. T. *
M
M. Reimann
M
Morten Rieger Hannemose
A
Anders Nymark Christensen
M
Miriam Kolko
A
Anders Bjorholm Dahl
J
Josefine Vilsbøll Sundgaard
DOI:10.1007/978-3-032-06593-3_19delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
While neural networks achieve strong performance in medical image analysis, effectively combining their predictions with human expertise remains a critical challenge for clinical deployment. We examine how different choices of stochastic parameter subsets used in approximate Bayesian inference impact the posterior predictive distributions and, consequently, the performance of a combined human-AI decision model. Using two medical classification tasks, we analyze the relationship between the resulting model and human uncertainty. We demonstrate that uncertainty estimates correlate differently with human uncertainty depending on the stochastic subsets. Building on these findings, we propose a framework that optimizes the choice of stochastic subsets to improve a final decision model that considers human uncertainty, enabling more reliable and interpretable integration of human and AI predictions in clinical settings. Our implementation is publicly available at https://github.com/mkreimann/uncertainty-guided-classification.
Keywords:
Uncertainty
Model calibration
Trustworthy AI

Journal

U
UNCERTAINTY FOR SAFE UTILIZATION OF MACHINE LEARNING IN MEDICAL IMAGING, UNSURE 2025
IF:
0
Papers:
22
Citations:
0

Organization

T
Technical University of Denmark
Scholars:
2.5K
Papers: 980
Citations: 1
U
university of copenhagen
Scholars:
7.2K
Papers: 2.9K
Citations: 0