PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
May 24, 20260 citationsOpen Access

Diabetes Prediction Modeling Using XGBoost Ensemble Learning with SHAP Explainability: A Machine Learning Approach on the Pima Indians Diabetes Dataset

View Full Paper
KAKhan Gulrez Shagufa Fazal Ahmed

Key Points

  • The aim is to develop a machine learning pipeline for accurately predicting diabetes status.
  • Used Pima Indians Diabetes Dataset (n=768) for analysis.
  • Employed XGBoost classifier with 300 estimators and cross-validation methods.
  • Performed systematic data cleaning and feature engineering, including outlier clipping.
  • Achieved accuracy of 87.0%, recall of 85.2%, and precision of 79.3%.
  • Model produced an ROC-AUC score of 94.7%.
  • SHAP analysis identified Glucose, Glucose x BMI interaction, and BMI as key predictors.

Abstract

Diabetes mellitus is a chronic metabolic disorder affecting over 537 million adults worldwide. This study presents a complete end-to-end machine learning pipeline for binary classification of diabetes status using the Pima Indians Diabetes Dataset (n=768). The pipeline integrates systematic data cleaning, group-median imputation, IQR-based outlier clipping, and six engineered interaction features. An XGBoost classifier was trained with 300 estimators, class-weighted loss, and L1/L2 regularization. Cross-validation was performed using a scikit-learn Pipeline to prevent data leakage. The model achieved accuracy of 87.0%, recall of 85.2%, precision of 79.3%, F1 score of 82.1%, and ROC-AUC of 94.7%. Five-fold CV AUC was 0.944 (SD=0.014). SHAP analysis identified Glucose, Glucose x BMI interaction, and BMI as the three most impactful predictors. Source code, trained model artifacts, and figures are publicly available on GitHub (https://github.com/randomthingsonlineatsk-cloud/diabetes-xgboost-prediction) and archived on Zenodo (DOI: 10.5281/zenodo.20332710).

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Khan Gulrez Shagufa Fazal Ahmed (2026) studied this question.

synapsesocial.com/papers/6a12959d48a0ea1665671c9ahttps://doi.org/10.5281/zenodo.20336854
Ask AI
Helpful
Bookmark
Share
View Full Paper