[论文解读] Deep-CLASS at ISIC Machine Learning Challenge 2018
该论文提出Deep-CLASS,一种19层卷积神经网络(CNN)框架,用于在ISIC 2018挑战赛中从皮肤镜图像对七种皮肤病变类型进行分类。通过在Caffe中使用数据增强和微调的CNN,并结合MATLAB,该模型在Dermatofibroma和Vascular lesions上的准确率最高达到98%,Melanoma的准确率为82%,展示了在多类皮肤疾病分类中的优异性能。
This paper reports the method and evaluation results of MedAusbild team for ISIC challenge task. Since early 2017, our team has worked on melanoma classification [1][6], and has employed deep learning since beginning of 2018 [7]. Deep learning helps researchers absolutely to treat and detect diseases by analyzing medical data (e.g., medical images). One of the representative models among the various deep-learning models is a convolutional neural network (CNN). Although our team has an experience with segmentation and classification of benign and malignant skin-lesions, we have participated in the task 3 of ISIC Challenge 2018 for classification of seven skin diseases, explained in this paper.
研究动机与目标
- 开发一种基于深度学习的系统,以实现对七种皮肤病变类型的准确分类。
- 通过在医学影像数据上进行端到端深度学习,提升黑色素瘤及其他皮肤癌的检测能力。
- 通过数据增强和模型优化,解决皮肤病变分类中数据不平衡的挑战。
- 在ISIC 2018基准数据集上评估自定义CNN框架在疾病分类中的性能。
提出的方法
- 设计了一种19层深度CNN架构,包含11个卷积层、5个最大池化层和3个全连接层。
- 输入图像被缩放为128×128×3,并通过旋转和翻转进行数据增强,以提升训练多样性。
- 使用Caffe深度学习框架实现模型,并通过微调超参数以获得最佳性能。
- 使用MATLAB进行系统开发,包括图像处理、并行计算以及通过App Designer进行应用部署。
- 应用类别权重和损失优化以处理数据集中类别不平衡的问题。
- 在HAM10000数据集的保留测试集(161张图像)上进行评估。
实验结果
研究问题
- RQ1深度CNN模型能否在从皮肤镜图像中分类七种不同皮肤病变类型方面实现高准确率?
- RQ2数据增强如何影响模型在数据不平衡的皮肤病变数据集上的泛化能力和性能?
- RQ3微调后的CNN模型在ISIC 2018任务3多类皮肤疾病分类基准上的表现如何?
- RQ4在真实世界临床数据集中,不同皮肤病变类型的精确率、召回率、特异性及F1值如何变化?
主要发现
- 该模型在Dermatofibroma(D)上达到98%的准确率,在Vascular lesions(VL)上也达到98%,表明对罕见类别的分类性能出色。
- Melanoma(M)的分类准确率为82%,但敏感度(召回率)仅为47%,精确率为30%,凸显了对恶性病例检测的挑战。
- Melanocytic nevus(N)的F1值最高(78%),精确率也达90%,表明模型对良性病变具有高度置信度和优异性能。
- 混淆矩阵显示,Melanoma常被误分类为Melanocytic nevus(N),提示两类之间存在类别相似性问题。
- Basal cell carcinoma(BCC)的准确率为92%,敏感度为52%,特异性为94%,表明其检测能力中等。
- 该模型在良性病变和血管性病变上表现出强鲁棒性,但在Melanoma和AK上表现欠佳,可能由于类别不平衡及视觉相似性所致。
更好的研究,从现在开始
从阅读论文到最终审阅,大幅缩短您的研究时间。
无需绑定信用卡
本解读由 AI 生成,并经人工编辑审核。