Conformal Prediction for Risk-Controlled Medical Entity Extraction Across Clinical Domains
Not on the map. Why?
Abstract (arXiv)
Large Language Models (LLMs) are increasingly used for medical entity extraction, yet their confidence scores are often miscalibrated, limiting safe deployment in clinical settings. We present a conformal prediction framework that provides finite-sample coverage guarantees for LLM-based extraction across two clinical domains. First, we extract structured entities from 1,000 FDA drug labels across eight sections using GPT-4.1, verified via FactScore-based atomic statement evaluation (97.7% accuracy over 128,906 entities). Second, we extract radiological entities from MIMIC-CXR reports using the RadGraph schema with GPT-4.1 and Llama-4-Maverick, evaluated against physician annotations (entity F1: 0.81 to 0.84). Our central finding is that miscalibration direction reverses across domains: on well-structured FDA labels, models are underconfident, requiring modest conformal thresholds (τ ≈ 0.06), while on free-text radiology reports, models are overconfident, demanding strict thresholds (τ up to 0.99). Despite this heterogeneity, conformal prediction achieves target coverage (≥ 90%) in both settings with manageable rejection rates (9--13%). These results demonstrate that calibration is not a global model property but depends on document structure, extraction category, and model architecture, motivating domain-specific conformal calibration for safe clinical deployment.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 11 of its 21 references that are in the map. First posted on arXiv 1 Mar 2026.
Builds on
11 of 21 references.
References · 11
- 2023GPT-4 Technical Reportcited by 18,132
- 2017On Calibration of Modern Neural Networkscited by 4,860
- 2021RadGraph: Extracting Clinical Entities and Relations from Radiology Reportscited by 310
- 2021Distribution-Free, Risk-Controlling Prediction Setscited by 278
- 2023A Survey of Confidence Estimation and Calibration in Large Language Modelscited by 253
- 2023Are You Sure? Challenging LLMs Leads to Performance Drops in The FlipFlop Experimentcited by 71
- 2024Conformal Prediction for Natural Language Processing: A Surveycited by 52
- 2025VeriFact: Verifying Facts in LLM-Generated Clinical Text with Electronic Health Recordscited by 17
- 2023Empirical Validation of Conformal Prediction for Trustworthy Skin Lesions Classificationcited by 11
- 2024Structured Extraction of Real World Medical Knowledge using LLMs for Summarization and Searchcited by 3
- 2025Mining Unstructured Medical Texts With Conformal Active Learningcited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026CARE: A Conformal Safety Layer for Medical Summarizationcited by 1
- 2026CARE: A Conformal Safety Layer for Medical Summarization8 Jun 2026