Back To Projects

Using Machine Learning Models to Predict Heart Disease and Attack with Non-Clinical and Non-Wearable Variables

Samuel P.

This study uses data from the CDC’s Behavioral Risk Factor Surveillance System (BRFSS) to train machine learning models that predict heart disease risk based on non-clinical factors. The research demonstrates the potential for cost-effective, data-driven screening tools for early heart disease detection, though further improvements are needed for real-world application.


Heart disease is the leading cause of death in the United States for over 100 years. However, electrocardiograms (EKG), the primary method of testing for heart issues, costs the average American ~100-5000 dollars per scan. Therefore, there should be other cheaper and time efficient methods for other methods of early onset heart disease detection. We used data from the Behavioral Risk Factor Surveillance System (BRFSS), a telephoned based survey with over 200 thousand participants and partnered with theCenters for Disease Control and Prevention (CDC), to find correlations between heart disease and attacks with non-clinical factors. Multiple machine learning models were trained on BRFSS 2015 data. The best model achieved 70% accuracy, while having a 21% precision and a 79% recall. These results show that while there is potential using non-clinical data combined with machine learning to provide a cost-effective preliminary screening tool for heart disease risk, significant improvements are needed before they can be applied effectively in real-world healthcare.

Explore More!

Samuel P.
Rami Abi-Akl

Related Projects

workspace_premium
The Utilization of Artificial Intelligence in Enabling the Early Detection of Brain Tumors

This research aims to investigate the application of machine learning to enhance diagnosis.
Shanzeh H.
Mentored by Odysseas Drosis
Computational Approaches to Enhance Idiopathic Intracranial Hypertension Diagnosis: A Neural Network-Based Framework for Improved Clinical Decision Support

A computer model, trained on eye images, can tell if optic nerves are healthy or not with 97% accuracy. This is important because it messes up less often than people do, making it a useful tool. The model looks at pictures of the back of the eye (fundus imaging) to make these predictions.
Rohan B.
Mentored by
Skin Cancer Detection

The goal of this research project is to predict whether or not a patient has skin cancer through a machine learning model that is developed from an image dataset. Skin cancer is extremely dangerous, as over 9500 people in the US are diagnosed with it daily. If detected early, patients will have a more likely chance of survival. I tested an MLP Classifier, Decision Tree Regressor, a Logistic Regression Model, and a KNN Model to compare various results and ultimately determine the best accuracy. The MLP Classifier had a 74.5% accuracy, the Decision Tree Regressor had a 74.1% accuracy, the Logistic Regression Model had a 68.8% accuracy, and the KNN Model had a 74.6% accuracy (all testing). We can see that the MLP Classifier, Decision Tree Regressor, and the KNN Model had around the same accuracy while outperforming the Logistic Regression Model. However, when comparing training data, there seems to be a large overfitting problem with most of the models.
Jaida G.
Mentored by Odysseas Drosis