LLM-Extracted Covariates for Clinical Causal Inference: Rethinking Integration Strategies
Not on the map. Why?
Abstract (arXiv)
Causal inference from electronic health records (EHR) is fundamentally limited by unmeasured confounding: critical clinical states such as frailty, goals of care, and mental status are documented in free-text notes but absent from structured data. Large language models can extract these latent confounders as interpretable, structured covariates, yet how to effectively integrate them into causal estimation pipelines has not been systematically studied. Using the MIMIC-IV database with 21,859 sepsis patients, we compare seven covariate-integration strategies for estimating the effect of early vasopressor initiation on 28-day mortality, spanning tabular-only baselines, traditional NLP representations, and three LLM-augmented approaches. A central finding is that not all integration strategies are equally effective: directly augmenting the propensity score model with LLM covariates achieves the best performance, while dual-caliper matching on text-derived categorical distances restricts the donor pool and degrades estimation. In semi-synthetic experiments with known ground-truth effects, LLM-augmented propensity scores reduce estimation bias from 0.0143 to 0.0003 relative to tabular-only methods, and this advantage persists under substantial simulated extraction error. On real data, incorporating LLM-extracted covariates reduces the estimated treatment effect from 0.055 to 0.027, directionally consistent with the CLOVERS randomized trial, and a doubly robust estimator yielding 0.031 confirms the robustness of this finding. Our results offer practical guidance on when and how text-derived covariates improve causal estimation in critical care. Our code is available at https://github.com/fpxlei/LLM-Covariates-Causal.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 16 of its 57 references that are in the map. First posted on arXiv 18 Apr 2026.
Builds on
16 of 57 references.
References · 16
- 2025Qwen3 Technical Reportcited by 8,820
- 2019Publicly Available Clinical BERT Embeddingscited by 1,002
- 2023Capabilities of GPT-4 on Medical Challenge Problemscited by 638
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
- 2022GatorTron: A Large Clinical Language Model to Unlock Patient Information from Unstructured Electronic Health Recordscited by 238
- 2022Large Language Models are Few-Shot Clinical Information Extractorscited by 207
- 2021Causal Inference in Natural Language Processing: Estimation, Prediction, Interpretation and Beyondcited by 178
- 2020Text and Causal Inference: A Review of Using Text to Remove Confounding from Causal Estimatescited by 59
- 2020Causal Effects of Linguistic Propertiescited by 39
- 2020Adjusting for Confounders with Text: Challenges and an Empirical Evaluation Framework for Causal Inferencecited by 13
- 2023Leveraging text data for causal inference using electronic health recordscited by 6
- 2018Cause-Effect Deep Information Bottleneck For Systematically Missing Covariatescited by 1
4 more
- 2018Estimating Causal Effects With Partial Covariates For Clinical Interpretabilitycited by 1
- 2019The Medical Deconfounder: Assessing Treatment Effects with Electronic Health Recordscited by 1
- 2024Controlling for Unobserved Confounding with Large Language Model Classification of Patient Smoking Statuscited by 1
- 2025REACT-LLM: A Benchmark for Evaluating LLM Integration with Causal Features in Clinical Prognostic Taskscited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025LLM-Augmented Symptom Analysis for Cardiovascular Disease Risk Prediction: A Clinical NLP
- 2025LLM-based Agents for Automated Confounder Discovery and Subgroup Analysis in Causal Inference
- 2026Linear-LLM-SCM: Benchmarking LLMs for Coefficient Elicitation in Linear-Gaussian Causal Models
- 2025Enhancing LLMs' Clinical Reasoning with Real-World Data from a Nationwide Sepsis Registry
- 2025ClinNoteAgents: An LLM Multi-Agent System for Predicting and Interpreting Heart Failure 30-Day Readmission from Clinical Notes
- 2025A Technical Exploration of Causal Inference with Hybrid LLM Synthetic Data
- 2025Aligning Probabilistic Beliefs under Informative Missingness: LLM Steerability in Clinical Reasoning
- 2025Visual Analytics for Causal Reasoning from Real-World Health Data ↗