Hyperparameter optimization: a comparative machine learning model analysis for enhanced heart disease prediction accuracy

被引:0
|
作者
Yagyanath Rimal
Navneet Sharma
机构
[1] IIS Deemed to be University,
来源
关键词
Bayesian optimization; Genetic optimization; GAsearchCV optimization; Optuna optimization; Gaussian; Random forest; Support vector machine; Principal component analysis;
D O I
暂无
中图分类号
学科分类号
摘要
An optimizer is the process of hyperparameter tuning that updates the machine learning model after each step of weight loss adjustment of input features. The permutation and combination of high and low learning rates with various step sizes ultimately leads to an optimal tuning model. The step size and learning rate sometimes take much smaller steps, allowing the derivatives of tangent to gradually reach global minima. The primary goal of this study is to compare the prediction accuracy of enhanced heart disease using various optimization algorithms. Heart disease treatment requires ensemble hyperparameter tuning for accurate prediction and classification due to multiple feature dependencies. The study analyzed model tuning techniques using the AUC and confusion matrix, revealing improvements in precision, recall, and f1 score from default to optimized models. The Hyper-opt in Bayesian optimizer and T-pot classifiers were used in genetic populations and offspring with 5 and 10 generations, while using Optuna optimization frozen trails was combined with a random forest algorithm. The default random forest (86.6%), Bayesian optimization with random forest (89%), and Bayesian optimization with support vector machines (90%) scored the highest accuracy among all. The generic algorithm with five generations (86.8%) and GAsearchCV with 10 generations (88.5%) scored the second highest accuracy, while Optuna's support vector machine model (84%) scored the least accuracy, respectively. This research further compares the machine learning accuracy, precision, recall, F1 score, macro average, and confusion matrix of each optimized model with their model's actual performance execution time. The predictive accuracy from exploratory data analysis and data pre-processing was further tested after the pipeline design of one-hot encoding and standard scaling of enhanced (31-featured) data sets and heart disease data (13 features). The gaussian algorithm (84%), logistic regression (83%), and classification models predict with higher accuracy than dummy classifiers (54%), when compared with standalone default machine learning models.
引用
收藏
页码:55091 / 55107
页数:16
相关论文
共 50 条
  • [31] Machine Learning Assisted Hyperparameter Tuning for Optimization
    Linkous, Lauren
    Lundquist, Jonathan
    Suche, Michael
    Topsakal, Erdem
    2024 IEEE INC-USNC-URSI RADIO SCIENCE MEETING (JOINT WITH AP-S SYMPOSIUM), 2024, : 107 - 108
  • [32] Sherpa: Robust hyperparameter optimization for machine learning
    Hertel, Lars
    Collado, Julian
    Sadowski, Peter
    Ott, Jordan
    Baldi, Pierre
    SOFTWAREX, 2020, 12
  • [33] Heart Disease Prediction Using Logistic Regression Machine Learning Model
    Hrvat, Faris
    Spahic, Lemana
    Aleta, Amina
    MEDICON 2023 AND CMBEBIH 2023, VOL 1, 2024, 93 : 654 - 662
  • [34] Prediction of Heart Disease Based on Machine Learning Using Jellyfish Optimization Algorithm
    Ahmad, Ahmad Ayid
    Polat, Huseyin
    DIAGNOSTICS, 2023, 13 (14)
  • [35] Hybrid XGBoost model with hyperparameter tuning for prediction of liver disease with better accuracy
    Dalal, Surjeet
    Onyema, Edeh Michael
    Malik, Amit
    WORLD JOURNAL OF GASTROENTEROLOGY, 2022, 28 (46) : 6551 - 6563
  • [36] Hybrid XGBoost model with hyperparameter tuning for prediction of liver disease with better accuracy
    Surjeet Dalal
    Edeh Michael Onyema
    Amit Malik
    World Journal of Gastroenterology, 2022, 28 (46) : 6551 - 6563
  • [37] A Comparative Study of Machine Learning Models with Hyperparameter Optimization Algorithm for Mapping Mineral Prospectivity
    Lin, Nan
    Chen, Yongliang
    Liu, Haiqi
    Liu, Hanlin
    MINERALS, 2021, 11 (02) : 1 - 31
  • [38] Predicting hydropower generation: A comparative analysis of Machine learning models and optimization algorithms for enhanced forecasting accuracy and operational efficiency
    Wang, Chunyang
    Li, Chao
    Feng, Yudong
    Wang, Shoufeng
    AIN SHAMS ENGINEERING JOURNAL, 2025, 16 (03)
  • [39] RETRACTED: An Intelligent and Reliable Hyperparameter Optimization Machine Learning Model for Early Heart Disease Assessment Using Imperative Risk Attributes (Retracted Article)
    Ansarullah, Syed Immamul
    Saif, Syed Mohsin
    Andrabi, Syed Abdul Basit
    Kumhar, Sajadul Hassan
    Kirmani, Mudasir M.
    Kumar, Pradeep
    JOURNAL OF HEALTHCARE ENGINEERING, 2022, 2022
  • [40] Hyperparameter optimization for machine learning models based on Bayesian optimization
    Wu J.
    Chen X.-Y.
    Zhang H.
    Xiong L.-D.
    Lei H.
    Deng S.-H.
    Journal of Electronic Science and Technology, 2019, 17 (01) : 26 - 40