Towards Transparent Diabetes Prediction: Combining AutoML and Explainable AI for Improved Clinical Insights

Raza Hasan*, Vishal Dattana, Salman Mahmood, Saqib Hussain

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Diabetes is a global health challenge that requires early detection for effective management. This study integrates Automated Machine Learning (AutoML) with Explainable Artificial Intelligence (XAI) to improve diabetes risk prediction and enhance model interpretability for healthcare professionals. Using the Pima Indian Diabetes dataset, we developed an ensemble model with 85.01% accuracy leveraging AutoGluon’s AutoML framework. To address the “black-box” nature of machine learning, we applied XAI techniques, including SHapley Additive exPlanations (SHAP), Local Interpretable Model-Agnostic Explanations (LIME), Integrated Gradients (IG), Attention Mechanism (AM), and Counterfactual Analysis (CA), providing both global and patient-specific insights into critical risk factors such as glucose and BMI. These methods enable transparent and actionable predictions, supporting clinical decision-making. An interactive Streamlit application was developed to allow clinicians to explore feature importance and test hypothetical scenarios. Cross-validation confirmed the model’s robust performance across diverse datasets. This study demonstrates the integration of AutoML with XAI as a pathway to achieving accurate, interpretable models that foster transparency and trust while supporting actionable clinical decisions.
Original languageEnglish
Article number7
Number of pages31
JournalInformation
Volume16
Issue number1
DOIs
Publication statusPublished - 26 Dec 2024

Cite this