Paper Overview
Field: Machine Learning Authors: Jun Ni Du, Lukas Adamek, Maxim Kryukov, Flavio Dormont Published: 2026-08-22 arXiv: 2608.20315
Abstract
Prediction models on structured electronic health records (EHR) remain central to medical machine learning, yet few approaches jointly emphasize quantitative laboratory information and explainability relative to input medical events. This paper proposes BERT-LER, a BERT-style model encoding EHR timelines, pre-trained and fine-tuned on a de-identified EHR dataset of 75 million patients.
The model encodes laboratory test results as discrete tokens while preserving graded information via percentile-based binning. Combined with Integrated Gradients, it provides token-level attributions grounded in the input EHR sequence.
Evaluation
- EHRShot benchmark: competitive predictive performance against public baselines
- Asthma severity progression study (real-world data): strong results on lab-related tasks
- Explainability: token-level attributions align with clinically known risk factors
Conclusion
BERT-LER typically outperforms publicly available baseline models on lab-related tasks while offering clinically meaningful explanations. The architecture and interpretability methodology can be applied to many therapeutic areas and prediction tasks.
--- *Automatically collected on 2026-08-22*