用于照片中皮肤病变自动诊断的深度迁移学习
计算机视觉与模式识别
2020-11-19 v3 机器学习
图像与视频处理
摘要
黑色素瘤并非最常见的皮肤癌形式,却是最致命的。目前该疾病由专家皮肤科医生诊断,费用高昂且需要及时获得医疗服务。深度学习的近期进展有潜力改善诊断表现、加快紧急转诊并减轻临床医生负担。通过智能手机,该技术可触达通常无法获得此类医疗服务的人群,例如世界偏远地区因经济限制者,或 2020 年因 COVID-19 取消就诊者。为此,我们研究了多种迁移学习方法,利用在 ImageNet 上预训练的模型参数并在黑色素瘤检测上微调。我们比较了 EfficientNet、MnasNet、MobileNet、DenseNet、SqueezeNet、ShuffleNet、GoogleNet、ResNet、ResNeXt、VGG 以及一个简单 CNN 在有无迁移学习下的表现。我们发现移动网络 EfficientNet(含迁移学习)取得最佳平均表现,受试者工作特征曲线下面积(AUROC)为 0.9310.005,精确率-召回率曲线下面积(AUPRC)为 0.8400.010。这显著优于全科医生(0.830.03 AUROC)与皮肤科医生(0.910.02 AUROC)。
引用
@article{arxiv.2011.04475,
title = {Deep Transfer Learning for Automated Diagnosis of Skin Lesions from Photographs},
author = {Emma Rocheteau and Doyoon Kim},
journal= {arXiv preprint arXiv:2011.04475},
year = {2020}
}
备注
Machine Learning for Mobile Health (ML4MH) Workshop at NeurIPS 2020