research-article

Can Attention Be Used to Explain EHR-Based Mortality Prediction Tasks: A Case Study on Hemorrhagic Stroke

Authors:

Forhan Bin Emdad,

Zhe HeAuthors Info & Claims

BCB '23: Proceedings of the 14th ACM International Conference on Bioinformatics, Computational Biology, and Health Informatics

Article No.: 26, Pages 1 - 6

https://doi.org/10.1145/3584371.3613002

Published: 04 October 2023 Publication History

Abstract

Stroke is a significant cause of mortality and morbidity, necessitating early predictive strategies to minimize risks. Traditional methods for evaluating patients, such as Acute Physiology and Chronic Health Evaluation (APACHE II, IV) and Simplified Acute Physiology Score III (SAPS III), have limited accuracy and interpretability. This paper proposes a novel approach: an interpretable, attention-based transformer model for early stroke mortality prediction. This model seeks to address the limitations of previous predictive models, providing both interpretability (providing clear, understandable explanations of the model) and fidelity (giving a truthful explanation of the model's dynamics from input to output). Furthermore, the study explores and compares fidelity and interpretability scores using Shapley values and attention-based scores to improve model explainability. The research objectives include designing an interpretable attention-based transformer model, evaluating its performance compared to existing models, and providing feature importance derived from the model.

References

[1]

Yuhan Deng et al. 2022. Explainable time-series deep learning models for the prediction of mortality, prolonged length of stay and 30-day readmission in intensive care patients. Frontiers in Medicine 9 (2022).

[2]

Forhan Bin Emdad et al. 2023. Towards A Unified Utilitarian Ethics Framework for Healthcare Artificial Intelligence. (2023).

[3]

Forhan Bin Emdad et al. 2023. Towards Interpretable Multimodal Predictive Models for Early Mortality Prediction of Hemorrhagic Stroke Patients. AMIA Summits on Translational Science Proceedings 2023 (2023), 128--137.

[4]

Jake Grigsby et al. 2021. Long-range transformers for dynamic spatiotemporal forecasting. arXiv preprint arXiv:2109.12218 (2021).

[5]

Gaspard Harerimana et al. 2021. A deep attention model to forecast the Length Of Stay and the in-hospital mortality right on admission from ICD codes and demographic data. Journal of Biomedical Informatics 118 (2021), 103778.

Digital Library

[6]

Jay Heo et al. 2018. Uncertainty-aware attention for reliable interpretation and prediction. Advances in neural information processing systems 31 (2018).

[7]

Sepp Hochreiter and Jürgen Schmidhuber. 1997. Long short-term memory. Neural computation 9, 8 (1997), 1735--1780.

Digital Library

[8]

Ming-Hsia Hu et al. 2010. Early and intensive rehabilitation predicts good functional outcomes in patients admitted to the stroke intensive care unit. Disability and Rehabilitation 32, 15 (2010), 1251--1259.

[9]

Jiann-Shing Jeng et al. 2008. Predictors of survival and functional outcome in acute stroke patients admitted to the stroke intensive care unit. Journal of the neurological sciences 270, 1--2 (2008), 60--66.

[10]

Eunji Jun et al. 2019. Stochastic imputation and uncertainty-aware attention to EHR for mortality prediction. In 2019 International Joint Conference on Neural Networks (IJCNN). IEEE, 1--7.

[11]

William A Knaus et al. 1991. The APACHE III prognostic system: risk prediction of hospital mortality for critically III hospitalized adults. Chest 100, 6 (1991), 1619--1636.

[12]

Ninghao Liu et al. 2022. Interpretability in Graph Neural Networks. Graph Neural Networks: Foundations, Frontiers, and Applications (2022), 121--147.

[13]

Hui Wen Loh et al. 2022. Application of explainable artificial intelligence for healthcare: A systematic review of the last decade (2011--2022). Computer Methods and Programs in Biomedicine (2022), 107161.

[14]

Scott M Lundberg and Su-In Lee. 2017. A Unified Approach to Interpreting Model Predictions. In Advances in Neural Information Processing Systems 30, I. Guyon, U. V. Luxburg, S. Bengio, H. Wallach, R. Fergus, S. Vishwanathan, and R. Garnett (Eds.). Curran Associates, Inc., 4765--4774.

[15]

Wilson E Marcílio and Danilo M Eler. 2020. From explanations to feature selection: assessing SHAP values as feature selection mechanism. In 2020 33rd SIBGRAPI conference on Graphics, Patterns and Images (SIBGRAPI). Ieee, 340--347.

[16]

Aniek F Markus et al. 2021. The role of explainability in creating trustworthy artificial intelligence for health care: a comprehensive survey of the terminology, design choices, and evaluation strategies. Journal of Biomedical Informatics 113 (2021), 103655.

Digital Library

[17]

Writing Group Members et al. 2009. Heart disease and stroke statistics---2009 update: a report from the American Heart Association Statistics Committee and Stroke Statistics Subcommittee. Circulation 119, 3 (2009), e21--e181.

[18]

Seyedeh Neelufar Payrovnaziri et al. 2020. Explainable artificial intelligence models using real-world electronic health record data: a systematic scoping review. Journal of the American Medical Informatics Association 27, 7 (2020), 1173--1185.

[19]

Juliana Tolles and William J Meurer. 2016. Logistic regression: relating patient characteristics to outcomes. Jama 316, 5 (2016), 533--534.

[20]

Petar Veličković et al. 2017. Graph attention networks. arXiv preprint arXiv:1710.10903 (2017).

Cited By

Fantozzi PNaldi M(2024)The Explainability of Transformers: Current Status and DirectionsComputers10.3390/computers1304009213:4(92)Online publication date: 4-Apr-2024
https://doi.org/10.3390/computers13040092
Ding JThao PPeng WWang JChug CHsieh MTseng YChen LLuo DWu CWang CHsu CChen YChen PLiu FHung F(2024)Large language multimodal models for new-onset type 2 diabetes prediction using five-year cohort electronic health recordsScientific Reports10.1038/s41598-024-71020-214:1Online publication date: 6-Sep-2024
https://doi.org/10.1038/s41598-024-71020-2
Rahman MEmdad FIslam CHe Z(2024)Uncovering the Interplay of Demographics and Healthcare Provider Availability on CMS HCC Risk Scores for Disabled BeneficiariesAdvances in Digital Health and Medical Bioengineering10.1007/978-3-031-62520-6_66(593-600)Online publication date: 31-Aug-2024
https://doi.org/10.1007/978-3-031-62520-6_66

Index Terms

Can Attention Be Used to Explain EHR-Based Mortality Prediction Tasks: A Case Study on Hemorrhagic Stroke
1. Computing methodologies
  1. Machine learning
    1. Machine learning approaches
      1. Neural networks

Recommendations

Mortality Prediction in ICU Patients Suffering from Stroke
SETN '22: Proceedings of the 12th Hellenic Conference on Artificial Intelligence

Ischemic stroke is a medical emergency that requires hospitalization and occasionally, specialized care at the Intensive Care Unit. Mortality prediction in the ICUs has been a challenge for intensivists, since prompt identification could impact medical ...
EHR coding with hybrid attention and features propagation on disease knowledge graph
Abstract
And sentences associated with these attributes and relationships have been neglected. in this paper ►We propose an end-to-end model called Knowledge Graph Enhanced neural network (KGENet) to address the above shortcomings. specifically ►We first ...
Highlights
- Disease Knowledge Graph with multi-view attributes and relationships.
- Hybrid Attention Module for label-specific feature extraction.
- Graph Propagation Module for interactive feature extraction.
- Attribute-aware and relationship-...
Mortality prediction using medical time series on TBI patients
Abstract Background and objective
Traumatic Brain Injury (TBI) is one of the leading causes of injury-related mortality in the world, with severe cases reaching mortality rates of 30-40%. It is highly heterogeneous both in causes and consequences making ...
Highlights
- The learning models developed for the mortality prediction achieved promising performance results.
- The study of the features with high predictive value improves the interpretability of the models.
- The developed models can be quite ...

Comments

Information & Contributors

Information

Published In

cover image ACM Conferences

BCB '23: Proceedings of the 14th ACM International Conference on Bioinformatics, Computational Biology, and Health Informatics

September 2023

626 pages

ISBN:9798400701269

DOI:10.1145/3584371

General Chairs:
May D. Wang
SGeorgia Institute of Technology
,
Byung-Jun Yoon

Copyright © 2023 ACM.

Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

Sponsors

SIGBio: ACM Special Interest Group on Bioinformatics

Publisher

Association for Computing Machinery

New York, NY, United States

Publication History

Published: 04 October 2023

Permissions

Request permissions for this article.

Request Permissions

Check for updates

Author Tags

Qualifiers

Research-article

Funding Sources

National Center for Advancing Translational Sciences
National Library of Medicine

Conference

BCB '23

Sponsor:

SIGBio

BCB '23: 14th ACM International Conference on Bioinformatics, Computational Biology, and Health Informatics

September 3 - 6, 2023

TX, Houston, USA

Acceptance Rates

Overall Acceptance Rate 254 of 885 submissions, 29%

Upcoming Conference

Contributors

Other Metrics

View Article Metrics

Bibliometrics & Citations

Bibliometrics

Article Metrics

2
Total Citations
View Citations
127
Total Downloads

Downloads (Last 12 months)127
Downloads (Last 6 weeks)11

Reflects downloads up to 15 Oct 2024

Other Metrics

View Author Metrics

Citations

Cited By

Fantozzi PNaldi M(2024)The Explainability of Transformers: Current Status and DirectionsComputers10.3390/computers1304009213:4(92)Online publication date: 4-Apr-2024
https://doi.org/10.3390/computers13040092
Ding JThao PPeng WWang JChug CHsieh MTseng YChen LLuo DWu CWang CHsu CChen YChen PLiu FHung F(2024)Large language multimodal models for new-onset type 2 diabetes prediction using five-year cohort electronic health recordsScientific Reports10.1038/s41598-024-71020-214:1Online publication date: 6-Sep-2024
https://doi.org/10.1038/s41598-024-71020-2
Rahman MEmdad FIslam CHe Z(2024)Uncovering the Interplay of Demographics and Healthcare Provider Availability on CMS HCC Risk Scores for Disabled BeneficiariesAdvances in Digital Health and Medical Bioengineering10.1007/978-3-031-62520-6_66(593-600)Online publication date: 31-Aug-2024
https://doi.org/10.1007/978-3-031-62520-6_66

View Options

Get Access

Login options

Check if you have access through your login credentials or your institution to get full access on this article.

Full Access

Get this Publication

View options

PDF

View or Download as a PDF file.

eReader

View online with eReader.

Media

Figures

Other

Tables

View Table of Contents