Scaling Recurrence-aware Foundation Models for Clinical Records via Next-Visit Prediction
About
While large-scale pretraining has revolutionized language modeling, its potential remains underexplored in healthcare with structured electronic health records (EHRs). We present RAVEN, a novel generative pretraining strategy for sequential EHR data based on Recurrence-Aware next-Visit EveNt prediction. Leveraging a dataset of over one million unique individuals, our model learns to autoregressively generate tokenized clinical events for the next visit conditioned on patient history. We introduce regularization on predicting repeated events and highlight a key pitfall in EHR-based foundation model evaluations: repeated event tokens can inflate performance metrics when new onsets are not distinguished from subsequent occurrences. Furthermore, we empirically investigate the scaling behaviors in a data-constrained, compute-saturated regime, showing that simply increasing model size is suboptimal without commensurate increases in data volume. We evaluate our model via zero-shot prediction for forecasting the incidence of a diverse set of diseases, where it rivals fully fine-tuned representation-based Transformer models and outperforms widely used simulation-based next-token approaches. Finally, without additional parameter updates, we show that RAVEN can generalize to an external patient cohort under lossy clinical code mappings and feature coverage gaps.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Disease onset forecasting | Patient EHR dataset Dementia 2-year horizon (test) | AUROC78.9 | 5 | |
| Disease onset forecasting | Patient EHR dataset Pancreatic Cancer 2-year horizon (test) | AUROC0.693 | 5 | |
| Disease onset forecasting | Patient EHR dataset Prostate Cancer 2-year horizon (test) | AUROC91 | 5 | |
| Disease onset forecasting | Patient EHR dataset Disease Suite Aggregate 2-year horizon (test) | AUROC0.78 | 5 | |
| Disease onset forecasting | Patient EHR dataset Knee OA 2-year horizon (test) | AUROC0.726 | 5 | |
| Disease onset forecasting | Patient EHR dataset COPD 2-year horizon (test) | AUROC69.1 | 5 | |
| Disease onset forecasting | Patient EHR dataset CHF 2-year horizon (test) | AUROC85.7 | 5 | |
| Disease onset forecasting | Patient EHR dataset Acute MI 2-year horizon (test) | AUROC79.3 | 5 | |
| Disease onset forecasting | EHR 5-year horizon (test) | Macro AUROC0.752 | 5 |