Machine Learning Based Cardiovascular Disease Detection Using Least Absolute Shrinkage and Selection Operator Method

Waqas Ahmad; Narub Iqbal; Shairose; Muhammad Nadeem

Author(s)

Waqas Ahmad Department of Computer Science and Technology, University of Science and Technology Beijing, Beijing 10083, China

Narub Iqbal Department of Computer Science, Alhamd Islamic University, Islamabad 44000, Pakistan

Shairose Department of Computer Science, Alhamd Islamic University, Islamabad 44000, Pakistan

Muhammad Nadeem Department of Computer Science and Technology, University of Science and Technology Beijing, Beijing 10083, China

Abstract

Early detection of cardiovascular disease is paramount as it stands among the most fatal and devastating illnesses globally. Despite extensive research efforts, crucial evaluation parameters like the area under the receiver operating characteristic curve (AUC-ROC), pivotal for diagnostic model assessment, have often been overlooked. AUC-ROC, accuracy, sensitivity, specificity, precision, and other assessment metrics have also been considered in a few research, however their conclusions have not been successful. This paper introduces a model utilizing the Least Absolute Shrinkage and Selection Operator (LASSO) feature selection technique within artificial neural networks (ANN). Alongside ANN, several machine learning classifiers including Random Forest (RF), Decision Tree (DT), Support Vector Machine (SVM), and K Nearest Neighbors (KNN) are employed to compare their efficacy in identifying cardiovascular illness, using a dataset of 70,000 medical records from Kaggle. According to the experimental findings, the suggested approach using ANN obtained accuracy, sensitivity, specificity, precision, and ROC score of 90.39%, 90.92%, 89.97%, 87.79%, and 0.95, respectively. In contrast, KNN, SVM, RF, and DT achieved cardiovascular diagnosis accuracies of 76.47%, 83.40%, 84.98%, and 79.58%, respectively. The findings of the classifiers utilized in this study clearly show that ANN performed better at identifying cardiovascular illness.

Publication Details

Page(s) 86-101

DOI DOI not available

Published Journal: The Sciencetech, Volume: 5, Issue: 4, Year: 2024

Request the full text from PASTIC

Keywords

machine learning, Cardiovascular disease, Least Absolute Shrinkage and Selection Operator, Artificial Neural Network

References

Arias-Duart A.,Mariotti E.,Garcia-Gasulla D.,Alonso-Moral J. M. .2023 .A confusion matrix for evaluating feature attribution methods. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, : 3709-3714.

Baghdadi N. A.,Farghaly Abdelaliem S. M.,Malki A.,Gad I.,Ewis A.,Atlam E. .2023 .Advanced machine learning techniques for cardiovascular disease early detection and diagnosis. Journal of Big Data, 10(1) : 144.

Gharehchopogh F. S.,Abdollahzadeh B. .2022 .An efficient harris hawk optimization algorithm for solving the travelling salesman problem. Cluster Computing, 25(3) : 1981-2005.

Ghosh P.,Azam S.,Jonkman M.,Karim A.,Shamrat F. J. M.,Ignatious E.,De Boer F. .2021 .Efficient prediction of cardiovascular disease using machine learning algorithms with relief and LASSO feature selection techniques. IEEE Access, 9 : 19304-19326.

Hoo Z. H.,Candlish J.,Teare D. .2017 .What is an ROC curve?. Emergency Medicine Journal, 34(6) : 357-359.

Jiang Y.,Zhang X.,Ma X.,Wang X.,Liu J.,Keerman M.,Guo H. .2021 .Cardiovascular disease prediction by machine learning algorithms based on cytokines in Kazakhs of China. , : 417-428.

Kasbe T.,Pippal R. S. .2018 .Enhancement in diagnosis of coronary artery disease using fuzzy expert system. Int. J. Sci. Res. Comput. Sci. Eng, 3(3) : 1324-1331.

Kumar A.,Das S.,Tyagi V.,Shaw R. N.,Ghosh A. .2021 .Analysis of classifier algorithms to detect anti-money laundering. Computationally intelligent systems and their applications, : 143-152.

Latif G.,Iskandar G.,Bashar D. A.,Alghazo A.,J. A. .2022 .Glioma Tumors' classification using deep-neuralnetwork-based features with SVM classifier. Diagnostics, 12(4) : 1018.

Maini E.,Venkateswarlu B.,Maini B.,Marwaha D. .2021 .Machine learning-based heart disease prediction system for Indian population: An exploratory study done in South India. medical journal armed forces india, 77(3) : 302-311.

Mienye I. D.,Sun Y.,Wang Z. .2020 .Improved sparse autoencoder based artificial neural network approach for prediction of heart disease. Informatics in Medicine Unlocked, 18 : 100307.

Ng M.,Fleming T.,Robinson M.,Thomson B.,Graetz N.,Margono C.,Gakidou E. .2014 .The lancet. , 384(9945) : 766-781.

Ogunpola A.,Saeed F.,Basurra S.,Albarrak A. M.,Qasem S. N. .2024 .Machine learning-based predictive models for detection of cardiovascular diseases. Diagnostics, 14(2) : 144.

Polat K.,Şahan S.,Güneş S. .2007 .Automatic detection of heart disease using an artificial immune recognition system (AIRS) with fuzzy resource allocation mechanism and k-nn (nearest neighbour) based weighting preprocessing. Expert Systems with Applications, 32(2) : 625-631.

Rani K. S.,Chaitanya M. S.,Kiran G. S. .2018 .A Heart Disease Prediction Model Using Logistic Regression By Cleveland DataBase. International Journal of Trend in Scientific Research and Development, 2(3) : 1467-1470.

Rippe J. M.,Angelopoulos T. J. .2019 .Lifestyle strategies for risk factor reduction, prevention and treatment of cardiovascular disease. Lifestyle Medicine, Third Edition, : 19-36.

Shah D. .2020 ."Heart disease prediction using machine learning techniques. " SN Computer Science, 1(6) : 345.

Shokrzade A.,Ramezani M.,Tab F. A.,Mohammad M. A. .2021 .A novel extreme learning machine based kNN classification method for dealing with big data. Expert Systems with Applications, 183 : 115293.

Tu M. C.,Shin D.,Shin D. .2009 .Effective diagnosis of heart disease through bagging approach. In 2009 2nd international conference on biomedical engineering and informatics, : 1-4.

Update A. S. .2017 ."Heart disease. update." Circulation, 135 : e146-e603.

Walek B. and P.,Fajmon B. and P. .2023 ."A hybrid recommender system for an online store using a fuzzy expert system. " Expert Systems with Applications, 212 : 118565.

Yang L.,Wu H.,Jin X.,Zheng P.,Hu S.,Xu X.,Yan J. .2020 .Study of cardiovascular disease prediction model based on random forest in eastern China. Scientific reports, 10(1) : 5245.