Enhancing diabetes risk prediction: A comparative evaluation of bagging, boosting, and ensemble classifiers with SMOTE oversampling
Diabetes is a major global health concern, with millions of individuals at risk of developing this chronic condition. Early prediction and intervention are essential for effective diabetes management. This study explores advanced machine learning techniques, specifically bagging, boosting, and ensemble methods to improve diabetes risk prediction. Using three diverse datasets, namely, the Centers for Disease Control and Prevention (CDC) Diabetes Health Indicators dataset, the Early Stage Diabetes Risk Prediction System (ESDRP) dataset, and the PIMA Indian Diabetes dataset are utilized to evaluate the adaptability and robustness of the proposed models. Our approach addresses critical gaps in existing research, including the handling of highly imbalanced datasets through the Synthetic Minority Over- sampling Technique (SMOTE), the necessity of feature selection, and the underutilization of the CDC dataset in diabetes studies. We find that applying SMOTE to the CDC dataset significantly enhances model performance, with the CATBoost algorithm achieving an accuracy of 91%. For the ESRPS dataset, ensemble methods demonstrate even stronger results, achieving 98% accuracy using the top five features. This study not only contributes to the development of more accurate predictive models for diabetes risk but also provides insights into enhancing the robustness of machine learning methods in healthcare. • Enhance diabetes risk prediction through advanced machine learning techniques, including bagging, boosting, and ensemble methods, applied to well-known diabetes datasets. • Addresses class imbalance with the Synthetic Minority Over-sampling Technique (SMOTE), which improves model performance, particularly on the CDC dataset. • Incorporates weighted feature importance methods to identify the most influential factors affecting diabetes risk, enhancing model interpretability and robustness. • Provides improvements over existing research, setting a new benchmark for future diabetes risk prediction.
Authors
- Rabia Asif (ORCID: https://orcid.org/0000-0003-2155-0952)
- Darshana Upadhyay (ORCID: https://orcid.org/0000-0001-5822-0020)
- Marzia Zaman (ORCID: https://orcid.org/0000-0002-0610-0470)
- Srini Sampalli
Institutions
- Dalhousie University (CA)
- Cistel Technology (Canada) (CA)
Publication Details
- Journal
- Informatics in Medicine Unlocked
- Published
- 2025-01-01
- DOI
- https://doi.org/10.1016/j.imu.2025.101661
- Citations
- 4
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 11.75