Predicting Heart Disease with Random Forest: An In-Depth Analysis of Machine Learning Techniques for Early Detection and Risk Assessment

Heart disease is a major health concern, and this study investigates how machine learning, particularly Random Forest (RF), can be used to identify it early and assess risk. We tested various algorithms using a large dataset (BRFSS 2022) and found that RF performed best, achieving an accuracy of 93% and an AUC of 98%. To make the model run faster, we identified the four most important features and focused on those. This reduced accuracy slightly (86% for RF and Extra Trees) but with AUCs of 0.92 and 0.91 respectively. Importantly, training and testing times improved significantly. This trade-off between accuracy and speed makes these models more suitable for real-world use, potentially aiding healthcare professionals in the early detection and prevention of heart disease.

Authors

Publication Details

Published
2024-11-18
DOI
https://doi.org/10.1109/healthcom60970.2024.10880829
Citations
3
Primary Topic
Artificial Intelligence in Healthcare
Type
article
Field-Weighted Citation Impact
2.12
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Predicting Heart Disease with Random Forest: An In-Depth Analysis of Machine Learning Techniques for Early Detection and Risk Assessment

Ahmed Mehaoua, Osman Salem, Quoc Bao Vo, Philippe Ea
3 citations
Artificial Intelligence in Healthcare
2.12
article

Predicting Heart Disease with Random Forest: An In-Depth Analysis of Machine Learning Techniques for Early Detection and Risk Assessment

Ahmed Mehaoua, Osman Salem, Quoc Bao Vo, Philippe Ea
article en
3 citations

Abstract

Heart disease is a major health concern, and this study investigates how machine learning, particularly Random Forest (RF), can be used to identify it early and assess risk. We tested various algorithms using a large dataset (BRFSS 2022) and found that RF performed best, achieving an accuracy of 93% and an AUC of 98%. To make the model run faster, we identified the four most important features and focused on those. This reduced accuracy slightly (86% for RF and Extra Trees) but with AUCs of 0.92 and 0.91 respectively. Importantly, training and testing times improved significantly. This trade-off between accuracy and speed makes these models more suitable for real-world use, potentially aiding healthcare professionals in the early detection and prevention of heart disease.

Openalex Percentile: Top 9%
Artificial Intelligence in Healthcare
2.12
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.