Computer Science & Electrical

Computer Science & Electrical

Archive
Join as an Editor/Reviewer

Analyzing the major contribution factor for heart attack using ML algorithms and LIME

Volume: 155  ,  Issue: 1 , August    Published Date: 25 August 2024
Publisher Name: IJRP
Views: 245  ,  Download: 79 , Pages: 151 - 158    
DOI: 10.47119/IJRP1001551820247098

Authors

# Author Name
1 Bishal Ghimire
2 Manoj Rai
3 Nishant Uprety
4 Rubim Shrestha

Abstract

Heart disease, a leading cause of global mortality, can significantly impact overall health. Timely prediction and identification of key risk factors are essential yet challenging. This study employs Machine Learning and Explainable Artificial Intelligence (XAI) to predict and identify major contributors to heart disease. Using a publicly available cardiovascular disease (CVD) dataset from Kaggle, consisting of 1319 samples with eight variables—age, gender, heart rate, systolic and diastolic blood pressure, blood sugar, CK-MB, and Test-Troponin—we trained and evaluated models for heart attack prediction. Supervised learning techniques, including SVM, Decision Tree, and Random Forest, were compared. The LIME technique was used to elucidate the influence of factors in different test cases. The study aimed to enhance classification and detection of key features leading to heart attacks. Model performance was assessed using metrics such as confusion matrix, accuracy, precision, recall, and F1 score. In the medical context, minimizing false negatives (missed actual positive cases) is critical. Random Forest emerged as the most effective model for reducing false negatives, making it a promising tool for heart disease prediction.

Keywords

  • explainable ai
  • lime
  • heart attack