用数学约束提升可解释性,让AI诊断皮肤癌更可信且适应少样本场景。
IViT: A Novel Interpretable Visual Transformer for Skin Disease Detection
- 通过二次规划约束特征选择,匹配临床诊断逻辑。
- 准确率93.80%,特征冗余降低29.5%,仅比基线低0.21%。
- 适合需要高可信度的医疗智能诊断场景,尤其少样本情况。
皮肤疾病临床诊断易受病灶类间相似性干扰,过度依赖医生经验易引发主观偏差。现有深度学习辅助诊断方法虽达较高准确率,但视觉变换器(ViT)存在黑箱问题,且在医疗少样本场景下适应性差。主流可解释算法在提升可解释性时往往导致显著准确率下降。本文提出一种受二次规划(QP)约束的可解释视觉变换器(IViT),引入预训练迁移学习以适应少样本特征提取。构建离散QP特征选择框架,筛选出与临床诊断逻辑一致的通用且判别性特征。设计多目标损失函数,在保持分类性能的同时减少特征冗余、优化激活分布。在六个标准皮肤疾病数据集上的实验表明,IViT达到93.80%准确率,仅比基线低0.21%,特征冗余降低29.5%。其核心激活区域与临床关注病灶区域高度一致。该模型在准确率与可解释性之间取得平衡,为少样本智能皮肤疾病诊断的临床部署提供可靠方案。
原文摘要 · Abstract (English)
The clinical diagnosis of skin diseases is susceptible to interference from inter-class similarity of skin lesions, and over-reliance on clinicians'experience easily leads to subjective bias. Although existing deep learning aided diagnosis methods achieve competitive accuracy, they suffer from the black-box opacity of Vision Transformer (ViT) and poor adaptability to medical few-shot scenarios. Moreover, mainstream explainable algorithms generally face the bottleneck of significant accuracy degradation when improving interpretability. This paper proposes an interpretable ViT (IViT) constrained by Quadratic Programming (QP). The introduced pre-trained transfer learning adapts to few-shot feature extraction. A discrete QP feature selection framework is constructed to screen generic and discriminative features consistent with clinical diagnostic logic. A multi-objective loss function is designed to reduce feature redundancy and optimize activation distribution while preserving classification performance. Experimental results on six standard skin disease datasets show that IViT achieves an accuracy of 93.80%, only 0.21% lower than the baseline, with feature redundancy reduced by 29.5%. Its core activation regions are consistent with clinically concerned lesion areas. The proposed model balances accuracy and interpretability, providing a reliable solution for the clinical deployment of few-shot intelligent skin disease diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。