Early Detection of Heart Disease Using Gated Recurrent Neural Network
Article Sidebar
Main Article Content
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.
Article Details
References
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