Skip to content

Bayesian uncertainty estimation improves clinical decision making in medical AI agents

Jul 2026 · arXiv.org · Vol abs/2607.20582 · 0 citations · 27 references
Computer Science

TL;DR

It is shown that Monte Carlo dropout, applied to a multi-task chest-radiograph classifier, provides an epistemic uncertainty signal that tracks generalisation across training-set scales and flags confident yet error-prone predictions.

Abstract

Machine learning models for medical image analysis typically lack a reliable measure of confidence, limiting their use in ambiguous or atypical cases. Here we show that Monte Carlo dropout, applied to a multi-task chest-radiograph classifier (eight thoracic findings, 137,593 training images), provides an epistemic uncertainty signal that tracks generalisation across training-set scales and flags confident yet error-prone predictions. Adding this signal to the point prediction raised error-detection AUROC from 0.74 to 0.77 ($\Delta$AUROC +0.023, 95% CI [+0.014, +0.033]). In a controlled 2x2 factorial experiment, a clinical-decision-support agent exploited this uncertainty only when it was delivered as a binary error-risk flag rather than as raw scores, cutting confident misdiagnoses on unreliable findings from 8.5% to 2.7%. Epistemic uncertainty estimation thus carries decision-relevant information beyond point predictions, but its value for downstream agents depends on how it is communicated.

View source

Similar papers

Open access 2026

Evaluating Uncertainty Quantification in Clinical Machine Learning: Calibration, Robustness, and Decision Utility under Distribution Shift

A rigorous empirical framework is presented for comparing three uncertainty quantification approaches on two clinical prediction tasks, in-hospital mortality and 30-day readmission, using 74,829 ICU admissions from the MIMIC-IV database to support a more demanding evaluation standard for UQ in clinical machine learning...

Isaac Tosin Adisa, Francis Mawutor Amuyao, Ezekiel Olaoluwa Joaquim · 0 citations
#artificial intelligence Preprint Aug 2026

Explainable Uncertainty Estimation for Reliable Medical AI

The Expected Gradients Reconstruction Uncertainty Estimate (egRUE), which incorporates prediction explanations into its uncertainty computation and decomposes uncertainty into feature-wise contributions, and proves theoretical properties of egRUE and shows that it improves reliability and interpretability compared to e...

Li-Rong Wang, J. Duell, Xin-Ran Xu et al. · 0 citations
Review Open access Jul 2026

Uncertainty quantification for artificial intelligence in medical imaging: what every radiologist needs to know.

Although UQ has the potential to improve the safety and interpretability of AI-assisted screening, challenges remain, including calibration, threshold selection, computational cost, and the need for prospective clinical validation.

Fernando Vega Lara, Lisa Koopmans, Christian Roest et al. · 1 citation
Open access Aug 2026

CBEC: a simple retrieval-based framework for population-grounded prediction reliability estimation in chest X-ray diagnosis

It is suggested that case-level consistency provides a meaningful practical signal for reliability estimation in medical imaging, whose relationship to the classical epistemic–aleatoric decomposition warrants further theoretical investigation across broader clinical settings.

Muhannad Faleh Alanazi, B. Z. Shakhreet, Hattan Ali A. Asiri et al. · 0 citations
Open access Sep 2026

Uncertainty-Aware Machine Learning for Reliable Decision-Making in Data-Driven Systems

An uncertainty-aware machine learning framework that integrates probabilistic modeling techniques into conventional predictive architectures to jointly estimate epistemic and aleatoric uncertainty is proposed, indicating that incorporating uncertainty estimation enhances trustworthiness and robustness without sacrifici...

Evi Yulianingsih, E. Noche, V. Yadav et al. · 0 citations
Open access Aug 2026

A knowledge-based Bayesian network reveals shared prediction errors with physicians in Gamma Knife radiosurgery

A knowledge-based expert system using a Bayesian Network to explicitly model an expert neurosurgeon's clinical reasoning for predicting outcomes of Gamma Knife radiosurgery highlights a limitation of predictive paradigms based on current expert knowledge and suggests the need for iterative refinement of knowledge-based...

Yusuke Uchiyama, Soichiro Fujiki, Kensaku Nomoto et al. · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.