Feature Importance and Correlation for Enhancing ML Performance in Predictive Healthcare Analytics

Cardiovascular disease (CVD) is a predominant cause of death worldwide, highlighting the critical necessity for effective early detection and accurate diagnostic systems. As healthcare technology advances, machine learning has emerged as a transformative tool to improve diagnostic precision. This study investigates the impact of feature correlation on the performance of machine learning models for heart disease prediction across diverse datasets: Heart Disease Dataset (1190 instances) and Cardiovascular Disease (42,500 instances). We evaluated nine algorithms: Logistic Regression (LR), Extreme Gradient Boosting (XGBoost), Random Forest (RF), Decision Tree (DT), Naive Bayes (NB), Support Vector Machine (SVM), K-Nearest Neigh-bor(KNN), Gradient Boosting (GB), and Multilayer Perceptron (MLP) using performance metrics such as accuracy, precision, recall, and F1 score. The analysis reveals that models like XG-Boost, RF, and MLP consistently deliver superior performance in datasets where key features are strongly correlated with the target outcome. Conversely, datasets lacking such correlations, particularly the Cardiovascular Disease dataset, exhibit significantly lower accuracy and overall model performance. These findings emphasize the crucial role of feature importance and correlation in shaping model accuracy and robustness, indicating that predictive model effectiveness depends on the relevance and strength of the selected features.

Authors

Institutions

Publication Details

Published
2024-12-20
DOI
https://doi.org/10.1109/iccit64611.2024.11021982
Primary Topic
Artificial Intelligence in Healthcare
Type
article
Field-Weighted Citation Impact
0.00

Funders

Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Feature Importance and Correlation for Enhancing ML Performance in Predictive Healthcare Analytics

Md. Saidur Rahman Kohinoor, Khadiza Akther, Humayra Kabir Nisa, Sajid Abdullah Al-Hafiz
Artificial Intelligence in Healthcare
article

Feature Importance and Correlation for Enhancing ML Performance in Predictive Healthcare Analytics

Md. Saidur Rahman Kohinoor, Khadiza Akther, Humayra Kabir Nisa, Sajid Abdullah Al-Hafiz
article en

Abstract

Cardiovascular disease (CVD) is a predominant cause of death worldwide, highlighting the critical necessity for effective early detection and accurate diagnostic systems. As healthcare technology advances, machine learning has emerged as a transformative tool to improve diagnostic precision. This study investigates the impact of feature correlation on the performance of machine learning models for heart disease prediction across diverse datasets: Heart Disease Dataset (1190 instances) and Cardiovascular Disease (42,500 instances). We evaluated nine algorithms: Logistic Regression (LR), Extreme Gradient Boosting (XGBoost), Random Forest (RF), Decision Tree (DT), Naive Bayes (NB), Support Vector Machine (SVM), K-Nearest Neigh-bor(KNN), Gradient Boosting (GB), and Multilayer Perceptron (MLP) using performance metrics such as accuracy, precision, recall, and F1 score. The analysis reveals that models like XG-Boost, RF, and MLP consistently deliver superior performance in datasets where key features are strongly correlated with the target outcome. Conversely, datasets lacking such correlations, particularly the Cardiovascular Disease dataset, exhibit significantly lower accuracy and overall model performance. These findings emphasize the crucial role of feature importance and correlation in shaping model accuracy and robustness, indicating that predictive model effectiveness depends on the relevance and strength of the selected features.

King Fahd University of Petroleum and Minerals (SA), Sylhet International University (BD)
BD
Openalex Percentile: Top 59%
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.