EHR2Trace: Auditable EHR Data Infrastructure for Patient World Models and Clinical Agents
cs.LG, cs.AI, cs.DB, q-bio.QM
Submitted: 2026-09-18
Updated: 2026-09-18
Comments: 13 pages, 3 figures, 6 tables. Code and experiment records: https://github.com/Yangxinyee/ehr2trace
Code: https://github.com/Yangxinyee/ehr2trace
Project page: https://ohdsi.github.io/CommonDataModel/cdm54
License: http://creativecommons.org/licenses/by/4.0/
The gist: Patient world models and clinical agents aim to predict changes in patients' health and support clinical work.
Terminology
Abstract
Patient world models and clinical agents aim to predict changes in patients' health and support clinical work. Developing these systems requires reliable histories of patient conditions, treatments, and the information available at each decision. Electronic health records (EHRs) contain these histories, but differences in how events are recorded make them difficult to use consistently. We present EHR2Trace, a system that converts EHRs from different sources into traceable patient events for model training and evaluation. It links events to source records, separates event time from information availability, and distinguishes medication orders, dispensing, and administration. A shared event representation supports both OMOP and MEDS exports, with automated validation and reproducible builds. Across three clinical datasets, EHR2Trace converted 846.4 million events, with every applicable check passing except one unit-consistency check on MIMIC-IV, and detected all 28 injected faults. A controlled prediction experiment showed that assigning later diagnoses to admission time substantially inflated measured performance, and that a model trained on such data lost accuracy when deployed on histories filtered by availability. EHR2Trace provides a reusable data foundation for patient world models and clinical agents, helping researchers inspect patient histories, check conversion decisions, and evaluate models with explicit data rules.
Sources
- PhysicianBench: Evaluating LLM Agents in Real-World EHR Environments
- EHRWorld: A Patient-Centric Medical World Model for Long-Horizon Clinical Trajectories
- ACES: Automatic Cohort Extraction System for Event-Stream Datasets
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks