Early Detection of Heart Disease Using Gated Recurrent Neural Network

Main Article Content

Shawni Dutta
Samir Kumar Bandyopadhyay

Abstract

Cardiovascular disease (CVD) is an important factor in life since it may cause the death of human by effecting the heart and blood vessels of the body. Early detection of this disease is necessary for securing patients life. For this purpose, an automated tool is proposed in this paper for detecting patients with CVD and assisting health care systems also. A stacked-GRU based Recurrent Neural Network model, abbreviated as, stk-G, is proposed in this paper that considers interfering factors from past health records while detecting patients with cardiac problems. This proposed model is compared with two benchmark classifiers known as Support Vector Machine (SVM) and K-Nearest Neighbour (K-NN). The comparative analysis concludes that the proposed model offers enhanced efficiency for heart disease prediction. A promising result is given by the proposed method with an accuracy of 84.37%, F1-Score of 0.84 and MSE of 0.16.

Keywords:
Heart disease prediction, machine learning, deep learning, GRU, SVM, K-NN

Article Details

How to Cite
Dutta, S., & Bandyopadhyay, S. K. (2020). Early Detection of Heart Disease Using Gated Recurrent Neural Network. Asian Journal of Cardiology Research, 3(1), 8-15. Retrieved from https://journalajcr.com/index.php/AJCR/article/view/30101
Section
Original Research Article

References

Sylva M, Van den Hoff MJB, Moorman AFM. Development of the human heart,” Am. J. Med. Genet. Part A. 2014;164(6): 1347–1371.
DOI: 10.1002/ajmg.a.35896.

Chung J, Gulcehre C, Cho K, Bengio Y. Empirical evaluation of gated recurrent neural networks on sequence modeling. 2014;1–9.

Evgeniou T, Pontil M. Workshop on support vector machines: Theory and applications. Mach. Learn. Its Appl. Adv. Lect. 2001;249–257.
DOI: 10.1007/3-540-44673-7

Cunningham P, Delany SJ. K-Nearest neighbour classifiers. Mult. Classif. Syst. 2007;1–17.
DOI: 10.1016/S0031-3203(00)00099-6

Chauhan A, Jain A, Sharma P, Deep V. Heart disease prediction using evolutionary rule learning. Int. Conf. Computational Intell. Commun. Technol. CICT. Cict. 2018; 1–4.
DOI: 10.1109/CIACT.2018.8480271

Gonsalves AH, Thabtah F, Mohammad RMA, Singh G. Prediction of coronary heart disease using machine learning: An experimental analysis. ACM Int. Conf. Proceeding Ser. 2019;51–56.
DOI: 10.1145/3342999.3343015

Austin PC, Tu JV, Ho JE, Levy D, Lee DS. Using methods from the data-mining and machine-learning literature for disease classification and prediction: A case study examining classification of heart failure subtypes. J. Clin. Epidemiol. 2013;66(4): 398–407.
DOI: 10.1016/j.jclinepi.2012.11.008

Kirmani M. Cardiovascular disease prediction using data mining techniques. Orient. J. Comput. Sci. Technol. 2017; 10(2):520–528.
DOI: 10.13005/ojcst/10.02.38

Sai PP, Reddy C. Heart disease prediction using ann algorithm in data mining. International Journal of Computer Science and Mobile Computing. 2017;6(4):168-172.

Bahrami B, Hosseini Shirvani M. Prediction and diagnosis of heart disease by data mining techniques. J. Multidiscip. Eng. Sci. Technol. 2015;2(2):3159–40.

Mohan S, Thirumalai C, Srivastava G. Effective heart disease prediction using hybrid machine learning techniques. IEEE Access. 2019;7:81542–81554.
DOI: 10.1109/ACCESS.2019.2923707

Izci E, Ozdemir MA, Degirmenci M, Akan A. Cardiac arrhythmia detection from 2d ecg images by using deep learning technique. TIPTEKNO 2019 - Tip Teknol. Kongresi. 2019;1–4.
DOI: 10.1109/TIPTEKNO.2019.8895011

Blake C, Keogh E, Merz CJ. UCI repository of machine learning databases. Department of Information and Computer Science, University of California, Irvine, CA; 1998.
Available:http://www.ics.uci.edu/∼mlearn/MLRepository.html

Lipton ZC, Berkowitz J, Elkan C. A critical review of recurrent neural networks for sequence learning. 2015;1–38.

Nwankpa C, Ijomah W, Gachagan A, Marshall S. Activation functions: Comparison of trends in practice and research for deep learning. 2018;1–20.

Canbek G, Temizel TT, Sagiroglu S, Baykal N. Binary classification performance measures/metrics: A comprehensive visualized roadmap to gain new insights,” 2nd Int. Conf. Comput. Sci. Eng. UBMK. 2017;821–826.
DOI: 10.1109/UBMK.2017.8093539

HM, SMN. a review on evaluation metrics for data classification evaluations. Int. J. Data Min. Knowl. Manag. Process. 2015;5(2):01–11.
DOI: 10.5121/ijdkp.2015.5201