Log In Sign Up

Deep Learning with Heterogeneous Graph Embeddings for Mortality Prediction from Electronic Health Records

by   Tingyi Wanyan, et al.

Computational prediction of in-hospital mortality in the setting of an intensive care unit can help clinical practitioners to guide care and make early decisions for interventions. As clinical data are complex and varied in their structure and components, continued innovation of modeling strategies is required to identify architectures that can best model outcomes. In this work, we train a Heterogeneous Graph Model (HGM) on Electronic Health Record data and use the resulting embedding vector as additional information added to a Convolutional Neural Network (CNN) model for predicting in-hospital mortality. We show that the additional information provided by including time as a vector in the embedding captures the relationships between medical concepts, lab tests, and diagnoses, which enhances predictive performance. We find that adding HGM to a CNN model increases the mortality prediction accuracy up to 4%. This framework serves as a foundation for future experiments involving different EHR data types on important healthcare prediction tasks.


page 1

page 2

page 3

page 4


Multi-Task Prediction of Clinical Outcomes in the Intensive Care Unit using Flexible Multimodal Transformers

Recent deep learning research based on Transformer model architectures h...

Unsupervised Probabilistic Models for Sequential Electronic Health Records

We develop an unsupervised probabilistic model for heterogeneous Electro...

Evaluation of Embeddings of Laboratory Test Codes for Patients at a Cancer Center

Laboratory test results are an important and generally highly dimensiona...

Precisely Predicting Acute Kidney Injury with Convolutional Neural Network Based on Electronic Health Record Data

The incidence of Acute Kidney Injury (AKI) commonly happens in the Inten...

Feature Engineering Combined with 1 D Convolutional Neural Network for Improved Mortality Prediction

The intensive care units (ICUs) are responsible for generating a wealth ...

Modeling Irregularly Sampled Clinical Time Series

While the volume of electronic health records (EHR) data continues to gr...