EHR-MPC: Inference-Time Control for Sepsis Treatment with Generative Patient Digital Twins
Not on the map. Why?
Abstract (arXiv)
Sepsis is a leading cause of mortality, yet optimal treatment policies remain contested. Existing reinforcement learning (RL) approaches learn fixed strategies for sepsis treatment, limiting adaptability to changing clinical objectives during inference. We propose EHRMPC, a framework that decouples learning patient dynamics from optimizing treatment by training a patient digital twin in the form of a generative electronic health record (EHR) model. The digital twin predicts clinical trajectories under interventions and enables model predictive control (MPC) to optimize treatments via inference-time planning over simulations. We evaluate EHR-MPC on a multicenter ICU sepsis cohort spanning 8 hospitals in the Mass General Brigham health system using both off-policy importance sampling and on-policy simulation-based evaluation. Relative to RL baselines, EHR-MPC achieves comparable off-policy performance and improved simulation performance. Unlike RL, this work frames sepsis treatment optimization as inference-time control over learned patient dynamics, establishing a general framework for decision making with generative clinical models.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 19 of its 62 references that are in the map. First posted on arXiv 7 Jul 2026.
Builds on
19 of 62 references.
References · 19
- 2019BioBERT: a pre-trained biomedical language representation model for biomedical text miningcited by 1,935
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
- 2023Exposing flaws of generative model evaluation metrics and their unfair treatment of diffusion modelscited by 199
- 2019Counterfactual Off-Policy Evaluation with Gumbel-Max Structural Causal Modelscited by 114
- 2020Controlling generative models with continuous factors of variationscited by 93
- 2024Zero Shot Health Trajectory Prediction Using Transformercited by 74
- 2017Deep Reinforcement Learning for Sepsis Treatmentcited by 71
- 2017Continuous State-Space Models for Optimal Sepsis Treatment - a Deep Reinforcement Learning Approachcited by 68
- 2018Evaluating Reinforcement Learning Algorithms in Observational Health Settingscited by 60
- 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
- 2019Improving Sepsis Treatment Strategies by Combining Deep and Kernel-Based Reinforcement Learningcited by 34
- 2018Model-Based Reinforcement Learning for Sepsis Treatmentcited by 21
7 more
- 2025MIRA: Medical Time Series Foundation Model for Real-World Health Datacited by 21
- 2025Foundation Model of Electronic Medical Records for Adaptive Risk Estimationcited by 17
- 2021Unifying Cardiovascular Modelling with Deep Reinforcement Learning for Uncertainty Aware Control of Sepsis Treatmentcited by 16
- 2021Offline reinforcement learning with uncertainty for treatment strategies in sepsiscited by 9
- 2020Optimizing Medical Treatment for Sepsis in Intensive Care: from Reinforcement Learning to Pre-Trial Evaluationcited by 8
- 2025TrajSurv: Learning Continuous Latent Trajectories from Electronic Health Records for Trustworthy Survival Predictioncited by 4
- 2025Traj-CoA: Patient Trajectory Modeling via Chain-of-Agents for Lung Cancer Risk Predictioncited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.