arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.14065cs.LG

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

Hasso Plattner Institute for Digital Health at Mount Sinai(西奈山哈索·普拉特纳数字健康研究所);Icahn School of Medicine at Mount Sinai(西奈山伊坎医学院);School of Informatics, Computing, and Engineering, Indiana University(印第安纳大学信息学、计算与工程学院);Dell Medical School, University of Texas at Austin(德克萨斯大学奥斯汀分校戴尔医学院);School of Informatics, University of Texas at Austin(德克萨斯大学奥斯汀分校信息学院)

Tingyi Wanyan, Hossein Honarvar, Ariful Azad, Ying Ding, Benjamin S. Glicksberg

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑