About this item:

101 Views | 63 Downloads

Author Notes:

Correspondence: maywang@bme.gatech.edu

The authors are grateful to Li Tong and Janani Venugopalan for their valuable comments and suggestions.

Subjects:

Research Funding:

None declared

Keywords:

  • Science & Technology
  • Life Sciences & Biomedicine
  • Mathematical & Computational Biology
  • Medical Informatics
  • Health care
  • electronic health records
  • deep learning
  • recurrent neural networks
  • attention
  • interpretability
  • visualization

Interpretable Predictions of Clinical Outcomes with An Attention-based Recurrent Neural Network

Tools:

Journal Title:

ACM-BCB '17: Proceedings of the 8th ACM International Conference on Bioinformatics, Computational Biology,and Health Informatics

Volume:

Volume 2017

Publisher:

, Pages 233-240

Type of Work:

Article | Post-print: After Peer Review

Abstract:

The increasing accumulation of healthcare data provides researchers with ample opportunities to build machine learning approaches for clinical decision support and to improve the quality of health care. Several studies have developed conventional machine learning approaches that rely heavily on manual feature engineering and result in task-specific models for health care. In contrast, healthcare researchers have begun to use deep learning, which has emerged as a revolutionary machine learning technique that obviates manual feature engineering but still achieves impressive results in research fields such as image classification. However, few of them have addressed the lack of the interpretability of deep learning models although interpretability is essential for the successful adoption of machine learning approaches by healthcare communities. In addition, the unique characteristics of healthcare data such as high dimensionality and temporal dependencies pose challenges for building models on healthcare data. To address these challenges, we develop a gated recurrent unit-based recurrent neural network with hierarchical attention for mortality prediction, and then, using the diagnostic codes from the Medical Information Mart for Intensive Care, we evaluate the model. We find that the prediction accuracy of the model outperforms baseline models and demonstrate the interpretability of the model in visualizations.

Copyright information:

© 2017 ACM, Inc.

Export to EndNote