Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

MedBayes-Lite: Bayesian Uncertainty Quantification for Safe Clinical Decision Support

About

We propose MedBayes-Lite, a lightweight Bayesian enhancement for transformer-based clinical language models that improves reliability through uncertainty-aware prediction. The framework operates without retraining, architectural modification, or additional trainable parameters, and integrates three components: Bayesian Embedding Calibration via Monte Carlo dropout, Uncertainty-Weighted Attention for reliability-aware token aggregation, and Confidence-Guided Decision Shaping for abstention under uncertainty. Across MedQA, PubMedQA, and MIMIC-III, MedBayes-Lite improves calibration and trustworthiness, reducing overconfidence by 32--48\%. In simulated clinical settings, it further supports safer decision-making by flagging uncertain predictions for human review, particularly under distribution shift. For closed API models, the framework remains applicable through sampling-based predictive uncertainty and confidence-guided abstention, while full embedding- and attention-level uncertainty propagation is evaluated on open-weight transformer models.

Elias Hossain, Md Mehedi Hasan Nipu, Maleeha Sheikh, Rajib Rana, Subash Neupane, Niloofar Yousefi• 2025

Related benchmarks

TaskDatasetResultRank
Medical Question AnsweringPubMedQA (test)
CUS Score7.11
4
Biomedical Question AnsweringPubMedQA (test)
CUS0.254
2
Clinical Text AnalysisMIMIC-III
CUS0.38
2
Medical Question AnsweringMedQA
CUS31
2
Medical Question AnsweringPubMedQA
CUS Score66
2
Biomedical Question AnsweringMedQA (test)
CUS28.9
2
Medical Question AnsweringMIMIC-III (test)
CUS Score23.12
2
Showing 7 of 7 rows

Other info

Follow for update