Development of a Grey Wolf Optimized-Gradient Boosted Decision Tree Metamodel for Heart Disease Prediction

Authors

  • Narayanan Ganesh School of Computer Science & Engineering, Vellore Institute of Technology, Chennai 600 027, India
  • M. Balamurugan Department of Computer Science, Kristu Jayanti College (Autonomous), Bengaluru, 560077, India
  • Jasgurpreet Singh Chohan Department of Mechanical Engineering and University Centre for Research & Development, Chandigarh University, Mohali, 140413, India
  • Kanak Kalita Department of Mechanical Engineering, Vel Tech Rangarajan Dr. Sagunthala R&D Institute of Science and Technology, Avadi 600062, India

Keywords:

decision tree, heart disease, optimization, metamodel, machine learning, prediction

Abstract

In this paper, a comprehensive study of various machine learning (ML) metamodels for heart disease detection is presented. The comparison includes conventional metamodels such as Logistic Regression, Support Vector Machine, K-Nearest Neighbors, Decision Trees, Random Forest as well as more advanced metamodels including Deep Learning, ML, Deep Neural Networks, Gradient Boosted Decision Trees and the proposed Grey Wolf Optimizer-Gradient Boosted Decision Trees (GWO-GBDT). The metamodels are assessed based on their performance in terms of accuracy, recall, precision, F1 measure and specificity. The results reveal that the developed GWO-GBDT metamodel outperforms the other metamodels in most metrics, offering superior prediction capabilities for heart disease diagnosis. This study provides a valuable reference for researchers and practitioners seeking efficient ML metamodels for heart disease prediction.

 

Downloads

Download data is not yet available.

References

A. Dewan and M. Sharma, "Prediction of heart disease using a hybrid technique in data mining classification," in 2015 2nd International Conference on Computing for Sustainable Global Development (INDIACom), 2015.

P. K. Anooj, "Clinical decision support system: Risk level prediction of heart disease using weighted fuzzy rules," Journal of King Saud University-Computer and Information Sciences, vol. 24, p. 27–40, 2012.

S. Sharanyaa, S. Lavanya, M. R. Chandhini, R. Bharathi and K. Madhulekha, "Hybrid Machine Learning Techniques for Heart Disease Prediction," International Journal of Advanced Engineering Research and Science, vol. 7, p. 44–8, 2020.

N. A. Rajendran and D. R. Vincent, "Heart disease prediction system using ensemble of machine learning algorithms," Recent Patents on Engineering, vol. 15, p. 130–139, 2021.

R. Tr, U. K. Lilhore, Poongodi, S. Simaiya, A. Kaur and M. Hamdi, "Predictive analysis of heart diseases with Machine Learning approaches," Malays. J. Comput. Sci., p. 132–148, March 2022.

V. Shorewala, "Early detection of coronary heart disease using ensemble techniques," Informatics in Medicine Unlocked, vol. 26, p. 100655, 2021.

A. Tiwari, A. Chugh and A. Sharma, "Ensemble framework for cardiovascular disease prediction," Computers in Biology and Medicine, vol. 146, p. 105624, 2022.

A. Rehman, S. Naz and I. Razzak, "Leveraging big data analytics in healthcare enhancement: trends, challenges and opportunities," Multimed. Syst., vol. 28, p. 1339–1371, August 2022.

K. Shaik, J. Ramesh, M. Mahdal, M. Rahman, S. Khasim and K. Kalita, "Big Data Analytics Framework Using Squirrel Search Optimized Gradient Boosted Decision Tree for Heart Disease Diagnosis," Applied Sciences , vol. 13, no. 9, p. 5236, 2023.

V. Chang, V. R. Bhavani, A. Q. Xu and M. A. Hossain, "An artificial intelligence model for heart disease detection using machine learning algorithms," Healthcare Analytics, vol. 2, p. 100016, November 2022.

U. Nagavelli, D. Samanta and P. Chakraborty, "Machine learning technology-based heart disease detection models," J. Healthc. Eng., vol. 2022, p. 7351061, February 2022.

S. Ketu and P. K. Mishra, "Empirical analysis of machine learning algorithms on imbalance electrocardiogram based arrhythmia dataset for heart disease detection," Arabian Journal for Science and Engineering, p. 1–23, 2022.

K. V. V. Reddy, I. Elamvazuthi, A. A. Aziz, S. Paramasivam, H. N. Chua and S. Pranavanand, "Heart disease risk prediction using machine learning classifiers with attribute evaluators," Applied Sciences, vol. 11, p. 8352, 2021.

A. Baccouche, B. Garcia-Zapirain, C. Castillo Olea and A. Elmaghraby, "Ensemble deep learning models for heart disease classification: A case study from Mexico," Information, vol. 11, p. 207, 2020.

A. Almulihi, H. Saleh, A. M. Hussien, S. Mostafa, S. El-Sappagh, K. Alnowaiser, A. A. Ali and M. Refaat Hassan, "Ensemble Learning Based on Hybrid Deep Learning Model for Heart Disease Early Prediction," Diagnostics, vol. 12, p. 3215, 2022.

T. Yoon and D. Kang, "Multi-Modal Stacking Ensemble for the Diagnosis of Cardiovascular Diseases," Journal of Personalized Medicine, vol. 13, p. 373, 2023.

A. Menshawi, M. M. Hassan, N. Allheeib and G. Fortino, "A Hybrid Generic Framework for Heart Problem Diagnosis Based on a Machine Learning Paradigm," Sensors, vol. 23, p. 1392, 2023.

D. Cenitta, R. Vijaya Arjunan and K. V. Prema, "Ischemic heart disease prediction using optimized squirrel search feature selection algorithm," IEEE Access, vol. 10, p. 122995–123006, 2022.

R. Bharti, A. Khamparia, M. Shabaz, G. Dhiman, S. Pande and P. Singh, "Prediction of heart disease using a combination of machine learning and deep learning," Comput. Intell. Neurosci., vol. 2021, p. 8387680, July 2021.

Y.-F. Ko, P.-H. Kuo, C.-F. Wang, Y.-J. Chen, P.-C. Chuang, S.-Z. Li, B.-W. Chen, F.-C. Yang, Y.-C. Lo, Y. Yang and others, "Quantification analysis of sleep based on smartwatch sensors for Parkinson’s disease," Biosensors, vol. 12, p. 74, 2022.

K. H. Miao and J. H. Miao, "Coronary heart disease diagnosis using deep neural networks," international journal of advanced computer science and applications, vol. 9, 2018.

M. S. Nawaz, B. Shoaib and M. A. Ashraf, "Intelligent Cardiovascular Disease Prediction Empowered with Gradient Descent Optimization," Heliyon, vol. 7, p. e06948, May 2021.

Downloads

Published

10.12.2023

How to Cite

Ganesh, N. ., Balamurugan, M. ., Chohan, J. S. ., & Kalita, K. . (2023). Development of a Grey Wolf Optimized-Gradient Boosted Decision Tree Metamodel for Heart Disease Prediction. International Journal of Intelligent Systems and Applications in Engineering, 12(8s), 515–522. Retrieved from https://ijisae.org/index.php/IJISAE/article/view/4160

Issue

Section

Research Article