mlsanityData to 14 Oct 2026

Tokenization Tradeoffs in Structured EHR Foundation Models

arXiv 2603.15644 ↗First posted 3 Mar 2026cs.LG cs.CLcited by 6

Not on the map. Why?

Abstract (arXiv)

Foundation models for structured electronic health records (EHRs) are pretrained on longitudinal sequences of timestamped clinical events to learn adaptable patient representations. Tokenization -- how these timelines are converted into discrete model inputs -- determines what information is preserved, how efficiently it is encoded, and which relationships must be learned versus precomputed. Yet the impact of tokenization design choices on downstream performance and computational efficiency remains largely unexplored. Here, we pretrained a transformer on pediatric EHR data under a factorial design, varying tokenization along event encoding, time encoding, and workflow annotation. We evaluated area-under-the-receiver-operating-characteristic curve across 74 clinical prediction tasks. Joint event encoding and positional time encoding outperformed their alternatives (73/74 and 71/74 tasks) while requiring 39.5% and 9.6% fewer pretraining floating-point operations, respectively. Targeted ablations traced the joint encoding advantage to local binding efficiency, that is, code-attribute pairs are combined into single tokens, rather than split across tokens that the model must learn to associate during pretraining. External evaluation on an adult intensive care unit cohort demonstrated that this advantage generalizes despite substantial vocabulary mismatch, while temporal and workflow effects remain institution-specific. These results establish tokenization as a tractable lever for improving both the performance and efficiency of EHR foundation models.

Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 20 of its 39 references that are in the map. First posted on arXiv 3 Mar 2026.

Builds on

20 of 39 references.

References · 20

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  4. 2020Scaling Laws for Neural Language Modelscited by 6,586
  5. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  6. 2019Generating Long Sequences with Sparse Transformerscited by 1,546
  7. 2024Smarter, Better, Faster, Longer: A Modern Bidirectional Encoder for Fast, Memory Efficient, and Long Context Finetuning and Inferencecited by 554
  8. 2023Scaling Data-Constrained Language Modelscited by 403
  9. 2020Med-BERT: pre-trained contextualized embeddings on large-scale structured electronic health records for disease predictioncited by 373
  10. 2020On the Binding Problem in Artificial Neural Networkscited by 233
  11. 2023EHRSHOT: An EHR Benchmark for Few-Shot Evaluation of Foundation Modelscited by 105
  12. 2024Tokenization Is More Than Compressioncited by 89
8 more
  1. 2020Language Models Are An Effective Patient Representation Learning Technique For Electronic Health Record Datacited by 87
  2. 2024Zero Shot Health Trajectory Prediction Using Transformercited by 77
  3. 2021CEHR-BERT: Incorporating temporal information from structured EHR data to improve prediction taskscited by 76
  4. 2024Unpacking Tokenization: Evaluating Text Compression and its Correlation with Model Performancecited by 61
  5. 2024Context Clues: Evaluating Long Context Models for Clinical Prediction Tasks on EHRscited by 39
  6. 2022GenHPF: General Healthcare Predictive Framework with Multi-task Multi-source Learningcited by 36
  7. 2023A Multi-Center Study on the Adaptability of a Shared Foundation Model for Electronic Health Recordscited by 31
  8. 2025Foundation Model of Electronic Medical Records for Adaptive Risk Estimationcited by 19

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 6.

6 papers.