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,
来源
Multimedia Tools and Applications | 2024年 / 83卷
关键词
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 条
[41]   Readmission Risk Prediction After Total Hip Arthroplasty Using Machine Learning and Hyperparameter Optimized with Bayesian Optimization [J].
Purbasari, Intan Yuniar ;
Bayuseno, Athanasius Priharyoto ;
Isnanto, R. Rizal ;
Winarni, Tri Indah .
INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2025, 16 (02) :887-898
[42]   Special Issue on Using Machine Learning Algorithms in the Prediction of Kyphosis Disease: A Comparative Study [J].
Dankwa, Stephen ;
Zheng, Wenfeng .
APPLIED SCIENCES-BASEL, 2019, 9 (16)
[43]   Advanced hyperparameter optimization of deep learning models for wind power prediction [J].
Hanifi, Shahram ;
Cammarono, Andrea ;
Zare-Behtash, Hossein .
RENEWABLE ENERGY, 2024, 221
[44]   Comparative analysis of machine learning models for coronary artery disease prediction with optimized feature selection [J].
Olawade, David B. ;
Soladoye, Afeez A. ;
Omodunbi, Bolaji A. ;
Aderinto, Nicholas ;
Adeyanju, Ibrahim A. .
INTERNATIONAL JOURNAL OF CARDIOLOGY, 2025, 436
[45]   CHURN PREDICTION - A COMPARATIVE ANALYSIS WITH SUPERVISED MACHINE LEARNING ALGORITHMS [J].
Gangadharan, Chika K. ;
Alex, Roshni ;
Sabu, M. K. .
ADVANCES AND APPLICATIONS IN MATHEMATICAL SCIENCES, 2021, 20 (12) :3049-3060
[46]   Mortality Prediction using Machine Learning Techniques: Comparative Analysis [J].
Verma, Akash ;
Goyal, Shreya ;
Thakur, Shridhar Kumar ;
Gupta, Archit ;
Gupta, Indrajeet .
PROCEEDINGS OF THE 2019 IEEE 9TH INTERNATIONAL CONFERENCE ON ADVANCED COMPUTING (IACC 2019), 2019, :230-234
[47]   Comparative Analysis of Stock Price Prediction Using Machine Learning [J].
Osama, Abdelrahman ;
Saeid, Hager ;
Mohsen, Seham ;
Eldin, Sarah Saad .
2024 INTERNATIONAL MOBILE, INTELLIGENT, AND UBIQUITOUS COMPUTING CONFERENCE, MIUCC 2024, 2024, :69-75
[48]   Comparative Analysis of Machine Learning Techniques for Cryptocurrency Price Prediction [J].
Salehi, Sara .
JOURNAL OF INFORMATION AND ORGANIZATIONAL SCIENCES, 2024, 48 (02) :341-352
[49]   Application of Machine Learning in Animal Disease Analysis and Prediction [J].
Zhang, Shuwen ;
Su, Qiang ;
Chen, Qin .
CURRENT BIOINFORMATICS, 2021, 16 (07) :972-982
[50]   Prediction Model for Bollywood Movie Success: A Comparative Analysis of Performance of Supervised Machine Learning Algorithms [J].
Verma, Hemraj ;
Verma, Garima .
REVIEW OF SOCIONETWORK STRATEGIES, 2020, 14 (01) :1-17