A Comparative Study of Machine Learning Models for Cardiovascular Risk Prediction Using PCA-Transformed Framingham Data
Students & Supervisors
Student Authors
Supervisors
Abstract
Cardiovascular disease (CVD) remains the leadingcause of mortality globally, highlighting the urgent need foreffective early detection systems. Supervised machine learningalgorithms, such as Logistic Regression (LR), SGDClassifier(SDGC), Support Vector Classifier (SVC), K-Nearest Neighbors (KNN), Decision Tree (DT), Random Forest (RF), GradientBoosting (GB), AdaBoost (AB), LightGBM (LGBM), and XG-Boost (XGB), are applied to the Framingham Heart Study (FHS) data set to predict mortality risk and are presented in this research in a comparative analysis. The proposed research includes Principal Component Analysis (PCA) for dimensionality reduction to reduce model dimensionality and improve model efficiency and generalization. The accuracy, precision, recall, F1 score, and AUC-ROC were used as performance metrics for the model, and 10-fold cross validation was used to test the stability of the model. More results show that ensemble models are much better at predictive accuracy and generalization than linear and moderate accuracy classifiers. Test accuracy was 98.05% and F1 score 0.98 for RF and 98.04% for XGB with a similar robust-ness and greater precision recall balance. PCA is also shown to successfully retain important variance and decrease model complexity in the research. Integration of ensemble learning and dimensionality reduction techniques into wearable compatible, real time cardiovascular monitoring systems helps to scale and accurately provide an early intervention for at risk people, and these findings support it.
Keywords
Publication Details
- Type of Publication: Conference
- Conference Name: International Conference on Quantum Photonics, Artificial Intelligence, and Networking (QPAIN)
- Date of Conference: 31/07/2025 - 31/07/2025
- Venue: Bangladesh Army University of Science and Technology (BAUST), Saidpur, Nilphamari, Rangpur Division, Bangladesh
- Organizer: IEEE Photonics Society Bangladesh Chapter