An Integrated LSTM-HeteroRGNN Model for Interpretable Opioid Overdose Risk Prediction

28 Pages Posted: 12 Jan 2022

See all articles by Xinyu Dong

Xinyu Dong

State University of New York (SUNY), Stony Brook

Rachel Wong

State University of New York (SUNY), Stony Brook

Weimin Lyu

State University of New York (SUNY), Stony Brook

Kayley Abell-Hart

State University of New York (SUNY), Stony Brook

Janos G. Hajagos

State University of New York (SUNY), Stony Brook

Richard N. Rosenthal

State University of New York (SUNY), Stony Brook

Chao Chen

State University of New York (SUNY), Stony Brook

Fusheng Wang

State University of New York (SUNY), Stony Brook University - Department of Biomedical Informatics

Abstract

Opioid overdose (OD) has become a leading cause of accidental death in the United States, and overdose deaths reached a record high during the COVID-19 pandemic. Combating the opioid crisis requires targeting high-need populations by identifying individuals at risk of OD. While deep learning emerges as a powerful method for building predictive models using large scale electronic health records (EHR), it is challenged by the complex intrinsic relationships among EHR data. Further, its utility is limited by the lack of clinically meaningful explainability, which is necessary for making informed clinical or policy decisions using such models. In this paper, we present LIGHTED, an integrated deep learning model combining long short term memory (LSTM) and graph neural networks (GNN) to predict patients’ OD risk. The LIGHTED model can incorporate the temporal effects of disease progression and the knowledge learned from interactions among clinical features. We evaluated the model using Cerner’s HealthFacts database with over 5 million patients. Our experiments demonstrated that the model outperforms traditional machine learning methods and other deep learning models. We also proposed a novel interpretability method by exploiting embeddings provided by GNNs and clustering patients and EHR features respectively, then conducted qualitative feature cluster analysis for clinical interpretations. Our study shows that LIGHTED can take advantage of longitudinal EHR data and the intrinsic graph structure of EHRs among patients to provide effective and interpretable OD risk predictions that may potentially improve clinical decision support.

Note:
Funding Information: This work was funded partially by the Stony Brook University OVPR Seed Grant 1158484-63845-6.

Declaration of Interests: The authors do not have any conflicts of interest to disclose.

Keywords: Opioid Overdose, Opioid Poisoning, deep learning, Clinical Decision Support, Electronic Health Records, Long Short-Term Memory, Graph Neural Network

Suggested Citation

Dong, Xinyu and Wong, Rachel and Lyu, Weimin and Abell-Hart, Kayley and Hajagos, Janos G. and Rosenthal, Richard N. and Chen, Chao and Wang, Fusheng, An Integrated LSTM-HeteroRGNN Model for Interpretable Opioid Overdose Risk Prediction. AIIM-D-22-00049, Available at SSRN: https://ssrn.com/abstract=4007155 or http://dx.doi.org/10.2139/ssrn.4007155

Xinyu Dong

State University of New York (SUNY), Stony Brook ( email )

Health Science Center
Stony Brook, NY 11794
United States

Rachel Wong

State University of New York (SUNY), Stony Brook ( email )

Health Science Center
Stony Brook, NY 11794
United States

Weimin Lyu

State University of New York (SUNY), Stony Brook ( email )

Health Science Center
Stony Brook, NY 11794
United States

Kayley Abell-Hart

State University of New York (SUNY), Stony Brook ( email )

Health Science Center
Stony Brook, NY 11794
United States

Janos G. Hajagos

State University of New York (SUNY), Stony Brook ( email )

Health Science Center
Stony Brook, NY 11794
United States

Richard N. Rosenthal

State University of New York (SUNY), Stony Brook ( email )

Health Science Center
Stony Brook, NY 11794
United States

Chao Chen

State University of New York (SUNY), Stony Brook ( email )

Health Science Center
Stony Brook, NY 11794
United States

Fusheng Wang (Contact Author)

State University of New York (SUNY), Stony Brook University - Department of Biomedical Informatics ( email )

Stony Brook, NY 11794
United States

Do you have a job opening that you would like to promote on SSRN?

Paper statistics

Downloads
52
Abstract Views
4,878
rank
515,880
PlumX Metrics