The Patient is not a Moving Document: A World Model Training Paradigm for Longitudinal EHR
On the map: World models → Medical and surgical world models
Abstract (arXiv)
Large language models (LLMs) trained with next-word-prediction have achieved success as clinical foundation models. Representations from these language backbones yield strong linear probe performance across biomedical tasks, suggesting that patient semantics emerge from next-token prediction at scale. However, this paradigm treats patients as a document to be summarized rather than a dynamical system to be simulated; a patient's trajectory emerges from their state evolving under interventions and time, requiring models that simulate dynamics rather than predict tokens. To address this, we introduce SMB-Structure, a world model for structured EHR that grounds a joint-embedding prediction architecture (JEPA) with next-token prediction (SFT). SFT grounds our model to reconstruct future patient states in token space, while JEPA predicts those futures in latent space from the initial patient representation alone, forcing trajectory dynamics to be encoded before the next state is observed. We validate across two large-scale cohorts: Memorial Sloan Kettering (23,319 oncology patients; 323,000+ patient-years) and INSPECT (19,402 pulmonary embolism patients). Using a linear probe evaluated at multiple points along the disease trajectory, we demonstrate that our training paradigm learns embeddings that capture disease dynamics not recoverable by autoregressive baselines, enabling SMB-Structure to achieve competitive performance on complex tasks characterized by high patient heterogeneity. Model weights are available at https://huggingface.co/standardmodelbio/SMB-v1-1.7B-Structure.
Key facts. Cited by 3 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 24 of its 33 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
24 of 33 references.
References · 24
- 2022Large Language Models Encode Clinical Knowledgecited by 1,807
- 2018World Modelscited by 1,444
- 2023LLaVA-Med: Training a Large Language-and-Vision Assistant for Biomedicine in One Daycited by 1,287
- 2023Mastering Diverse Domains through World Modelscited by 1,193
- 2019Publicly Available Clinical BERT Embeddingscited by 995
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 982
- 2025V-JEPA 2: Self-Supervised Video Models Enable Understanding, Prediction and Planningcited by 738
- 2024Revisiting Feature Prediction for Learning Visual Representations from Videocited by 490
- 2016DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Networkcited by 354
- 2023Causal Reasoning and Large Language Models: Opening a New Frontier for Causalitycited by 348
- 2019Transfer Learning in Biomedical Natural Language Processing: An Evaluation of BERT and ELMo on Ten Benchmarking Datasetscited by 298
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 227
12 more
- 2023Causal Parrots: Large Language Models May Talk Causality But Are Not Causalcited by 146
- 2023EHRSHOT: An EHR Benchmark for Few-Shot Evaluation of Foundation Modelscited by 101
- 2020Language Models Are An Effective Patient Representation Learning Technique For Electronic Health Record Datacited by 85
- 2022Causal Machine Learning for Healthcare and Precision Medicinecited by 68
- 2025VL-JEPA: Joint Embedding Predictive Architecture for Vision-languagecited by 61
- 2025LLM-JEPA: Large Language Models Meet Joint Embedding Predictive Architecturescited by 35
- 2022Can Foundation Models Talk Causality?cited by 30
- 2024Advancing High Resolution Vision-Language Models in Biomedicinecited by 5
- 2025Building the EHR Foundation Model via Next Event Predictioncited by 5
- 2025Zero-shot Medical Event Prediction Using a Generative Pre-trained Transformer on Electronic Health Recordscited by 3
- 2025Foundation Models for Clinical Records at Health System Scalecited by 3
- 2025Patient-specific Biomolecular Instruction Tuningcited by 1
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Medical world models: representing medical states, modelling clinical dynamics and guiding intervention policiescited by 3
- 2026Clin-JEPA: A Multi-Phase Co-Training Framework for Joint-Embedding Predictive Pretraining on EHR Patient Trajectoriescited by 2
- 2026Generated Context versus Governed State: Functional Conditions for Accountable Longitudinal Clinical Reasoningcited by 0
- 2026Generated Context versus Governed State: Functional Conditions for Accountable Longitudinal Clinical Reasoning14 Aug 2026
- 2026Medical world models: representing medical states, modelling clinical dynamics and guiding intervention policies15 Jun 2026
- 2026Clin-JEPA: A Multi-Phase Co-Training Framework for Joint-Embedding Predictive Pretraining on EHR Patient Trajectories11 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026EHRWorld: A Patient-Centric Medical World Model for Long-Horizon Clinical Trajectories
- 2026ChronoMedicalWorld: A Medical World Model for Learning Patient Trajectories from Longitudinal Care Data
- 2026Beyond Prediction: Longitudinal Reasoning in EHR-Integrated Clinical AI ↗
- 2026Future Querying: Can LLMs Serve as Implicit Medical World Models?
- 2026ChatHealthAI: Aligning Electronic Health Record Representations with Large Language Models for Grounded Clinical Reasoning
- 2026EHR-RAG: Bridging Long-Horizon Structured Electronic Health Records and Large Language Models via Enhanced Retrieval-Augmented Generation
- 2026Medical world models in healthcare: foundations, applications, and challenges for trustworthy clinical translation
- 2025Rethinking Tokenization for Clinical Time Series: When Less is More