• E-ISSN3022-5388

Article Detail

Home > Article Detail
  • E-ISSN 3022-5388

A Comparative Study on the Performance of Machine Learning Algorithms and Key Feature Analysis for Predicting Heart Attack

Journal of Korean Artificial Intelligence Association / Journal of Korean Artificial Intelligence Association, (E)3022-5388
2024, v.2 no.2, pp.31-37
https://doi.org/10.24225/jkaia.2024.2.2.31
KOH JunSu (Eulji University)
KANG MinSoo (Eulji University)
HAN Dong Hoon (MIIC)
  • Downloaded
  • Viewed

Abstract

In this study, we compare the performance of various machine learning algorithms for predicting heart attacks, a major cause of mortality globally, with a focus on identifying key predictive features. Using a dataset of 918 records, the research evaluates models such as Random Forest, Logistic Regression, XGBoost, SVM, KNN, and Decision Tree to enhance prediction accuracy for heart attack risks. The methodology emphasizes robust preprocessing techniques, including feature scaling and handling class imbalances through Stratified K-Fold cross-validation, to improve model reliability. Results reveal that ensemble models, particularly Random Forest, achieve the highest ROC AUC score of 0.9301, significantly outperforming traditional algorithms. Key predictors, such as ST_Slope, were identified as critical variables in determining heart attack risks, while less influential features, such as RestingECG, had minimal impact. The findings underscore the efficacy of ensemble learning in predicting heart attacks and highlight the importance of feature importance analysis in enhancing model interpretability. This study provides valuable insights into the integration of machine learning in personalized healthcare, offering a foundation for future research to refine predictive models and improve early detection and prevention strategies for cardiovascular diseases.

keywords
Heart Attack, Machine learning, Decision Tree, Random Forest, Recall, ROC AUC


Submission Date
2024-11-13
Revised Date
2024-12-12
Accepted Date
2024-12-14
상단으로 이동

Journal of Korean Artificial Intelligence Association