JudgEHR: LLM-Guided Cohort Inference for Clinical Decision Confidence Estimation

Abstract

Clinical decisions, such as diagnosing conditions, prescribing medications, and recommending procedures, are rarely made with absolute certainty. Instead, they reflect probabilistic judgments shaped by evolving patient information and incomplete evidence. However, current EHR systems and knowledge graphs encode such decisions as deterministic triples, lacking a mechanism to represent the subjective confidence inherent in clinical reasoning. We present JudgEHR, a framework for clinical decision confidence estimation that leverages large language models (LLMs) to perform cohort-based collective inference over structured patient records by representing clinical events as knowledge graph triples and integrating them into LLM prompts. JudgEHR groups related clinical concepts into cohorts using LLM-driven relational inference, and then jointly evaluates the plausibility of all clinical decision triples within each cohort by considering patient visit history and background medical knowledge. We apply our method to the MIMIC-III dataset. Our statistical analysis shows that JudgEHR generates semantically consistent confidence scores, with similar concepts receiving closer values, whereas dissimilar replacements yield large confidence differences. Experiments on the MIMIC-III dataset show that incorporating the confidence into a zero-shot LLM-based pipeline improves relative AUROC by ${1 4. 6 \%}$ and AUPRC by 21.8% on the mortality prediction task.