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

Brain Tumor Classification

Early classification and diagnosis of Brain Tumors are essential for providing the right treatment to a patient. It is crucial to get treatment as soon as possible because the survival rate for someone with an untreated brain tumor can range from as low as 3 months to as high as 5 years. In this project, we classified brain tumor images into 4 categories: glioma, meningioma, pituitary, and no tumor. With the use of baseline and deep learning models, the deep learning models demonstrated a significantly higher performance due to their ability to analyze images. The model with the highest accuracy was the MobileNet, a pre-trained transfer learning model trained on 5,608 images. This model yielded a validation accuracy of 98.24%. Using metrics including Kappa cohen score, precision, and recall, we validated the machine learning model's performance. We deployed the MobileNet model to a web app using Streamlit, where users submit MRI images and receive diagnoses of tumor class. We found that the model performed very well while utilizing the web app, indicating that it is safe to be used. However, since we only have 4 classes and there are over 150 total types of brain tumors, it could easily get a diagnosis wrong if it is not in one of these 4 classes.
Rohan S.
Mentored by Sriram Hathwar
Predicting Mental Health Conditions Using Student Demographic Information

With this, we will train the data to extrapolate what symptoms the user has based on their demographics and academic/social life. This can help understand what symptoms a person has of certain characteristics.
Ashwith Y.
Mentored by
Modeling Sleep Efficiency

This study explores how alcohol use, exercise, and smoking affect sleep quality by analyzing a dataset containing various lifestyle and sleep-related factors. Using a regression-based predictive model, the research identifies relationships between these behaviors and sleep outcomes, with the final linear regression model achieving a mean absolute error of 0.051.
Nimeesha K.
Mentored by Serena Hughes