用自适应融合提升ResNet-50,更准识别皮肤癌病变。
Skin Lesion Classification Based on ResNet-50 Enhanced With Adaptive Spatial Feature Fusion
- 引入自适应空间特征融合模块,动态整合多尺度特征。
- 在ISIC 2020上达93.18%准确率,AUC超0.967。
- 可视化更聚焦病灶区,适合医学图像诊断研究者。
皮肤癌分类因类别间相似性高、类内差异大及皮肤镜图像伪影而具挑战性。为此,本文提出增强版ResNet-50模型,结合自适应空间特征融合(ASFF)机制,自适应融合多尺度语义与表层特征以优化表示并减少过拟合。模型采用双分支结构,融合高层语义与中层细节特征,通过全局平均池化和全连接层生成空间权重,强化病灶相关区域。在随机选取的ISIC 2020平衡子集(3,297张图像)上,该模型超越多个CNN基线,实现93.182%准确率,且精度、召回率、特异性与F1均表现优异,同时达到0.9670 AUC(P-R)与0.9717 AUC(ROC)。Grad-CAM可视化显示模型对病灶区域关注更精准。该模型在ISIC 2019外部验证集上亦表现优于原ResNet-50。开源代码、权重及混淆矩阵见https://github.com/Grapesea/ASFF-ResNet50-enhanced。
原文摘要 · Abstract (English)
Skin cancer classification is challenging due to high inter-class similarity, intra-class variability, and artifacts in dermoscopic images. To address these issues, we propose an improved ResNet-50 with Adaptive Spatial Feature Fusion (ASFF), which adaptively integrates multi-scale semantic and surface features to refine representations and reduce overfitting. The ResNet-50 model is enhanced with an adaptive feature fusion mechanism to achieve more effective multi-scale feature extraction and improve overall performance. Specifically, a dual-branch design fuses high-level semantic and mid-level detail features which use global average pooling and fully connected layers to produce spatial weights, and emphasizes lesion-relevant regions. Evaluated on a balanced subset of ISIC 2020 (3,297 images, randomly selected from the original dataset), the ASFF-based ResNet-50 outperforms multiple CNN baselines, achieving 93.182% accuracy with superior precision, recall, specificity, and F1. It also reaches 0.9670 AUC (P-R) and 0.9717 AUC (ROC). Grad-CAM visualizations show more accurate focus on lesion areas.The proposed model also generalizes well to ISIC 2019 external validation, outperforming the ResNet-50 baseline. These findings demonstrate that the proposed approach provides a more effective and efficient solution for computer-aided skin cancer diagnosis. The generation codes, weights and confusion matrices are open sourced in https://github.com/Grapesea/ASFF-ResNet50-enhanced.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。