用可解释的神经网络提升皮肤癌诊断准确率与透明度。
Hybrid Interpretable Deep Learning Framework for Skin Cancer Diagnosis: Integrating Radial Basis Function Networks with Explainable AI
- 结合CNN与径向基函数网络,通过局部特征映射提升可解释性。
- 在ISIC 2016和2017数据集上分别达到83.02%和72.15%准确率。
- 适合临床医生使用,让AI决策过程可追溯、可信。
皮肤癌是全球最常见且可能致命的疾病之一,早期准确诊断对改善患者预后至关重要。传统诊断方法依赖临床经验和组织病理分析,常耗时长、主观性强且结果变异大。为此,本文提出一种新型混合深度学习框架,将卷积神经网络(CNN)与径向基函数(RBF)网络结合,实现高分类准确率与增强的可解释性。引入RBF网络因其内在可解释性和对输入特征的局部响应能力,适用于需要透明决策的任务。不同于依赖全局特征表示的传统模型,RBF网络可将图像片段映射到选定原型,捕捉单幅图像中的显著特征,使临床医生能追踪预测至具体可解释模式。框架采用基于分割的特征提取、主动学习选择原型及K-Medoids聚类,聚焦显著特征。在ISIC 2016和ISIC 2017数据集上的评估表明,该模型使用ResNet50分别达到83.02%和72.15%的分类准确率,优于基于VGG16的配置。通过生成预测的可解释解释,框架契合临床工作流程,弥合预测性能与可信度之间的鸿沟。本研究凸显了混合模型在提供可行动洞察方面的潜力,推动高风险医疗应用中可靠AI辅助诊断工具的发展。
原文摘要 · Abstract (English)
Skin cancer is one of the most prevalent and potentially life-threatening diseases worldwide, necessitating early and accurate diagnosis to improve patient outcomes. Conventional diagnostic methods, reliant on clinical expertise and histopathological analysis, are often time-intensive, subjective, and prone to variability. To address these limitations, we propose a novel hybrid deep learning framework that integrates convolutional neural networks (CNNs) with Radial Basis Function (RBF) Networks to achieve high classification accuracy and enhanced interpretability. The motivation for incorporating RBF Networks lies in their intrinsic interpretability and localized response to input features, which make them well-suited for tasks requiring transparency and fine-grained decision-making. Unlike traditional deep learning models that rely on global feature representations, RBF Networks allow for mapping segments of images to chosen prototypes, exploiting salient features within a single image. This enables clinicians to trace predictions to specific, interpretable patterns. The framework incorporates segmentation-based feature extraction, active learning for prototype selection, and K-Medoids clustering to focus on these salient features. Evaluations on the ISIC 2016 and ISIC 2017 datasets demonstrate the model's effectiveness, achieving classification accuracies of 83.02\% and 72.15\% using ResNet50, respectively, and outperforming VGG16-based configurations. By generating interpretable explanations for predictions, the framework aligns with clinical workflows, bridging the gap between predictive performance and trustworthiness. This study highlights the potential of hybrid models to deliver actionable insights, advancing the development of reliable AI-assisted diagnostic tools for high-stakes medical applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。