URGen: Uncertainty-guided Retrieval-augmented Generation for Guideline-Grounded Medical Question Answering

Mengieong Hoi, Chen Zhao, Chaoyu Lei, Huifang Zhou, Zhedong Zheng


Abstract
Large language models (LLMs) achieve strong fluency in open-domain question answering, yet often hallucinate in specialized medical QA and lack reliable uncertainty quantification, posing risks in safety-critical applications. We propose Uncertainty-guided Retrieval-augmented Generation (URGen), a training-free framework for guideline-grounded medical multiple-choice QA, where authoritative clinical documents (e.g., consensus statements, treatment protocols) are available and answer correctness depends on precise evidence retrieval. URGen is a four-stage pipeline that combines adaptive retrieval depth with uncertainty calibration: (1) multi-temperature knowledge retrieval to rank candidate keypoints, (2) zero-shot adaptive retrieval depth (ZARD) that expands context when rankings are ambiguous, and contracts it when rankings are consistent, (3) multi-temperature answer generation with aggregation over candidates, and (4) entropy-based uncertainty estimation over the answer distribution to enable abstention on high-uncertainty queries. Extensive experiments on two expert-annotated ophthalmology benchmarks, Thyroid Eye Disease and Graves’ Disease, show that URGen improves accuracy by 9.29% and 6.19% over a naive RAG-based Meditron-7B baseline, while achieving 3.01% and 4.57% expected calibration error (ECE). URGen is LLM-agnostic, consistently improving diverse backbones (, LLaMA-3.1 and Mistral) and providing interpretable uncertainty indicators for safety-critical deployment.
Anthology ID:
2026.findings-emnlp.184
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
3918–3936
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.184/
DOI:
Bibkey:
Cite (ACL):
Mengieong Hoi, Chen Zhao, Chaoyu Lei, Huifang Zhou, and Zhedong Zheng. 2026. URGen: Uncertainty-guided Retrieval-augmented Generation for Guideline-Grounded Medical Question Answering. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 3918–3936, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
URGen: Uncertainty-guided Retrieval-augmented Generation for Guideline-Grounded Medical Question Answering (Hoi et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.184.pdf