mlsanityData to 7 Oct 2026

ChronoMedicalWorld: A Medical World Model for Learning Patient Trajectories from Longitudinal Care Data

arXiv 2605.21963 ↗First posted 21 May 2026cs.LG cs.AIcited by 4

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

  1. 2026EHRWorld: A Patient-Centric Medical World Model for Long-Horizon Clinical Trajectories3 Feb

Other references · 14

  1. 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
  2. 2023Mastering Diverse Domains through World Modelscited by 1,193
  3. 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
  4. 2018Recurrent World Models Facilitate Policy Evolutioncited by 761
  5. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 368
  6. 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 344
  7. 2019BEHRT: Transformer for Electronic Health Recordscited by 258
  8. 2026LeWorldModel: Stable End-to-End Joint-Embedding Predictive Architecture from Pixelscited by 237
  9. 2021Multi-Time Attention Networks for Irregularly Sampled Time Seriescited by 234
  10. 2019Latent ODEs for Irregularly-Sampled Time Seriescited by 182
  11. 2020Estimating Counterfactual Treatment Outcomes over Time Through Adversarially Balanced Representationscited by 154
  12. 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
2 more
  1. 2023Transformer-based Time-to-Event Prediction for Chronic Kidney Disease Deteriorationcited by 4
  2. 2025Development and Validation of a Dynamic Kidney Failure Prediction Model based on Deep Learning: A Real-World Study with External Validationcited by 3

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026The Patient is not a Moving Document: A World Model Training Paradigm for Longitudinal EHR
  2. 2026Clin-JEPA: A Multi-Phase Co-Training Framework for Joint-Embedding Predictive Pretraining on EHR Patient Trajectories
  3. 2026Learning Continuous Patient Trajectories from Electronic Health Records
  4. 2026LongMoE: Longitudinal Multimodal Learning via Trajectory-Aware Mixture-of-Experts
  5. 2026Toward World Modeling of Physiological Signals with Chaos-Theoretic Balancing and Latent Dynamics
  6. 2026Simulating clinical interventions with a generative multimodal model of human physiology
  7. 2026CalTwin: Towards Calibrated, Shift-Robust Medical World Models via Fisher-Information Regularisation
  8. 2026LLM4EHR: Aligning Clinical Time Series with Medical Event Sequences via Large Language Models