arXiv:2506.10302cs.CVcs.AI2025-06被引 2

提升皮肤癌诊断可信度,用多步方法融合特征与不确定性量化

A Quad-Step Approach to Uncertainty-Aware Deep Learning for Skin Cancer Classification

  • 分四步:预训练模型选型、降维优化、集成不确定性评估、融合模型加熵损失训练
  • 在HAM10000数据集上,特征融合+预测熵损失使准确率与不确定性可靠性双提升
  • 适合医疗AI研究者,尤其关注模型可信度与临床部署的团队

准确的皮肤癌诊断对早期治疗和患者预后至关重要。深度学习模型在自动化皮肤病变分类中展现出潜力,但受限于数据稀缺和不确定性感知不足。本研究在HAM10000数据集上,系统评估了基于迁移学习和不确定性量化(UQ)的深度学习皮肤癌分类方法。对比了CLIP系列、ResNet50、DenseNet121、VGG16和EfficientNet-V2-Large等预训练特征提取器,结合SVM、XGBoost和逻辑回归等传统分类器,并测试了多种主成分分析(PCA)设置(64, 128, 256, 512)。其中,LAION CLIP ViT-H/14和ViT-L/14在PCA-256下表现最优。在不确定性评估阶段,采用蒙特卡洛丢弃(MCD)、集成(Ensemble)和集成蒙特卡洛丢弃(EMCD)方法,通过不确定性感知指标(UAcc, USen, USpe, UPre)进行评估,发现集成方法在准确性与可靠性间取得最佳平衡。进一步通过融合表现最佳提取器的特征,在PCA-256下实现性能提升。最终提出一种基于特征融合并使用预测熵(PE)损失函数训练的模型,其在标准与不确定性感知评价中均优于所有先前配置,推动了可信赖的深度学习皮肤癌诊断发展。

原文摘要 · Abstract (English)

Accurate skin cancer diagnosis is vital for early treatment and improved patient outcomes. Deep learning (DL) models have shown promise in automating skin cancer classification, yet challenges remain due to data scarcity and limited uncertainty awareness. This study presents a comprehensive evaluation of DL-based skin lesion classification with transfer learning and uncertainty quantification (UQ) on the HAM10000 dataset. We benchmark several pre-trained feature extractors -- including CLIP variants, ResNet50, DenseNet121, VGG16, and EfficientNet-V2-Large -- combined with traditional classifiers such as SVM, XGBoost, and logistic regression. Multiple principal component analysis (PCA) settings (64, 128, 256, 512) are explored, with LAION CLIP ViT-H/14 and ViT-L/14 at PCA-256 achieving the strongest baseline results. In the UQ phase, Monte Carlo Dropout (MCD), Ensemble, and Ensemble Monte Carlo Dropout (EMCD) are applied and evaluated using uncertainty-aware metrics (UAcc, USen, USpe, UPre). Ensemble methods with PCA-256 provide the best balance between accuracy and reliability. Further improvements are obtained through feature fusion of top-performing extractors at PCA-256. Finally, we propose a feature-fusion based model trained with a predictive entropy (PE) loss function, which outperforms all prior configurations across both standard and uncertainty-aware evaluations, advancing trustworthy DL-based skin cancer diagnosis.

皮肤癌分类不确定性量化特征融合医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。