Learning Latent Space Representations to Predict Patient Outcomes

Abstract

Background: Scalable and accurate health outcome prediction using electronic health record data (EHR) has gained much attention in research recently. Previous machine learning models mostly ignore relations between different types of clinical data (i.e., laboratory, ICD, and medication).

Objective: In this study, we model such clinical data types and build predictive models using Intensive Care Units’ (ICUs) EHR data. We developed innovative neural network (NN) models and compared them with the widely used logistic regression model and other state-of-the-art NN models to predict patient’s mortality using his/her own longitudinal EHR data.

Methods: We built a set of NN models we collectively called as long short-term memory (LSTM) outcome prediction using comprehensive feature relations (CLOUT). Our CLOUT models use a correlational neural network model to identify a latent space representation between different types of discrete clinical features during a patient’s encounter and integrate the latent representation into a LSTM based predictive model framework. In addition, we designed an ablation experiment to identify risk factors from our CLOUT models. Using physicians’ input as the gold standard, we compared the risk factors identified by both CLOUT and logistic regression models.

Results: Experiments on the MIMIC-III (Medical Information Mart for Intensive Care) dataset show that CLOUT has surpassed logistic regression and other baseline NN models. Also, physicians’ agreement with the CLOUT-derived risk factor rankings was statistically significantly higher than the agreement with the logistic regression model.

Conclusions: Our results strongly support the applicability of CLOUT for real-world clinical use in identifying patients at high risk of mortality.

Publication
In the Journal of Medical Internet Research (JMIR)