Exploring the Efficiency of Advanced ML Techniques for Multi-Dimensional Cardiovascular Disease Prediction and Prognostic Analysis

Since cardiovascular diseases (CVDs) remain a main cause of death globally, accurate and efficientdiagnosis tools are necessary. This work explores the use of machine learning methods, namely K-NearestNeighbors (KNN) and Random Forest (RF), for the prediction of cardiovascular diseases based on theCleveland Heart Disease dataset. Including age, blood pressure, cholesterol, and ECG measurements, thedataset comprises 14 continuous and categorical traits covering 7,000 individuals. Many data preparationmethods feature scaling, data splitting, one-hot encoding, and handling missing values were applied toensure the dataset was fit for model training. The predictive power of both Random Forest and KNN wasselected; Random Forest is an ensemble method combining forecasts from numerous decision trees, whilstKNN is a non-parametric classification algorithm based on the majority vote of its nearest neighbors.Hyperparameter adjustment for both models using grid search and cross-validation helped to identifyoptimal values for model performance. The models were evaluated using key performance measurescomprising accuracy, sensitivity, specificity, and Area Under the ROC Curve (AUC).Random Forest performed better than KNN in all evaluated criteria: its 92% accuracy, 89% sensitivity,94% specificity, and 0.95 AUC show this. Conversely, KNN boasts an accuracy of 86% but a reducedsensitivity and specificity. The results show that since Random Forest is better at separating patients withcardiovascular illness from those without, it offers a potential tool for early diagnosis. The results of thestudy indicate that by increasing early identification of cardiovascular diseases, machine learning modelsRandom Forest have enormous potential to improve patient outcomes and save healthcare expenses.

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2025-04-14
DOI
https://doi.org/10.5281/zenodo.15207914
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

Exploring the Efficiency of Advanced ML Techniques for Multi-Dimensional Cardiovascular Disease Prediction and Prognostic Analysis

Zenodo (CERN European Organization for Nuclear Research)
Artificial Intelligence in Healthcare
article

Exploring the Efficiency of Advanced ML Techniques for Multi-Dimensional Cardiovascular Disease Prediction and Prognostic Analysis

article en

Abstract

Since cardiovascular diseases (CVDs) remain a main cause of death globally, accurate and efficientdiagnosis tools are necessary. This work explores the use of machine learning methods, namely K-NearestNeighbors (KNN) and Random Forest (RF), for the prediction of cardiovascular diseases based on theCleveland Heart Disease dataset. Including age, blood pressure, cholesterol, and ECG measurements, thedataset comprises 14 continuous and categorical traits covering 7,000 individuals. Many data preparationmethods feature scaling, data splitting, one-hot encoding, and handling missing values were applied toensure the dataset was fit for model training. The predictive power of both Random Forest and KNN wasselected; Random Forest is an ensemble method combining forecasts from numerous decision trees, whilstKNN is a non-parametric classification algorithm based on the majority vote of its nearest neighbors.Hyperparameter adjustment for both models using grid search and cross-validation helped to identifyoptimal values for model performance. The models were evaluated using key performance measurescomprising accuracy, sensitivity, specificity, and Area Under the ROC Curve (AUC).Random Forest performed better than KNN in all evaluated criteria: its 92% accuracy, 89% sensitivity,94% specificity, and 0.95 AUC show this. Conversely, KNN boasts an accuracy of 86% but a reducedsensitivity and specificity. The results show that since Random Forest is better at separating patients withcardiovascular illness from those without, it offers a potential tool for early diagnosis. The results of thestudy indicate that by increasing early identification of cardiovascular diseases, machine learning modelsRandom Forest have enormous potential to improve patient outcomes and save healthcare expenses.

Zenodo (CERN European Organization for Nuclear Research)
Openalex Percentile: Top 43%
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.