Imputation of Unknown Missingness in Sparse Electronic Health Records

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Han, Jun, Nassar, Josue, Batra, Sanjit Singh, Cordova-Palomera, Aldo, Nori, Vijay, Tillman, Robert E.
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908849764564992
author Han, Jun
Nassar, Josue
Batra, Sanjit Singh
Cordova-Palomera, Aldo
Nori, Vijay
Tillman, Robert E.
author_facet Han, Jun
Nassar, Josue
Batra, Sanjit Singh
Cordova-Palomera, Aldo
Nori, Vijay
Tillman, Robert E.
contents Machine learning holds great promise for advancing the field of medicine, with electronic health records (EHRs) serving as a primary data source. However, EHRs are often sparse and contain missing data due to various challenges and limitations in data collection and sharing between healthcare providers. Existing techniques for imputing missing values predominantly focus on known unknowns, such as missing or unavailable values of lab test results; most do not explicitly address situations where it is difficult to distinguish what is missing. For instance, a missing diagnosis code in an EHR could signify either that the patient has not been diagnosed with the condition or that a diagnosis was made, but not shared by a provider. Such situations fall into the paradigm of unknown unknowns. To address this challenge, we develop a general purpose algorithm for denoising data to recover unknown missing values in binary EHRs. We design a transformer-based denoising neural network where the output is thresholded adaptively to recover values in cases where we predict data are missing. Our results demonstrate improved accuracy in denoising medical codes within a real EHR dataset compared to existing imputation approaches and leads to increased performance on downstream tasks using the denoised data. In particular, when applying our method to a real world application, predicting hospital readmission from EHRs, our method achieves statistically significant improvement over all existing baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2602_20442
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Imputation of Unknown Missingness in Sparse Electronic Health Records
Han, Jun
Nassar, Josue
Batra, Sanjit Singh
Cordova-Palomera, Aldo
Nori, Vijay
Tillman, Robert E.
Machine Learning
Artificial Intelligence
Machine learning holds great promise for advancing the field of medicine, with electronic health records (EHRs) serving as a primary data source. However, EHRs are often sparse and contain missing data due to various challenges and limitations in data collection and sharing between healthcare providers. Existing techniques for imputing missing values predominantly focus on known unknowns, such as missing or unavailable values of lab test results; most do not explicitly address situations where it is difficult to distinguish what is missing. For instance, a missing diagnosis code in an EHR could signify either that the patient has not been diagnosed with the condition or that a diagnosis was made, but not shared by a provider. Such situations fall into the paradigm of unknown unknowns. To address this challenge, we develop a general purpose algorithm for denoising data to recover unknown missing values in binary EHRs. We design a transformer-based denoising neural network where the output is thresholded adaptively to recover values in cases where we predict data are missing. Our results demonstrate improved accuracy in denoising medical codes within a real EHR dataset compared to existing imputation approaches and leads to increased performance on downstream tasks using the denoised data. In particular, when applying our method to a real world application, predicting hospital readmission from EHRs, our method achieves statistically significant improvement over all existing baselines.
title Imputation of Unknown Missingness in Sparse Electronic Health Records
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2602.20442