Deep Interpretable Mortality Model for Intensive Care Unit Risk Prediction

No Thumbnail Available
File version
Author(s)
Shi, Z
Chen, W
Liang, S
Zuo, W
Yue, L
Wang, S
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2019
Size
File type(s)
Location

Dalian, China

License
Abstract

Estimating the mortality of patients plays a fundamental role in an intensive care unit (ICU). Currently, most learning approaches are based on deep learning models. However, these approaches in mortality prediction suffer from two problems: (i) the specificity of causes of death are not considered in the learning process due to the different diseases, and symptoms are mixed-used without diversification and localization; (ii) the learning outcome for the mortality prediction is not self-explainable for the clinicians. In this paper, we propose a Deep Interpretable Mortality Model (DIMM), which employs Multi-Source Embedding, Gated Recurrent Units (GRU), Attention mechanism and Focal Loss techniques to prognosticate mortality prediction. We intensified the mortality prediction by considering the different clinical measures, medical treatments and the heterogeneity of the disease. More importantly, for the first time, in this framework, we use a separate evidence-based interpreter named Highlighter to interpret the prediction model, which makes the prediction understandable and trustworthy to clinicians. We demonstrate that our approach achieves state-of-the-art performance in mortality prediction and can get an interpretable prediction on four different diseases.

Journal Title
Conference Title

Advanced Data Mining and Applications

Book Title
Edition
Volume

11888

Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject

Artificial intelligence

Persistent link to this record
Citation

Shi, Z; Chen, W; Liang, S; Zuo, W; Yue, L; Wang, S, Deep Interpretable Mortality Model for Intensive Care Unit Risk Prediction, Advanced Data Mining and Applications , 2019, 11888, pp. 617-631