Representation Before Training: A Practical Benchmark for Generative Medical Event Model Tokenization
Not on the map. Why?
Abstract (arXiv)
Generative medical event models use tokenized sequences of patient timelines as input, but practical guidance on the many decisions around tokenization is limited. We benchmark quantization granularity, reference-range anchoring, code--value fusion, numeric and temporal encodings, and native versus harmonized event representations from an expert-mapped common data model. Using both Llama and Qwen architectures, 156 models were trained on full hospitalizations from three initialization seeds, with each configuration following a shared training recipe for up to five epochs. We evaluated learned representations from the first 24 hours of hospitalization with linear probes to predict binary and continuous outcomes during hours 24-48. Fused tokens pairing codes with value deciles increased performance across all eight outcome families relative to the equivalent unfused tokenized input with area under the receiver operating characteristic curve (AUROC) gains of +0.002 to +0.033 and Spearman correlation gains of +0.025 to +0.114. Neither anchoring value bins to reference ranges nor increasing quantization granularity consistently improved performance, while xVal variants underperformed both discrete and soft encodings. Alternatives to explicit time tokens, such as event-order and admission-relative rotary position embeddings (RoPE), yielded higher family-mean point estimates across all eight families while reducing input length. When evaluating native input against input mapped to the Common Longitudinal Intensive Care Unit Data Format (CLIF), the CLIF full-hospitalization training sequences contained 28.6% as many tokens as the native sequences and improved performance across six of eight outcome families. These findings show that tokenization and event encoding are consequential design choices when learning patient representations for downstream classification and regression tasks.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 31 of its 41 references that are in the map. First posted on arXiv 18 Apr 2026.
Builds on
31 of 41 references.
References · 31
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 373
- 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 349
- 2019BEHRT: Transformer for Electronic Health Recordscited by 263
- 2019Time2Vec: Learning a Vector Representation of Timecited by 213
- 2015Doctor AI: Predicting Clinical Events via Recurrent Neural Networkscited by 208
- 2013Zero-Shot Learning by Convex Combination of Semantic Embeddingscited by 135
- 2017Dipole: Diagnosis Prediction in Healthcare via Attention-based Bidirectional Recurrent Neural Networkscited by 116
- 2023EHRSHOT: An EHR Benchmark for Few-Shot Evaluation of Foundation Modelscited by 105
- 2020Language Models Are An Effective Patient Representation Learning Technique For Electronic Health Record Datacited by 87
- 2021Representing Numbers in NLP: a Survey and a Visioncited by 83
19 more
- 2023MOTOR: A Time-To-Event Foundation Model For Structured Medical Recordscited by 78
- 2024Zero Shot Health Trajectory Prediction Using Transformercited by 77
- 2021CEHR-BERT: Incorporating temporal information from structured EHR data to improve prediction taskscited by 76
- 2023Event Stream GPT: A Data Pre-processing and Modeling Library for Generative, Pre-trained Transformers over Continuous-time Sequences of Complex Eventscited by 46
- 2021HiRID-ICU-Benchmark -- A Comprehensive Machine Learning Benchmark on High-resolution ICU Datacited by 39
- 2024EHRMamba: Towards Generalizable and Scalable Foundation Models for Electronic Health Recordscited by 32
- 2023xVal: A Continuous Numerical Tokenization for Scientific Language Modelscited by 27
- 2025Large Language Models are Powerful Electronic Health Record Encoderscited by 26
- 2021Unifying Heterogeneous Electronic Health Records Systems via Text-Based Code Embeddingcited by 24
- 2023DuETT: Dual Event Time Transformer for Electronic Health Recordscited by 17
- 2025CLIMB: Data Foundations for Large Scale Multimodal Clinical Foundation Modelscited by 13
- 2025Continuous Autoregressive Language Modelscited by 10
- 2025EHRStruct: A Comprehensive Benchmark Framework for Evaluating Large Language Models on Structured Electronic Health Record Taskscited by 6
- 2026Tokenization Tradeoffs in Structured EHR Foundation Modelscited by 6
- 2024Representation Learning of Structured Data for Medical Foundation Modelscited by 5
- 2024Temporal Cross-Attention for Dynamic Embedding and Tokenization of Multimodal Electronic Health Recordscited by 3
- 2025Quantifying surprise in clinical care: Detecting highly informative events in electronic health records with foundation modelscited by 3
- 2025Rethinking Tokenization for Clinical Time Series: When Less is Morecited by 3
- 2022Ontologizing Health Systems Data at Scale: Making Translational Discovery a Realitycited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026The Patient is not a Moving Document: A World Model Training Paradigm for Longitudinal EHR
- 2026From Token to Token Pair: Efficient Prompt Compression for Large Language Models in Clinical Prediction
- 2025Beyond Generative AI: World Models for Clinical Prediction, Counterfactuals, and Planning
- 2026Efficient Generative Prediction for EHR Foundation Models: The SCOPE and REACH Estimators
- 2026Tokenizer-Generator Coupling in Medical Image Generation
- 2025Quantized Large Language Models in Biomedical Natural Language Processing: Evaluation and Recommendation
- 2026What Makes a Good Medical Image Tokenizer? Rethinking Reconstruction and Generation in Medical Image Tokenization
- 2026LLM4EHR: Aligning Clinical Time Series with Medical Event Sequences via Large Language Models