Highrisk Prediction from Electronic Medical Records via Deep Attention Networks
arXiv:1712.00010
Abstract
Predicting highrisk vascular diseases is a significant issue in the medical domain. Most predicting methods predict the prognosis of patients from pathological and radiological measurements, which are expensive and require much time to be analyzed. Here we propose deep attention models that predict the onset of the high risky vascular disease from symbolic medical histories sequence of hypertension patients such as ICD-10 and pharmacy codes only, Medical History-based Prediction using Attention Network (MeHPAN). We demonstrate two types of attention models based on 1) bidirectional gated recurrent unit (R-MeHPAN) and 2) 1D convolutional multilayer model (C-MeHPAN). Two MeHPAN models are evaluated on approximately 50,000 hypertension patients with respect to precision, recall, f1-measure and area under the curve (AUC). Experimental results show that our MeHPAN methods outperform standard classification models. Comparing two MeHPANs, R-MeHPAN provides more better discriminative capability with respect to all metrics while C-MeHPAN presents much shorter training time with competitive accuracy.
arXiv commentAccepted poster at NIPS 2017 Workshop on Machine Learning for Health (https://ml4health.github.io/2017/)
References in corpus (10)
- Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling
- Show, Attend and Tell: Neural Image Caption Generation with Visual Attention
- Convolutional Sequence to Sequence Learning
- Deep Speech 2: End-to-End Speech Recognition in English and Mandarin
- Densely Connected Convolutional Networks
- A Structured Self-attentive Sentence Embedding
- Convolutional Neural Networks for Sentence Classification
- Language Modeling with Gated Convolutional Networks
- YOLO9000: Better, Faster, Stronger
- Deep Voice 2: Multi-Speaker Neural Text-to-Speech