A Comparative Study of Machine Learning Approaches for Diabetes Risk Prediction: Insights from SHAP and Feature Importance
This study leverages machine learning models, including Logistic Regression, XGBoost, and Random Forest, to predict diabetes using BRFSS data. XGBoost emerged as the topperforming model, achieving an AUC of 0.83. Feature importance was analyzed using SHAP values, highlighting general health, high blood pressure, and BMI as key predictors. This research demonstrates the potential of machine learning models in identifying individuals at risk of diabetes, aiding in early intervention and improved public health outcomes.
Authors
- Qimin Zhang (ORCID: https://orcid.org/0000-0002-6099-2799)
- Huili Zheng (ORCID: https://orcid.org/0000-0002-3991-7743)
- Shaohan Chen (ORCID: https://orcid.org/0000-0001-6843-2027)
- Yiru Gong
- Zheyan Liu
Institutions
- Columbia University (US)
Publication Details
- Published
- 2024-10-18
- DOI
- https://doi.org/10.1109/icmlca63499.2024.10754491
- Citations
- 7
- Primary Topic
- Artificial Intelligence in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 4.95