Development of a Grey Wolf Optimized-Gradient Boosted Decision Tree Metamodel for Heart Disease Prediction
Keywords:
decision tree, heart disease, optimization, metamodel, machine learning, predictionAbstract
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
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
How to Cite
Issue
Section
License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
All papers should be submitted electronically. All submitted manuscripts must be original work that is not under submission at another journal or under consideration for publication in another form, such as a monograph or chapter of a book. Authors of submitted papers are obligated not to submit their paper for publication elsewhere until an editorial decision is rendered on their submission. Further, authors of accepted papers are prohibited from publishing the results in other publications that appear before the paper is published in the Journal unless they receive approval for doing so from the Editor-In-Chief.
IJISAE open access articles are licensed under a Creative Commons Attribution-ShareAlike 4.0 International License. This license lets the audience to give appropriate credit, provide a link to the license, and indicate if changes were made and if they remix, transform, or build upon the material, they must distribute contributions under the same license as the original.