Explainable Transformer Models for Clinical Prediction Tasks on Structured Electronic Health Records
Not on the map. Why?
Abstract (arXiv)
Predictive models over structured electronic health records (EHRs) remain central to machine learning for healthcare, but few have jointly emphasized quantitative laboratory information and interpretability with respect to input medical events. We present BERT-LER, a BERT-style model for coded EHR timelines pretrained and fine-tuned from a de-identified EHR dataset of 75 million patients, that encodes laboratory test results as discrete tokens while retaining graded information through percentile-based binning, paired with Integrated Gradients for token-level attributions grounded in the input EHR sequence. We benchmark our approach on the public EHRShot benchmark suite and on an asthma severity progression study based on real-world data. This addresses a methodological gap in EHR foundation-style modeling by unifying laboratory value representation and explainability in a single framework, while assessing whether both predictive performance and explanations generalize beyond standard clinical prediction tasks. Across EHRShot and asthma tasks, BERT-LER achieves predictive performance that is competitive with, and on laboratory-related tasks often exceeds, publicly available benchmark models, and provides attributions that align with clinically known risk factors. Our architecture and explainability approach can be applied to many therapeutic areas and prediction tasks using language models trained on structured EHRs.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 13 of its 52 references that are in the map. First posted on arXiv 20 Aug 2026.
Builds on
13 of 52 references.
References · 13
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2017Axiomatic Attribution for Deep Networkscited by 3,384
- 2018Self-Attention with Relative Position Representationscited by 1,120
- 2020Captum: A unified and generic model interpretability library for PyTorchcited by 547
- 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
- 2023EHRSHOT: An EHR Benchmark for Few-Shot Evaluation of Foundation Modelscited by 105
- 2021Hi-BEHRT: Hierarchical Transformer-based model for accurate prediction of clinical events using multimodal longitudinal electronic health recordscited by 54
- 2021An explainable Transformer-based deep learning model for the prediction of incident heart failurecited by 12
- 2023ExBEHRT: Extended Transformer for Electronic Health Records to Predict Disease Subtypes & Progressionscited by 12
Built on by
Not cited yet.