Diagnosing Cardiovascular Diseases using Optimized Machine Learning Algorithms with GridSearchCV

Khalid Alemerien, Saleel Alsarayreh, Enshirah Altarawneh
Pages 1539 -1552
Abstract Views 1200
PDF Downloads 969
Crossref Citations -
DOI https://doi.org/10.47738/jads.v5i4.280

Abstract


Accurate and timely diseases diagnosis is the most important responsibility in the healthcare industry for protecting the people lives. Many lives can be spared from death if their cases diagnosed accurately and early. One of the dangerous diseases is cardiovascular disease (CVD), is the leading cause of death worldwide, making it one of the hardest conditions to diagnose. Globally, about 17.9 million of people are died because of the cardiovascular disease. In order to assist physicians in this mission, automated solutions based on machine learning and deep learning techniques are introduced. Therefore, machine learning algorithms can diagnose diseases quickly and accurately, which adds a huge value to the medical industry. This gives physicians and patients plenty of time. To address this issue, we utilized several supervised machine learning (ML) techniques with GridSearchCV optimizer. Using the optimization techniques can enhance the performance and accuracy of proposed ML-based models. Therefore, we conducted a comparative analysis study to identify the most efficient classification model using two benchmark real datasets from the online Kaggle repository. Seven popular machine learning techniques were utilized: Decision Tree (DT), Support Vector Machine (SVM), Logistic regression (LR), K-Nearest Neighbor (KNN), Random Forest (RF), XGBoost and Naïve Bayes (NB). The findings revealed that both Random Forest and XGBoost classifiers yields highest results in both of the datasets used in our study in terms of accuracy 95.38% and 98.54%, respectively. The rest of ML algorithms showed less performance in predicting the CVD in terms of accuracy, where DT and RF achieved an accuracy of 98.53% and 98.52%, respectively, on the first dataset. Furthermore, employing the proposed ML-based model in the diagnosing CVD process shows the expected implications for patients and physicians. In addition, it shows the impact of constructing a real comprehensive dataset to enhance the performance of proposed solutions.

Full Text:

PDF

References


Pouriyeh, S., Vahid, S., Sannino, G., De Pietro, G., Arabnia, H., & Gutierrez, J. (2017, July). A comprehensive investigation and comparison of machine learning techniques in the domain of heart disease. In 2017 IEEE symposium on computers and communications (ISCC) (pp. 204-207). IEEE.‏

D. Zhang, et al., "Common Mode Circulating Current Control of Interleaved Three-Phase Two-Level Voltage-Source Converters with Discontinuous Space-Vector Modulation," 2009 IEEE Energy Conversion Congress and Exposition, Vol 1-6, pp. 3906-3912, 2009.

Dangare, C. S., & Apte, S. S. (2012). Improved study of heart disease prediction system using data mining classification techniques. International Journal of Computer Applications, 47(10), 44-48.‏

B. Fida, M. Nazir, N. Naveed and S. Akram, "Heart disease classification ensemble optimization using Genetic algorithm," 2011 IEEE 14th International Multi-topic Conference, Karachi, Pakistan, 2011, pp. 19-24, doi: 10.1109/INMIC.2011.6151471.

Kirubha, V., & Priya, S. M. (2016). Survey on data mining

algorithms in disease prediction. International Journal of Computer Trends and Technology, 38(3), 124-128.‏

Dehkordi, S. K., & Sajedi, H. (2019). Prediction of disease based on prescription using data mining methods. Health and Technology, 9, 37-44.‏

Garg, A., & Mago, V. (2021). Role of machine learning in medical research: A survey. Computer science review, 40, 100370.

Roy, D., Mahmood, M. A., & Roy, T. J. (2023). An Analytical Model for Prediction of Heart Disease using Machine Learning Classifiers. Authorea Preprints.

Bhatt, C. M., Patel, P., Ghetia, T., & Mazzeo, P. L. (2023). Effective heart disease prediction using machine learning techniques. Algorithms, 16(2), 88.

Arumugam, K., Naved, M., Shinde, P. P., Leiva-Chauca, O., Huaman-Osorio, A., & Gonzales-Yanac, T. (2023). Multiple disease prediction using Machine learning algorithms. Materials Today: Proceedings, 80, 3682-3685.

Aladeyelu, A. C., & Adekunle, G. T. (2023). Predicting Heart Disease Using Machine Learning. Machine learning, 10(4).

Kadhim, M. A., & Radhi, A. M. (2023). Heart disease classification using optimized Machine learning algorithms. Iraqi Journal For Computer Science and Mathematics, 4(2), 31-42.‏

Chandrasekhar, N., & Peddakrishna, S. (2023). Enhancing Heart Disease Prediction Accuracy through Machine Learning Techniques and Optimization. Processes, 11(4), 1210.‏

Anderies, A., Tchin, J. A. R. W., Putro, P. H., Darmawan, Y. P., & Gunawan, A. A. S. (2022). Prediction of heart disease UCI dataset using machine learning algorithms. Engineering, MAthematics and Computer Science (EMACS) Journal, 4(3), 87-93.‏

Saboor, A., Usman, M., Ali, S., Samad, A., Abrar, M. F., & Ullah, N. (2022). A method for improving prediction of human heart disease using machine learning algorithms. Mobile Information Systems, 2022.‏

Shu Jiang, Heart disease prediction using machine learning algorithms, Master thesis, Los Angeles, 2022

Yilmaz, R., & YAĞIN, F. H. (2022). Early detection of coronary heart disease based on machine learning methods. Medical Records, 4(1), 1-6.‏

Alalawi, H. H., & Alsuwat, M. S. (2021). Detection of cardiovascular disease using machine learning classification models. International Journal of Engineering Research & Technology, 10(7), 151-7.‏

Katarya, R., & Meena, S. K. (2021). Machine learning techniques for heart disease prediction: a comparative study and analysis. Health and Technology, 11, 87-97.

Kumar, N. K., Sindhu, G. S., Prashanthi, D. K., & Sulthana, A. S. (2020, March). Analysis and prediction of cardio vascular disease using machine learning classifiers. In 2020 6th International Conference on Advanced Computing and Communication Systems (ICACCS) (pp. 15-21). IEEE.

Singh, A., & Kumar, R. (2020, February). Heart disease prediction using machine learning algorithms. In 2020 international conference on electrical and electronics engineering (ICE3) (pp. 452-457). IEEE.‏

Shah, D., Patel, S., & Bharti, S. K. (2020). Heart disease prediction using machine learning techniques. SN Computer Science, 1, 1-6.

Saqlain, M., Hussain, W., Saqib, N. A., & Khan, M. A. (2016, August). Identification of heart failure by using unstructured data of cardiac patients. In 2016 45th International Conference on Parallel Processing Workshops (ICPPW) (pp. 426-431). IEEE.‏

Pal, M., & Parija, S. (2021, March). Prediction of heart diseases using random forest. In Journal of Physics: Conference Series (Vol. 1817, No. 1, p. 012009). IOP Publishing.

Katarya, R., & Srinivas, P. (2020, July). Predicting heart disease at early stages using machine learning: A survey. In 2020 International Conference on Electronics and Sustainable Communication Systems (ICESC) (pp. 302-305). IEEE.‏

Osisanwo, F. Y., Akinsola, J. E. T., Awodele, O., Hinmikaiye, J. O., Olakanmi, O., & Akinjobi, J. (2017). Supervised machine learning algorithms: classification and comparison. International Journal of Computer Trends and Technology (IJCTT), 48(3), 128-138.‏

Beyene, C., & Kamat, P. (2018). Survey on prediction and analysis the occurrence of heart disease using data mining techniques. International Journal of Pure and Applied Mathematics, 118(8), 165-174.‏

Khanna, D., Sahu, R., Baths, V., & Deshpande, B. (2015). Comparative study of classification techniques (SVM, logistic regression and neural networks) to predict the prevalence of heart disease. International Journal of Machine Learning and Computing, 5(5), 414.‏

logistic Regression in Machine Learning, https://www.javatpoint.com/logistic-regression-in-machine-learning, Last accessed 13/01/2024.

Ayon, S. I., Islam, M. M., & Hossain, M. R. (2022). Coronary artery heart disease prediction: a comparative study of computational intelligence techniques. IETE Journal of Research, 68(4), 2488-2507.‏

Sharma, H., & Rizvi, M. A. (2017). Prediction of heart disease using machine learning algorithms: A survey. International Journal on Recent and Innovation Trends in Computing and Communication, 5(8), 99-104.‏

Cai, Z., Gu, J., Wen, C., Zhao, D., Huang, C., Huang, H., ... & Chen, H. (2018). An intelligent Parkinson’s disease diagnostic system based on a chaotic bacterial foraging optimization enhanced fuzzy KNN approach. Computational and mathematical methods in medicine, 2018.‏

Anggoro, D. A., & Kurnia, N. D. (2020). Comparison of accuracy level of support vector machine (SVM) and K-nearest neighbors (KNN) algorithms in predicting heart disease. International Journal, 8(5), 1689-1694.‏

Bora, N., Gutta, S., & Hadaegh, A. (2021). Using Machine Learning to Predict Heart Disease (Doctoral dissertation, California State University San Marcos).‏

Das, K., & Behera, R. N. (2017). A survey on machine learning: concept, algorithms and applications. International Journal of Innovative Research in Computer and Communication Engineering, 5(2), 1301-1309.‏

Yang, J., & Guan, J. (2022). A heart disease prediction model based on feature optimization and smote-Xgboost algorithm. Information, 13(10), 475.‏

Heart Disease Dataset, https://www.kaggle.com/datasets/johnsmith88/heart-disease-dataset, Last accessed in 13/01/2024

Mahesh, B. (2020). Machine learning algorithms-a review. International Journal of Science and Research (IJSR).[Internet], 9(1), 381-386.‏

Heart Statlog Cleveland Hungary Final, ‏https://www.kaggle.com/datasets/sid321axn/heart-statlog-cleveland-hungary-final, Last accessed in 13/01/2024

DeGroat, W., Abdelhalim, H., Patel, K., Mendhe, D., Zeeshan, S., & Ahmed, Z. (2024). Discovering biomarkers associated and predicting cardiovascular disease with high accuracy using a novel nexus of machine learning techniques for precision medicine. Scientific reports, 14(1), 1.

Kaur, D., Hughes, J. W., Rogers, A. J., Kang, G., Narayan, S. M., Ashley, E. A., & Perez, M. V. (2024). Race, Sex, and Age Disparities in the Performance of ECG Deep Learning Models Predicting Heart Failure. Circulation: Heart Failure, 17(1), e010879.

Thukral, A., Dhiman, S., Meher, R., & Bedi, P. (2023). Knowledge graph enrichment from clinical narratives using NLP, NER, and biomedical ontologies for healthcare applications. International Journal of Information Technology, 15(1), 53-65.‏

Marelli, A. J., Li, C., Liu, A., Nguyen, H., Moroz, H., Brophy, J. M., ... & Li, Y. (2024). Machine learning informed diagnosis for congenital heart disease in large claims data source. JACC: Advances, 3(2), 100801.

Bahbouh, N. M., Compte, S. S., Valdes, J. V., & Sen, A. A. A. (2023). An empirical investigation into the altering health perspectives in the internet of health things. International Journal of Information Technology, 15(1), 67-77.‏

Dubey, A. K., Sinhal, A. K., & Sharma, R. (2023). Heart disease classification through crow intelligence optimization-based deep learning approach. International Journal of Information Technology, 1-16.‏




DOI: https://doi.org/10.47738/jads.v5i4.280

Refbacks

  • There are currently no refbacks.


Barcode

Journal of Applied Data Sciences

ISSN:2723-6471 (Online)
Publisher:Bright Publisher
Website:http://bright-journal.org/JADS
Email:taqwa@amikompurwokerto.ac.id (principal contact)
  support@bright-journal.org (technical issues)

This work is licensed under a Creative Commons Attribution-ShareAlike 4.0