An automatic heart disease prediction using cluster-based bi-directional LSTM (C-BiLSTM) algorithm

被引:59
作者
Dileep, P. [1 ]
Rao, Kunjam Nageswara [2 ]
Bodapati, Prajna [2 ]
Gokuruboyina, Sitaratnam [3 ]
Peddi, Revathy [4 ]
Grover, Amit [5 ]
Sheetal, Anu [6 ]
机构
[1] Malla Reddy Coll Engn & Technol, Dept Comp Sci & Engn, Hyderabad 500100, Telangana, India
[2] Andhra Univ, Dept CS & SE, AUCE A, Visakhapatnam, Andhra Pradesh, India
[3] LENDI Inst Engn & Technol, Dept CSE, Vizianagaram, Andhra Pradesh, India
[4] ACE Engn Coll, Dept CSE, Hyderabad, Telangana, India
[5] Shaheed Bhagat Singh State Univ, Dept Elect & Commun Engn, Ferozepur, Punjab, India
[6] Guru Nanak Dev Univ, Dept Engn & Technol, Reg Campus, Gurdaspur, Punjab, India
关键词
Heart disease; K- means clustering; BiLSTM; Prediction; Performance metrics; LIFE-STYLE; DEEP; RISK;
D O I
10.1007/s00521-022-07064-0
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Heart disease involves many diseases like block blood vessels, heart attack, chest pain or stroke. Heart disease will affect the muscles, valves or heart rate, and bypass surgery or coronary artery surgery will be used to treat these problems. In this paper, UCI heart disease dataset and real time dataset are used to test the deep learning techniques which are compared with the traditional methods. To improve the accuracy of the traditional methods, cluster-based bi-directional long-short term memory (C-BiLSTM) has been proposed. The UCI and real time heart disease dataset are used for experimental results, and both the datasets are used as inputs through the K-Means clustering algorithm for the removal of duplicate data, and then, the heart disease has been predicted using C-BiLSTM approach. The conventional classifier methods such as Regression Tree, SVM, Logistic Regression, KNN, Gated Recurrent Unit and Ensemble are compared with C-BiLSTM, and the efficiency of the system is demonstrated in terms of accuracy, sensitivity and F1 score. The results show that the C-BiLSTM proves to be the best with 94.78% accuracy of UCI dataset and 92.84% of real time dataset compared to the six conventional methods for providing better prediction of heart disease.
引用
收藏
页码:7253 / 7266
页数:14
相关论文
共 30 条
[1]   An Automated Diagnostic System for Heart Disease Prediction Based on χ2 Statistical Model and Optimally Configured Deep Neural Network [J].
Ali, Liaqat ;
Rahman, Atiqur ;
Khan, Aurangzeb ;
Zhou, Mingyi ;
Javeed, Ashir ;
Khan, Javed Ali .
IEEE ACCESS, 2019, 7 :34938-34945
[2]  
Anbarasi M., 2010, INT J ENG SCI TECHNO, V2, P5370
[3]   Ensemble Deep Learning Models for Heart Disease Classification: A Case Study from Mexico [J].
Baccouche, Asma ;
Garcia-Zapirain, Begonya ;
Olea, Cristian Castillo ;
Elmaghraby, Adel .
INFORMATION, 2020, 11 (04)
[4]  
Blake C., 1998, Uci repository of machine learning databases
[5]  
Dileep P., 2019, INT J ADV SCI TECHNO, V28, P309
[6]  
Dong S, 2016, COMPUTING CARDIOLOGY
[7]  
Florence S., 2014, Int J Innov Res Comput Commun Eng, V11, P2320
[8]  
Javid I, 2020, INT J ADV COMPUT SC, V11, P540
[9]  
Kanimozhi V.A., 2016, Int J Adv Res Compu Commun Eng, V5, P552
[10]  
Karthikeyan T, 2017, Int. J. Adv. Res. Sci. Eng. Technol, V4, P3194