Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Haresh Rengaraj Rajamohan, Xiang Gao, Weicheng Zhu, Shih-Lun Huang, Long Chen, Gabe Schulman, Huizhen Jin, Shengduo Li, Yixuan Wang, Huidi Yang, Kyunghyun Cho, Cem M. Deniz, Narges Razavian• 2026

Related benchmarks

TaskDatasetResultRank
Disease onset forecastingPatient EHR dataset Dementia 2-year horizon (test)
AUROC78.9
5
Disease onset forecastingPatient EHR dataset Pancreatic Cancer 2-year horizon (test)
AUROC0.693
5
Disease onset forecastingPatient EHR dataset Prostate Cancer 2-year horizon (test)
AUROC91
5
Disease onset forecastingPatient EHR dataset Disease Suite Aggregate 2-year horizon (test)
AUROC0.78
5
Disease onset forecastingPatient EHR dataset Knee OA 2-year horizon (test)
AUROC0.726
5
Disease onset forecastingPatient EHR dataset COPD 2-year horizon (test)
AUROC69.1
5
Disease onset forecastingPatient EHR dataset CHF 2-year horizon (test)
AUROC85.7
5
Disease onset forecastingPatient EHR dataset Acute MI 2-year horizon (test)
AUROC79.3
5
Disease onset forecastingEHR 5-year horizon (test)
Macro AUROC0.752
5
Showing 9 of 9 rows

Other info

Follow for update