返回
Explainable diabetes classification using hybrid Bayesian-optimized TabNet architecture
DOI:10.1016/j.compbiomed.2022.106178.png)
摘要
En 中文
Diabetes is a deadly chronic disease that occurs when the pancreas is not able to produce ample insulin or when the body cannot use insulin effectively. If undetected, it may lead to a host of health complications. Hence, accurate and explainable early-stage detection of diabetes is essential for the proper administration of treatment options in leading a healthy and productive life. For this, we developed an interpretable TabNet model tuned via Bayesian optimization (BO). To achieve model-specific interpretability, the attention mechanism of TabNet ar-chitecture was used, which offered the local and global model explanations on the influence of the attributes on the outcomes. The model was further explained locally and globally using more robust model-agnostic LIME and SHAP eXplainable Artificial Intelligence (XAI) tools. The proposed model outperformed all benchmarked models by obtaining high accuracy of 92.2% and 99.4% using the Pima Indians diabetes dataset (PIDD) and the early -stage diabetes risk prediction dataset (ESDRPD), respectively. Based on the XAI results, it was clear that the most influential attribute for diabetes classification using PIDD and ESDRPD were Insulin and Polyuria, respectively. The feature importance values registered for insulin was 0.301 (PIDD) and for polyuria 0.206 was registered (ESDRPD). The high accuracy and ancillary interpretability of our objective model is expected to increase end -users trust and confidence in early-stage detection of diabetes.
Keyword:
Attention mechanism
Bayesian optimization
?Black-box? models
Diabetes classification
eXplainable artificial intelligence (XAI)
Interpretability
TabNet
期刊
IF:
6.3
论文数:
8.3K
被引数:
3.3W
机构
引用论文
A systematic study of the class imbalance problem in convolutional neural networks卷积神经网络中类不平衡问题的系统研究
NEURAL NETWORKS
IF6.3
A case-based ensemble learning system for explainable breast cancer recurrence prediction用于可解释的乳腺癌复发预测的基于案例的集成学习系统
Interpretable heartbeat classification using local model-agnostic explanations on ECGs使用ecg的局部模型不可知解释进行可解释的心跳分类
An Explainable Machine Learning Model for Early Detection of Parkinson's Disease using LIME on DaTSCAN Imagery一种可解释的机器学习模型,用于在DaTSCAN图像上使用LIME早期检测帕金森氏病
Prospects and limitations of non-invasive blood glucose monitoring using near-infrared spectroscopy近红外光谱无创血糖监测的前景和局限性

