Home > Archive > 2018 > Volume 8 Number 2 (Apr. 2018) >
IJMLC 2018 Vol.8(2): 117-120 ISSN: 2010-3700
DOI: 10.18178/ijmlc.2018.8.2.673

Diagnosis Prediction via Recurrent Neural Networks

Yangzi Mu, Mengxing Huang, Chunyang Ye, and Qingzhou Wu

Abstract—The prediction of patient’s future health information from the historical electronic health records (EHR) forms the core of the development of personalized healthcare research tasks. Patient EHR data consists of sequences of visits over time, where each visit contains multiple medical codes, including diagnosis, medication, and patient profile. Using historical data from the EHR, we can predict medical conditions and medication uses. Existing works model EHR data by using recurrent neural networks (RNNs). However, RNN-based approaches have certain limitations: the performance of RNNs drops when the length of sequences is large and they ignore some of the characteristics of the patients themselves. We propose an application of using bidirectional RNNs to remember all the information of both the past and future visits and add some patient’s characteristics as side information into this model. Experimental results on real world EHR datasets show that the proposed model can remarkably improve the prediction accuracy when compared with the diagnosis prediction approaches, and it can provide clinically meaningful interpretation.

Index Terms—Component, electronic health records, bidirectional recurrent neural networks, side information

The authors are with College of information Science &Technology, Hainan University, Haikou, China and State Key Laboratory of marine resource utilization in the South China Sea, Hainan University (e-mail: muyangzi521@163.com, huangmx09.com, cyye@ustc.com, wuqingzhou@21cn.com).


Cite: Yangzi Mu, Mengxing Huang, Chunyang Ye, and Qingzhou Wu, "Diagnosis Prediction via Recurrent Neural Networks," International Journal of Machine Learning and Computing vol. 8, no. 2, pp. 117-120, 2018.

General Information

  • ISSN: 2010-3700 (Online)
  • Abbreviated Title: Int. J. Mach. Learn. Comput.
  • Frequency: Bimonthly
  • DOI: 10.18178/IJMLC
  • Editor-in-Chief: Dr. Lin Huang
  • Executive Editor:  Ms. Cherry L. Chen
  • Abstracing/Indexing: Inspec (IET), Google Scholar, Crossref, ProQuest, Electronic Journals Library.
  • E-mail: ijmlc@ejournal.net

Article Metrics