BERT-LER : des modèles transformers explicables pour la prédiction clinique sur données EHR structurées

BERT-LER encode les résultats de tests de laboratoire en tokens discrets tout en préservant l'information graduée via un binning par percentiles, et associe cette représentation à des attributions token par token grâce aux Integrated Gradients. Testé sur le benchmark public EHRShot et sur une étude de progression de la sévérité de l'asthme, le modèle égale voire dépasse les benchmarks existants sur les tâches liées aux données de laboratoire, avec des explications alignées sur des facteurs de risque cliniques connus.