arXiv:2606.17504eess.IVcs.CV2026-06

分两阶段微调ResNet50,提升皮肤癌检测灵敏度。

Two-Stage Fine-Tuning of ResNet50 for High-Sensitivity Melanoma Detection on Dermoscopic Images

  • 先冻结主干只训练分类头,再整体低速微调避免遗忘
  • 在3826张图像上达到87.56%敏感度,AUC达0.9559
  • 适合医疗影像检测,尤其关注漏诊率的临床场景

黑色素瘤是危害最大的皮肤癌,早期发现可使五年生存率超过99%,但一旦扩散则急剧下降。本文提出并评估了一种针对ResNet50的两阶段微调方法,用于皮肤镜图像上的二分类黑色素瘤检测。核心挑战为类别不平衡及单阶段微调带来的迁移学习性能不足。在分层划分训练/验证/测试集后,仅对训练集进行随机过采样,实现1:1类别平衡。第一阶段仅训练分类头,冻结ResNet50主干;第二阶段以1e-5的学习率联合微调所有层,防止视觉特征灾难性遗忘。在独立测试集3,826张图像上,模型达到AUC-ROC 0.9559,准确率88.34%,敏感度87.56%,特异性89.13%,F1分数88.29%。消融实验表明,两阶段方案显著优于单阶段微调,敏感度提升超4%。Grad-CAM可视化显示病灶定位正确。配套提供可部署的Streamlit应用及全部训练代码。

原文摘要 · Abstract (English)

Melanoma is the most dangerous form of skin cancer with five-year survival rates exceeding 99% when detected early but falling sharply once the disease spreads. This paper proposes and evaluates a two-stage fine-tuning approach for ResNet50 applied to binary melanoma classification on dermoscopic images. The core challenges addressed are class imbalance and suboptimal transfer learning from single-stage fine-tuning. After stratified train/validation/test splitting, random oversampling was applied exclusively to the training set to achieve a 1:1 class balance. Stage 1 trained only the classification head with the ResNet50 base frozen, while Stage 2 fine-tuned all layers jointly at a low learning rate of 1e-5 to prevent catastrophic forgetting of learned visual features. On an independent test set of 3,826 images, the model achieved an AUC-ROC of 0.9559, accuracy of 88.34%, sensitivity of 87.56%, specificity of 89.13%, and F1-score of 88.29%. An ablation study confirms the two-stage protocol significantly outperforms single-stage fine-tuning, with sensitivity gains of over 4%. Grad-CAM visualizations demonstrate correct lesion localization. A fully deployable Streamlit detection application is provided alongside all training code.

皮肤癌检测两阶段微调医学图像ResNet50

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