Skip to main content

Gradient Importance Learning for Incomplete Observations

Publication ,  Journal Article
Gao, Q; Wang, D; Amason, JD; Yuan, S; Tao, C; Henao, R; Hadziahmetovic, M; Carin, L; Pajic, M
July 5, 2021

Though recent works have developed methods that can generate estimates (or imputations) of the missing entries in a dataset to facilitate downstream analysis, most depend on assumptions that may not align with real-world applications and could suffer from poor performance in subsequent tasks such as classification. This is particularly true if the data have large missingness rates or a small sample size. More importantly, the imputation error could be propagated into the prediction step that follows, which may constrain the capabilities of the prediction model. In this work, we introduce the gradient importance learning (GIL) method to train multilayer perceptrons (MLPs) and long short-term memories (LSTMs) to directly perform inference from inputs containing missing values without imputation. Specifically, we employ reinforcement learning (RL) to adjust the gradients used to train these models via back-propagation. This allows the model to exploit the underlying information behind missingness patterns. We test the approach on real-world time-series (i.e., MIMIC-III), tabular data obtained from an eye clinic, and a standard dataset (i.e., MNIST), where our imputation-free predictions outperform the traditional two-step imputation-based predictions using state-of-the-art imputation methods.

Duke Scholars

Publication Date

July 5, 2021
 

Citation

APA
Chicago
ICMJE
MLA
NLM
Gao, Q., Wang, D., Amason, J. D., Yuan, S., Tao, C., Henao, R., … Pajic, M. (2021). Gradient Importance Learning for Incomplete Observations.
Gao, Qitong, Dong Wang, Joshua D. Amason, Siyang Yuan, Chenyang Tao, Ricardo Henao, Majda Hadziahmetovic, Lawrence Carin, and Miroslav Pajic. “Gradient Importance Learning for Incomplete Observations,” July 5, 2021.
Gao Q, Wang D, Amason JD, Yuan S, Tao C, Henao R, et al. Gradient Importance Learning for Incomplete Observations. 2021 Jul 5;
Gao Q, Wang D, Amason JD, Yuan S, Tao C, Henao R, Hadziahmetovic M, Carin L, Pajic M. Gradient Importance Learning for Incomplete Observations. 2021 Jul 5;

Publication Date

July 5, 2021