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
- Ahmed Mehaoua (ORCID: https://orcid.org/0000-0002-4631-6962)
- Osman Salem (ORCID: https://orcid.org/0000-0003-2076-9733)
- Quoc Bao Vo (ORCID: https://orcid.org/0000-0002-7404-110X)
- Philippe Ea (ORCID: https://orcid.org/0009-0000-6737-5067)
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