Comparative Evaluation of Machine Learning Models for Cardiovascular Disease Prediction
DOI:
https://doi.org/10.62051/ns1f4t40Keywords:
Random Forest; Logistic Regression; XGBoost; Cardiovascular Diseases.Abstract
Cardiovascular diseases (CVDs) persist as a critical global health burden, accounting for over 30% of annual mortality worldwide. While machine learning approaches show promise in early-stage risk identification, existing research predominantly focuses on isolated model validation without systematic comparative analysis. To address this gap, our study conducts a rigorous multi-algorithm evaluation, leveraging Logistic Regression (LR), Random Forest (RF), and XGBoost classifiers trained on clinically validated CVD datasets. Methodologically, this study implement stratified 5-fold cross-validation with adaptive class-weight balancing to mitigate data imbalance issues, while embedding recursive feature elimination for optimal predictor selection. Performance benchmarking across three critical dimensions-precision, recall (sensitivity), and ROC-AUC-reveals RF’s consistent superiority in both discriminative power and operational stability. Specifically, RF generates the most favorable sensitivity-specificity trade-off curve and attains significantly higher diagnostic sensitivity compared to LR and XGBoost, essential for minimizing false negatives in clinical screening scenarios. The validated model establishes a scalable risk stratification pipeline for community healthcare systems, enabling timely interventions while demonstrating methodological replicability for predictive epidemiology in resource-constrained primary care settings. This framework bridges a critical feasibility gap in translating algorithmic innovations into deployable clinical tools.
Downloads
References
[1] Miao Lipeng, Ren Kehao, LI Mengdie, et al. Trend Analysis and Prediction of Cardiovascular Disease Mortality in China from 2009 to 2021 [J]. Chinese General Practice, 2024, 27 (18): 2260-2264.
[2] Mehra R. Global public health problem of sudden cardiac death [J]. Journal of electrocardiology, 2007, 40 (6): S118-S122. DOI: https://doi.org/10.1016/j.jelectrocard.2007.06.023
[3] Yaseliani M, Khedmati M. Prediction of heart diseases using logistic regression and likelihood ratios [J]. International Journal of Industrial Engineering & Production Research, 2023, 34 (1): 1-15.
[4] Ogunpola A, Saeed F, Basurra S, et al. Machine learning-based predictive models for detection of cardiovascular diseases [J]. Diagnostics, 2024, 14 (2): 144. DOI: https://doi.org/10.3390/diagnostics14020144
[5] Azmi J, Arif M, Nafis M T, et al. A systematic review on machine learning approaches for cardiovascular disease prediction using medical big data [J]. Medical engineering & physics, 2022, 105: 103825. DOI: https://doi.org/10.1016/j.medengphy.2022.103825
[6] Shi Shengyuan, Zhu Lei, Ye Lin, etc Research on Cardiovascular Disease Prediction Based on Random Forest Algorithm [J]. Intelligent Computer and Application, 2021, 11 (04): 176-178.
[7] Ye Suting, Pan Yuanyuan, Bi Yingchun Research on Heart Disease Early Warning Model Based on Decision Tree Algorithm [J]. Computer Knowledge and Technology, 2020, 16 (19): 187-189.
[8] Feng Yuan, Yu Rongrong, Sun Ziwei Research on the Classification and Prediction Model of Cardiac Cases [J]. Advances in Applied Mathematics, 2024, 13:4610. DOI: https://doi.org/10.12677/aam.2024.1310442
[9] Yang J, Guan J. A heart disease prediction model based on feature optimization and smote-Xgboost algorithm [J]. Information, 2022, 13 (10): 475. DOI: https://doi.org/10.3390/info13100475
[10] Ambrish G, Ganesh B, Ganesh A, et al. Logistic regression technique for prediction of cardiovascular disease [J]. Global Transitions Proceedings, 2022, 3 (1): 127-130. DOI: https://doi.org/10.1016/j.gltp.2022.04.008
[11] Magidi J, Nhamo L, Mpandeli S, et al. Application of the random forest classifier to map irrigated areas using google earth engine [J]. Remote Sensing, 2021, 13 (5): 876. DOI: https://doi.org/10.3390/rs13050876
Downloads
Published
Issue
Section
License
Copyright (c) 2025 Transactions on Computer Science and Intelligent Systems Research

This work is licensed under a Creative Commons Attribution-NonCommercial 4.0 International License.








