Enhancing Heart Disease Model Accuracy: A Study of Optimization Hyperparameter Tuning Techniques in Machine Learning Algorithms

Heart disease (HD) is a major global health issue, This study evaluates Machine learning (ML) models such as Logistic Regression (LR), Decision Tree (DT), K- Nearest Neighbors (KNN), and Support Vector Machine (SVM) for heart disease dataset using Train-Test Split(TTS) and K-Fold Cross-Validation (KF-CV). Initially, accuracy rates improved with KF-CV across all models. The DT model achieved 98.54 percent accuracy with TTS and 99.32 percent with KF-CV in the pre-optimization phase. To further enhance accuracy, four advanced optimization techniques—Grid Search CV, Random Search CV, Ant Colony Optimization (ACO), and Particle Swarm Optimization (PSO) were applied. The findings revealed that the DT model achieved perfect score of 100 percent accuracy with Grid Search CV and Random Search CV. SVM and KNN models also showed significant improvements with advanced tuning. PSO, in particular, was effective for optimizing hyperparameters of the LR model. This study underscores the value of advanced optimization techniques in improving the accuracy of heart disease prediction models.

Authors

Publication Details

Published
2024-09-13
DOI
https://doi.org/10.1109/icietdw61607.2024.10941101
Primary Topic
Artificial Intelligence in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Enhancing Heart Disease Model Accuracy: A Study of Optimization Hyperparameter Tuning Techniques in Machine Learning Algorithms

Shashank Bhargav, Om Prakash S, Charles Sagayaraj A C
Artificial Intelligence in Healthcare
article

Enhancing Heart Disease Model Accuracy: A Study of Optimization Hyperparameter Tuning Techniques in Machine Learning Algorithms

Shashank Bhargav, Om Prakash S, Charles Sagayaraj A C
article en

Abstract

Heart disease (HD) is a major global health issue, This study evaluates Machine learning (ML) models such as Logistic Regression (LR), Decision Tree (DT), K- Nearest Neighbors (KNN), and Support Vector Machine (SVM) for heart disease dataset using Train-Test Split(TTS) and K-Fold Cross-Validation (KF-CV). Initially, accuracy rates improved with KF-CV across all models. The DT model achieved 98.54 percent accuracy with TTS and 99.32 percent with KF-CV in the pre-optimization phase. To further enhance accuracy, four advanced optimization techniques—Grid Search CV, Random Search CV, Ant Colony Optimization (ACO), and Particle Swarm Optimization (PSO) were applied. The findings revealed that the DT model achieved perfect score of 100 percent accuracy with Grid Search CV and Random Search CV. SVM and KNN models also showed significant improvements with advanced tuning. PSO, in particular, was effective for optimizing hyperparameters of the LR model. This study underscores the value of advanced optimization techniques in improving the accuracy of heart disease prediction models.

Good health and well-being
Openalex Percentile: Top 61%
Artificial Intelligence in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.