ChronoMedicalWorld: A Medical World Model for Learning Patient Trajectories from Longitudinal Care Data
On the map: World models → Medical and surgical world models
Abstract (arXiv)
Long-horizon clinical simulation -- predicting how a patient's physiology evolves over years under specified interventions -- is central to chronic-disease care, yet existing electronic health record (EHR) models are predominantly discriminative, and general-purpose large language models drift under repeated interventions. We propose the ChronoMedicalWorld Model (CMWM), an action-conditioned latent world-model framework for learning patient trajectories from longitudinal care data. CMWM couples a joint-embedding state encoder with a wide action encoder that admits both structured intervention indicators and free-text communication embeddings, and trains a recurrent latent transition module under a six-term objective: next-observation supervision, next-latent prediction, SIGReg latent regularisation, and three physiology-aware shape priors (slope, continuity, large-jump penalty). A closed-loop rollout-prefix protocol matches training to deployment, so the model is optimised against the same multi-step error it exhibits at inference. As a concrete case study, we instantiate CMWM for annual estimated glomerular filtration rate (eGFR) trajectory forecasting in chronic kidney disease (CKD). On a 2,232-patient nephrology cohort, the CKD instantiation achieves a dynamic-50% history rollout test mean absolute error (MAE) of 7.384 and root-mean-square error (RMSE) of 10.256, against 7.964 and 11.069 for a tuned GPT-5.5 structured-prompting baseline (-7.28% MAE, -7.35% RMSE), with the gain dominated by the dialogue portion of patient--health-coach communication. The framework is not CKD-specific: its architecture, loss design, and training protocol apply to any chronic condition that can be cast as periodic clinical state interleaved with structured and conversational interventions.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Medical and surgical world models, one of 11 directions in World models. Builds on 15 of its 31 references that are in the map. First posted on arXiv 21 May 2026.
Builds on
15 of 31 references.
Closest ancestors · 1
Same direction, newest first
Other references · 14
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
- 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
- 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 344
- 2019BEHRT: Transformer for Electronic Health Recordscited by 258
- 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixelscited by 237
- 2021Multi-Time Attention Networks for Irregularly Sampled Time Seriescited by 234
- 2019Latent ODEs for Irregularly-Sampled Time Seriescited by 182
- 2020Estimating Counterfactual Treatment Outcomes over Time Through Adversarially Balanced Representationscited by 154
- 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Medical world models in healthcare: foundations, applications, and challenges for trustworthy clinical translationcited by 3
- 2026DRIFT: Direct-Recursive Intervention-Conditioned Forecasting of ICU Physiological Trajectoriescited by 0
- 2026Intervention Granularity Matters: Coherent Treatment Bundles in Counterfactual Simulation with Clinical World Modelscited by 0
- 2026Can LLMs Value the Right Evidence? Evidence-Value Misalignment in Dynamic Medical Diagnosiscited by 0
- 2026Can LLMs Value the Right Evidence? Evidence-Value Misalignment in Dynamic Medical Diagnosis28 Sep 2026
- 2026Intervention Granularity Matters: Coherent Treatment Bundles in Counterfactual Simulation with Clinical World Models18 Sep 2026
- 2026DRIFT: Direct-Recursive Intervention-Conditioned Forecasting of ICU Physiological Trajectories28 Jul 2026
- 2026Medical world models in healthcare: foundations, applications, and challenges for trustworthy clinical translation28 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026The Patient is not a Moving Document: A World Model Training Paradigm for Longitudinal EHR
- 2026Clin-JEPA: A Multi-Phase Co-Training Framework for Joint-Embedding Predictive Pretraining on EHR Patient Trajectories
- 2026Learning Continuous Patient Trajectories from Electronic Health Records
- 2026LongMoE: Longitudinal Multimodal Learning via Trajectory-Aware Mixture-of-Experts
- 2026Toward World Modeling of Physiological Signals with Chaos-Theoretic Balancing and Latent Dynamics
- 2026Simulating clinical interventions with a generative multimodal model of human physiology
- 2026CalTwin: Towards Calibrated, Shift-Robust Medical World Models via Fisher-Information Regularisation
- 2026LLM4EHR: Aligning Clinical Time Series with Medical Event Sequences via Large Language Models