用视觉Transformer ensemble提升肾病分期准确率,还让模型决策可解释。
XEns-CKD: An Explainable Ensemble-Based Approach for Chronic Kidney Disease Stage Detection
- 三组不同参数的ViT模型集成,基于超声图像做肾病分期。
- 整体准确率达86.36%,比现有方法高4个百分点。
- 结合注意力图技术,清晰展示病变区域变化,适合临床可信使用。
慢性肾病(CKD)是一种隐匿性疾病,其进展可能在日常生活中无明显表现。人肾功能可分为正常或五期CKD。早期识别分期有助于患者了解肾功能状态并采取措施延缓进展。本文提出XEns-CKD,一种基于视觉变压器(ViT)的新型集成方法,利用超声图像进行CKD分期分类。三个ViT在私有超声图像数据集上以不同训练参数训练,分别评估宏敏感性、宏特异性、宏精确率、宏F1分数、宏约登指数、马修斯相关系数(MCC)和宏平衡准确率。集成模型总体分类准确率为86.36%。本研究还关注识别并解释受CKD进展影响的肾区。采用LIME、LRP、Attention-Min与Attention-Max等可解释AI技术增强模型透明度与临床信任。结合Attention-Min与Attention-Max结果的注意力图,有效识别并解释了从一阶段到另一阶段的肾区变化,同时揭示了病情进展的影响。相比现有方法,该方法在五期CKD与正常肾状态分类上实现4%的准确率提升。
原文摘要 · Abstract (English)
Chronic kidney disease (CKD) is a silent disease. Its progression may not significantly hamper a person's daily routine. Human kidney function can be classified as normal or as one of the five stages of CKD. Early detection of the CKD stage can help patients understand the functional status of their kidneys and follow medical advice to slow CKD progression. In this paper, we propose XEns-CKD, a novel ensemble vision transformer-based scheme for CKD stage classification using ultrasound images. Three ViTs were trained on a private ultrasound image dataset using different training parameters. The performance of each ViT was evaluated using macro sensitivity, macro specificity, macro precision, macro F1-score, macro Youden index, the Matthews correlation coefficient (MCC), and macro balanced accuracy. The ensemble model achieved an overall classification accuracy of 86.36%. This work also emphasizes identifying and interpreting kidney regions affected by CKD progression. Explainable artificial intelligence techniques, including LIME, LRP, Attention-Min, and Attention-Max, were used to improve model transparency and clinical trust. An attention map combining the Attention-Min and Attention-Max results effectively identified and interpreted kidney regions affected during CKD progression from one stage to another. The attention map also highlighted the effects of CKD progression in these regions. Compared with existing methods, the proposed method classified the five CKD stages and normal kidney status with a 4% improvement in accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。