Enhancing Heart Disease Prediction through Ensemble Learning Techniques with Hyperparameter Optimization

被引:23
|
作者
Asif, Daniyal [1 ]
Bibi, Mairaj [1 ]
Arif, Muhammad Shoaib [2 ,3 ]
Mukheimer, Aiman [2 ]
机构
[1] COMSATS Univ Islamabad, Dept Math, Pk Rd, Islamabad 45550, Pakistan
[2] Prince Sultan Univ, Coll Humanities & Sci, Dept Math & Sci, Riyadh 11586, Saudi Arabia
[3] Air Univ, Dept Math, PAF Complex E-9, Islamabad 44000, Pakistan
关键词
heart disease; machine learning; ensemble learning; hyperparameter optimization; extra tree; XGBoost; CatBoost;
D O I
10.3390/a16060308
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Heart disease is a significant global health issue, contributing to high morbidity and mortality rates. Early and accurate heart disease prediction is crucial for effectively preventing and managing the condition. However, this remains a challenging task to achieve. This study proposes a machine learning model that leverages various preprocessing steps, hyperparameter optimization techniques, and ensemble learning algorithms to predict heart disease. To evaluate the performance of our model, we merged three datasets from Kaggle that have similar features, creating a comprehensive dataset for analysis. By employing the extra tree classifier, normalizing the data, utilizing grid search cross-validation (CV) for hyperparameter optimization, and splitting the dataset with an 80:20 ratio for training and testing, our proposed approach achieved an impressive accuracy of 98.15%. These findings demonstrated the potential of our model for accurately predicting the presence or absence of heart disease. Such accurate predictions could significantly aid in early prevention, detection, and treatment, ultimately reducing the mortality and morbidity associated with heart disease.
引用
收藏
页数:17
相关论文
共 50 条
  • [1] Enhancing Heart Disease Prediction Accuracy through Machine Learning Techniques and Optimization
    Chandrasekhar, Nadikatla
    Peddakrishna, Samineni
    PROCESSES, 2023, 11 (04)
  • [2] Disease prediction via Bayesian hyperparameter optimization and ensemble learning
    Gao Liyuan
    Ding Yongmei
    BMC RESEARCH NOTES, 2020, 13 (01)
  • [3] Disease prediction via Bayesian hyperparameter optimization and ensemble learning
    Liyuan Gao
    Yongmei Ding
    BMC Research Notes, 13
  • [4] Hyperparameter Optimization for Software Bug Prediction Using Ensemble Learning
    Al-Fraihat, Dimah
    Sharrab, Yousef
    Al-Ghuwairi, Abdel-Rahman
    Alshishani, Hamzeh
    Algarni, Abdulmohsen
    IEEE ACCESS, 2024, 12 : 51869 - 51878
  • [5] Enhancing groundwater quality prediction through ensemble machine learning techniques
    Karimi, Hadi
    Sahour, Soheil
    Khanbeyki, Matin
    Gholami, Vahid
    Sahour, Hossein
    Shahabi-Ghahfarokhi, Sina
    Mohammadi, Mohsen
    ENVIRONMENTAL MONITORING AND ASSESSMENT, 2024, 197 (01)
  • [6] Prediction of heart disease using Ensemble Learning and Particle Swarm Optimization
    Yekkala, Indu
    Dixit, Sunanda
    Jabbar, M. A.
    PROCEEDINGS OF THE 2017 INTERNATIONAL CONFERENCE ON SMART TECHNOLOGIES FOR SMART NATION (SMARTTECHCON), 2017, : 691 - 698
  • [7] A comparative analysis of ensemble learning algorithms with hyperparameter optimization for soil liquefaction prediction
    Demir, Alparslan Serhat
    Kurnaz, Talas Fikret
    Koekcam, Abdullah Hulusi
    Erden, Caner
    Dagdeviren, Ugur
    ENVIRONMENTAL EARTH SCIENCES, 2024, 83 (09)
  • [8] Hyperparameter optimization: a comparative machine learning model analysis for enhanced heart disease prediction accuracy
    Rimal, Yagyanath
    Sharma, Navneet
    MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 83 (18) : 55091 - 55107
  • [9] Hyperparameter optimization: a comparative machine learning model analysis for enhanced heart disease prediction accuracy
    Yagyanath Rimal
    Navneet Sharma
    Multimedia Tools and Applications, 2024, 83 : 55091 - 55107
  • [10] An Effective Heart Disease Prediction Framework based on Ensemble Techniques in Machine Learning
    Yewale, Deepali
    Vijayaragavan, S. P.
    Bairagi, V. K.
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (02) : 182 - 190