A Multi-Stage Validation Framework for Trustworthy Large-scale Clinical Information Extraction using Large Language Models
Authors: Maria Mahbub, Gregory M. Dams, Josh Arnold, Caitlin Rizy, Sudarshan Srinivasan, Elliot M. Fielstein, Minu A. Aghevli, Kamonica L. Craig, Elizabeth M. Oliva, Joseph Erdos, Jodie Trafton, Ioana Danciu
Organizations: Oak Ridge National Laboratory, Oak Ridge, TN, USA · Program Evaluation and Resource Center, Office of Mental Health and Office of Suicide Prevention, Department of Veterans Affairs, Menlo Park, CA, USA · Vanderbilt University Medical Center, Nashville, TN, USA · VA Maryland Health Care System, Baltimore, MD, USA · VA Desert Pacific Healthcare Network, Long Beach, CA, USA · VA Connecticut Health Care System, West Haven, CT, USA · Yale School of Medicine, New Haven, CT, USA
Large language models (LLMs) show promise for extracting clinically meaningful information from unstructured health records, yet their translation into real-world settings is constrained by the lack of scalable and trustworthy validation approaches. Conventional evaluation methods rely heavily on annotation-intensive reference standards or incomplete structured data, limiting feasibility at population scale. We propose a multi-stage validation framework for LLM-based clinical information extraction that enables rigorous assessment under weak supervision. The framework integrates prompt calibration, rule-based plausibility filtering, semantic grounding assessment, targeted confirmatory evaluation using an independent higher-capacity judge LLM, selective expert review, and external predictive validity analysis to quantify uncertainty and characterize error modes without exhaustive manual annotation. We applied this framework to extraction of substance use disorder (SUD) diagnoses across 11 substance categories from 919,783 clinical notes. Rule-based filtering and semantic grounding removed 14.59% of LLM-positive extractions that were unsupported, irrelevant, or structurally implausible. For high-uncertainty cases, the judge LLM's assessments showed substantial agreement with subject matter expert review (Gwet's AC1=0.80). Using judge-evaluated outputs as references, the primary LLM achieved an F1 score of 0.80 under relaxed matching criteria. LLM-extracted SUD diagnoses also predicted subsequent engagement in SUD specialty care more accurately than structured-data baselines (AUC=0.80). These findings demonstrate that scalable, trustworthy deployment of LLM-based clinical information extraction is feasible without annotation-intensive evaluation.
Figures & tables
Figure 1: A multi-stage framework for trustworthy large-scale LLM extraction of clinical information
Figure 2: Direct vs chain-of-thought prompting to extract SUD diagnoses information from clinical notes.
Figure 3: Prompt for judge LLM
Reasoning type
Number of shots
Precision (relaxed)
Recall (relaxed)
F1 (relaxed)
F1 (strict)
Direct
0
0.8782
0.8691
0.8647
0.8203
1
0.8772
0.8564
0.8584
0.8113
2
0.8648
0.8517
0.8514
0.8031
Chain-of-thought
0
0.9189
0.8973
0.9004
0.8582
Table 2: Prompt calibration results on the annotated dataset comparing zero-, one-, and two-shot prompting under direct and chain-of-thought strategies.
Substance category
Pre-filter LLM+
Count of flagged LLM+ extractions
Post-filter LLM+
Rule-based
Semantic grounding
0.50
0.55
0.60
0.65
0.70
Alcohol
261037
29692
972
1064
1260
1708
2270
229637
Opioid
63123
14123
824
855
947
1057
1326
47943
Cannabis
71777
5917
1065
1093
1150
1315
1579
64545
Cocaine
65796
4066
873
888
920
1025
1317
60705
Table 3: LLM+ extraction counts and flagged extraction counts from rule-based filtering and semantic grounding assessment by substance category.
Substance category
Precision (relaxed)
Recall (relaxed)
F1 (relaxed)
F1 (strict)
Alcohol
0.8678
0.8759
0.8689
0.8446
Opioid
0.8410
0.8448
0.8394
0.8106
Cannabis
0.8026
0.8076
0.8027
0.7805
Cocaine
0.7690
0.7735
0.7682
0.7401
Amphetamine
0.6155
0.6155
0.6139
0.5934
Caffeine
0.6745
0.6739
0.6730
0.6542
Table 4: Information extraction performance for SUD diagnoses as assessed by the judge LLM, considering only instances where the primary LLM produced an extraction.
Substance category
Overall Agreement (%)
Gwet’s AC1
Alcohol
90
0.87
Opioid
92
0.9
Cannabis
92
0.9
Cocaine
91
0.88
Amphetamine
83
0.79
Caffeine
95
0.95
Table 5: Inter-annotator agreement (Gwet’s AC1) between the judge LLM and SME for a random sample of 100 clinical notes across substance categories.
Figure 4: Predictive validity of LLM-extracted SUD diagnoses shown by ROC curves comparing outcome prediction using LLM-extracted, ICD-10–based, and combined features across cohorts (a) full study population, (b) concordant cohort, and (c) narrative-only cohort.
Department of Biomedical Informatics, School of Medicine, University of Pittsburgh, Pittsburgh, PA, USA · Department of Health Information Management, School of Health and Rehabilitation Sciences, University of Pittsburgh, Pittsburgh, PA, USA · OpenCura, Health Innovation Consortium +6