DCMN: Double Core Memory Network for Patient Outcome Prediction with Multimodal Data

More and more healthcare data are becoming readily available nowadays. These data can help the healthcare professionals and patient themselves to better understand the patient status and potentially lead to improved care quality. However, the analysis of these data are challenging because they are large-scale and heterogeneous, high-dimensional and sparse, temporal but irregularly sampled. In this paper, we propose a method called Double Core Memory Networks (DCMN) to integrate information from different modalities of the longitudinal patient data and learn a joint patient representation effective for downstream analytical tasks such as risk prediction. DCMN is designed not only to disentangle the temporal and non-linear intra-modal dependencies for the data within each modality but also to capture the long-term inter-modal interactions. DCMN models are the end-to-end memory networks with two external memory cores where each modality of data is compressed and stored. Each memory core has an information-flow controller named query to interact with an external memory module. In addition, we incorporate a gating mechanism into basic DCMN model to perform dynamic regulation of memory interaction. DCMN models have multiple computational layers (hops) allowing data of different modalities interacting with each other recurrently along with a mechanism of alternating access of external memory for each memory core hop-by-hop. We evaluate DCMN models on two outcome prediction tasks, including a mortality prediction on the public Medical Information Mart for Intensive Care III (MIMIC-III) database and a cost prediction on the Hospital Quality Monitoring System (HQMS) dataset. Experimental results demonstrate that our DCMN models are more competitive over the baseline methods in the multimodal prediction setting.

[1]  Gaël Varoquaux,et al.  Scikit-learn: Machine Learning in Python , 2011, J. Mach. Learn. Res..

[2]  Andrew Y. Ng,et al.  Cardiologist-Level Arrhythmia Detection with Convolutional Neural Networks , 2017, ArXiv.

[3]  Erik Cambria,et al.  Memory Fusion Network for Multi-view Sequential Learning , 2018, AAAI.

[4]  C. Peng,et al.  Fractal analysis of heart rate dynamics as a predictor of mortality in patients with depressed left ventricular function after acute myocardial infarction. TRACE Investigators. TRAndolapril Cardiac Evaluation. , 1999, The American journal of cardiology.

[5]  R. Fetter,et al.  Case mix definition by diagnosis-related groups. , 1980, Medical care.

[6]  Jeffrey M. Hausdorff,et al.  Physionet: Components of a New Research Resource for Complex Physiologic Signals". Circu-lation Vol , 2000 .

[7]  Selcan Kaplan Berkaya,et al.  A survey on ECG analysis , 2018, Biomed. Signal Process. Control..

[8]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[9]  H. Huikuri,et al.  Prediction of sudden cardiac death after acute myocardial infarction: role of Holter monitoring in the modern treatment era. , 2005, European heart journal.

[10]  Fei Wang,et al.  A Framework for Mining Signatures from Event Sequences and Its Applications in Healthcare Data , 2013, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[11]  Julien Perez,et al.  Gated End-to-End Memory Networks , 2016, EACL.

[12]  Ning Chen,et al.  Patient outcome prediction via convolutional neural networks based on multi-granularity medical concept embedding , 2017, 2017 IEEE International Conference on Bioinformatics and Biomedicine (BIBM).

[13]  Jason Weston,et al.  End-To-End Memory Networks , 2015, NIPS.

[14]  Fei Wang,et al.  Readmission prediction via deep contextual embedding of clinical concepts , 2018, PloS one.

[15]  F Halberg,et al.  Fractal analysis of heart rate variability and mortality in elderly community-dwelling people -- Longitudinal Investigation for the Longevity and Aging in Hokkaido County (LILAC) study. , 2005, Biomedicine & pharmacotherapy = Biomedecine & pharmacotherapie.

[16]  Jiann-Shiun Yuan,et al.  ECG Arrhythmia Classification Using Transfer Learning from 2- Dimensional Deep CNN Features , 2018, 2018 IEEE Biomedical Circuits and Systems Conference (BioCAS).

[17]  P. Stein,et al.  Heart rate variability in risk stratification of cardiac patients. , 2013, Progress in cardiovascular diseases.

[18]  Ping Zhang,et al.  Risk Prediction with Electronic Health Records: A Deep Learning Approach , 2016, SDM.

[19]  Joon Son Chung,et al.  Lip Reading Sentences in the Wild , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[20]  Gareth L. Ackland,et al.  Heart rate variability in critical care medicine: a systematic review , 2017, Intensive Care Medicine Experimental.

[21]  Fei Wang,et al.  Integrative Analysis of Patient Health Records and Neuroimages via Memory-Based Graph Convolutional Network , 2018, 2018 IEEE International Conference on Data Mining (ICDM).

[22]  Yan Liu,et al.  Benchmarking deep learning models on large healthcare datasets , 2018, J. Biomed. Informatics.

[23]  Richard Socher,et al.  Dynamic Memory Networks for Visual and Textual Question Answering , 2016, ICML.

[24]  Jimeng Sun,et al.  RAIM: Recurrent Attentive and Intensive Model of Multimodal Patient Monitoring Data , 2018, KDD.

[25]  Petr Sojka,et al.  Software Framework for Topic Modelling with Large Corpora , 2010 .

[26]  Fei Wang,et al.  Deep Learning in Medicine-Promise, Progress, and Challenges. , 2019, JAMA internal medicine.

[27]  Dinggang Shen,et al.  Predicting Alzheimer's Disease Cognitive Assessment via Robust Low-Rank Structured Sparse Model , 2017, IJCAI.

[28]  Tianqi Chen,et al.  XGBoost: A Scalable Tree Boosting System , 2016, KDD.

[29]  Aram Galstyan,et al.  Multitask learning and benchmarking with clinical time series data , 2017, Scientific Data.

[30]  Fei Wang,et al.  Towards heterogeneous temporal clinical event pattern discovery: a convolutional approach , 2012, KDD.

[31]  Masun Nabhan Homsi,et al.  Cardiac arrhythmia detection from ECG combining convolutional and long short-term memory networks , 2017, 2017 Computing in Cardiology (CinC).

[32]  Sergio Gomez Colmenarejo,et al.  Hybrid computing using a neural network with dynamic external memory , 2016, Nature.

[33]  Fei Wang,et al.  Deep learning for healthcare: review, opportunities and challenges , 2018, Briefings Bioinform..

[34]  Fei Wang,et al.  Predictive Modeling of the Hospital Readmission Risk from Patients’ Claims Data Using Machine Learning: A Case Study on COPD , 2019, Scientific Reports.

[35]  Svetha Venkatesh,et al.  Dual Memory Neural Computer for Asynchronous Two-view Sequential Learning , 2018, KDD.

[36]  Michael Tschannen,et al.  Convolutional recurrent neural networks for electrocardiogram classification , 2017, 2017 Computing in Cardiology (CinC).

[37]  Peter Szolovits,et al.  MIMIC-III, a freely accessible critical care database , 2016, Scientific Data.

[38]  Fei Wang,et al.  Combining Knowledge and Data Driven Insights for Identifying Risk Factors using Electronic Health Records , 2012, AMIA.