arrow
返回

Explainable diabetes classification using hybrid Bayesian-optimized TabNet architecture

delete2022-12-01
delete40
PRE
AI
L
Lionel Joseph
E
Erica A. Joseph
R
Ramendra Prasad *
DOI:10.1016/j.compbiomed.2022.106178delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

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

期刊

Computers in Biology and Medicine 封面图
Computers in Biology and Medicine
IF:
6.3
论文数:
8.3K
被引数:
3.3W

机构

U
University of Southern Queensland
学者数:
4.1K
论文数: 4.8K
被引数: 18
引用论文

引用论文

err分享
err收藏
Interpretable heartbeat classification using local model-agnostic explanations on ECGs使用ecg的局部模型不可知解释进行可解释的心跳分类
err2021-06-01
err50
errOAAI
errNeves, Ines; Folgado, Duarte; Santos, Sara; Barandas, Marilia; Campagner, Andrea; Ronzio, Luca; Cabitza, Federico; Gamboa, Hugo
err分享
err收藏
err分享
err收藏
学者 查看更多内容