Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

The Patient is not a Moving Document: A World Model Training Paradigm for Longitudinal EHR

About

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.

Irsyad Adam, Zekai Chen, David Laprade, Shaun Porwal, David Laub, Erik Reinertsen, Arda Pekis, Kevin Brown• 2026

Related benchmarks

TaskDatasetResultRank
Cancer Indication PredictionMSK cohort oncology
AUC-ROC Bladder0.778
15
Mortality PredictionINSPECT cohort PE 1.0 (30 day)
AUC-ROC0.806
12
Mortality PredictionINSPECT cohort (PE) 1.0 (180 day)
AUC-ROC0.803
12
Mortality PredictionINSPECT cohort (PE) 365-day 1.0
AUC-ROC81
12
PH predictionINSPECT cohort PE 1.0 (365-day)
AUC-ROC65.3
12
Readmission predictionINSPECT cohort PE 1.0 (30 day)
AUC-ROC69.1
12
Readmission predictionINSPECT cohort (PE) 1.0 (180 day)
AUC-ROC0.681
12
Readmission predictionINSPECT cohort (PE) 365-day 1.0
AUC-ROC68
12
Showing 8 of 8 rows

Other info

Follow for update