用轻量模型实现可解释的痤疮分级,准确率超93%
Interpretable Image-Level Acne Severity Grading via EfficientNet-B0 Transfer Learning and Grad-CAM

- 基于EfficientNet-B0迁移学习,结合Grad-CAM可视化提升可解释性
- 在2983张图像上达93.5%准确率,各类别F1值0.92~0.97
- 代码开源支持跨平台使用,适合临床与设备验证场景
寻常痤疮影响多数青少年和许多成年人。准确的严重程度分级有助于治疗、监测和临床试验终点设定,但人工评估依赖医师整体评价或Hayashi标准,受限于评分者间差异及成像条件不一致。本研究基于Hayashi标准构建四类痤疮严重程度分类器,采用ImageNet预训练的EfficientNet-B0模型进行迁移学习。模型在包含2,983张标注图像的公开ACNE04数据集上,通过AdamW优化、几何与光度增强,并基于验证集宏F1选择检查点进行微调。在保留的分层15%测试集上,分类器达到93.5%准确率和94.4%宏F1,各等级F1值介于0.92至0.97之间。83%的错误发生在相邻等级间。加权肯德尔一致性系数为0.956(95%置信区间:0.935–0.973),自助法置信区间显示性能稳定。最终卷积层的Grad-CAM可视化聚焦于额头、脸颊和下巴等临床相关区域。完整流程以Python(PyTorch和timm)和MATLAB R2026a的等效开源实现提供,包含面向临床的推理接口及无需专用预训练权重的备用主干选项。结果表明,轻量迁移学习可实现高精度、均衡且可解释的痤疮分级,为未来前瞻性及设备分层临床验证提供可复现的跨平台基准。
原文摘要 · Abstract (English)
Acne vulgaris affects most adolescents and many adults. Accurate severity grading guides treatment, monitoring, and clinical trial endpoints, but manual assessment using the Investigator's Global Assessment or Hayashi criteria is limited by inter-rater variability and inconsistent imaging conditions. We developed a four-class acne severity classifier based on the Hayashi criteria using transfer learning with an ImageNet-pretrained EfficientNet-B0 model. The model was fine-tuned on the public ACNE04 dataset of 2,983 labeled images using AdamW optimization, geometric and photometric augmentation, and checkpoint selection based on validation macro-F1. On a held-out stratified 15 percent test set, the classifier achieved 93.5 percent accuracy and 94.4 percent macro-F1, with per-class F1 scores from 0.92 to 0.97. Eighty-three percent of errors occurred between adjacent grades. Quadratic-weighted Cohen's kappa was 0.956, with a 95 percent confidence interval of 0.935 to 0.973. Bootstrap confidence intervals indicated stable performance. Grad-CAM visualizations from the final convolutional block focused on clinically relevant facial regions, including the forehead, cheeks, and chin. The complete pipeline is provided as functionally equivalent open-source implementations in Python using PyTorch and timm, and in MATLAB R2026a. The software includes a clinician-facing inference interface and a fallback backbone option that supports operation without specialized pretrained-weight packages. These results show that lightweight transfer learning can provide accurate, balanced, and interpretable acne severity grading while offering a reproducible cross-platform reference for future prospective and device-stratified clinical validation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。