Continuous Value Tokenization Improves Medical Event Foundation Models
McCann, K. A.; Shin, I.; Li, H.; White, D.; Melnick, E. R.; Iscoe, M. S.; Loza, A. J.
Show abstract
Medical foundation models convert patient records into token sequences for autoregressive prediction, but numeric values such as lab results, vital signs, and time intervals are typically discretized into bins, losing precision and misaligning with clinical thresholds. We trained decoder-only transformer models (47 million parameters) on MIMIC-IV data (364,627 patients; 375 million observations) to compare three tokenization strategies: Discrete (binned values), Continuous Factored (continuous values preserving sequence length), and Continuous Fused (continuous values fused with measurement-type tokens). We evaluated next-token prediction, numeric value prediction, and three clinical tasks: ED disposition at triage, ICD code prediction, and DRG prediction at discharge. Continuous Fused tokenization reduced median sequence length by 34\%, reached the Discrete model's final next-token loss in 30\% of training iterations, and improved numeric prediction accuracy by 30.25\% median nRMSE reduction. ICD code prediction favored Continuous Fused (AU-PRC 0.457 vs.\ 0.446; p < 0.001); DRG prediction was equivalent between Continuous Fused and Discrete; ED disposition accuracy was equivalent across all models ($\sim$0.900), though Discrete achieved better calibration. We additionally explain why predictive performance improves with Monte Carlo sample count and derive a scaling law to predict performance gains from increasing simulation budget. Continuous-value tokenization offers substantial efficiency and precision gains while maintaining comparable clinical task performance, with no modifications to the standard transformer architecture.
Matching journals
The top 1 journal accounts for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- EHR Foundation Models Improve Robustness in the Presence of Temporal Distribution Shift 95%
- Developing Machine Learning Models for Predicting Intensive Care Unit Resource Use During the COVID-19 Pandemic 94%
- Large Language Models Improve the Identification of Emergency Department Visits for Symptomatic Kidney Stones 94%
Similar papers in this journal
- Creating a computer assisted ICD coding system: performance metric choice and use of the ICD hierarchy 94%
- A methodology of phenotyping ICU patients from EHR data: high-fidelity, personalized, and interpretable phenotypes estimation 93%
- Mining for Equitable Health: Assessing the Impact of Missing Data in Electronic Health Records 93%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.