[发明专利]一种基于深度学习的皮肤病图片病变类型分类方法

说明书

本发明公开了一种基于深度学习的皮肤病图片病变类型分类方法，本方法在现有ResNet50和InceptionV3网络的基础上设计了一种专门针对皮肤病变图片分类的模型算法，该模型在使用相关病变图片进行训练、调优之后能够对包括黑色素瘤、痣、基底细胞癌、光化性角化病、良性角化病、皮肤纤维瘤和血管瘤在内的七种皮肤病变进行分类诊断。使用该方法在与训练数据不同的测试数据图片库上进行测试，最终能够达到92.48％的准确率和86.76％的召回率，超过医生平均的准确率。

技术领域

本发明涉及深度学习医学图像分类算法，尤其是一种基于深度学习的医学图像分类方法。

背景技术

现有的基于深度学习的皮肤病分类诊断主要是诊断是否存在皮肤病变，即判断获取的图像是否存在有皮肤癌变。例如，公告号为CN107909566A的中国发明专利，公开了“一种基于深度学习的皮肤癌黑色素瘤的图像识别方法”，其方法通过深度学习技术来对是否存在黑色素瘤进行诊断。此种仅对于是否存在癌症的方法过于简单，皮肤是否存在病变通过普通人的肉眼即刻观测出，而且当皮肤发生病变且不为黑色素瘤时该方法未必能够正确诊断。因此大大加大了误诊的可能性，同时分类诊断的准确率不高且不能够准确诊断出病变属于哪种疾病。

发明内容

为了解决现有技术中存在误诊、准确率过低且不能诊断出处于哪种病变等缺陷，本发明提出了一种基于深度学习的皮肤病图片病变类型分类方法，将针对皮肤病图片重新设计深度学习模型，并与皮肤病变图像扩增技术相结合，在能够对包括黑色素瘤、痣、基底细胞癌、光化性角化病、良性角化病、皮肤纤维瘤和血管瘤在内的七种皮肤病变进行分类诊断。并且超过人类专家诊断的真确率。

本发明所采用的技术方案是：一种基于深度学习的皮肤病图片病变类型分类方法，改方法包含以下技术步骤：

步骤1、获取不同的皮肤病变图片；对皮肤病变图片进行数据预处理，对不同尺寸的皮肤病变图片进行缩放至设置好的两个固定尺寸。

步骤2、皮肤病变图片数据扩增，对经过预处理的皮肤病变图片进行包括随机裁剪、随机翻转、随机旋转和随机形变在内的变换，所有的变换均有概率发生，即可能发生变换也可能不发生变换。数据扩增的目的在于在训练阶段可以防止模型的过拟合。

步骤3、将步骤2扩增后的数据输入到卷积神经网络；所述的卷积神经网络为改进了ResNet50、InceptionV3网络，保留ResNet50平均池化层和InceptionV3平均池化层之前的部分，并对两个改进后的网络的输出张量在深度的维度上进行拼接，然后送入到分类器中；所述的分类器包含两个全连接层，第一个全连接层用于整合之前的张量特征。第二个全连接层用于产生分类结果。

步骤4、针对数据样本分布不平衡的问题，采用损失函数加权的方式解决，损失函数的加权值为每一类样本出现频率的倒数。

步骤5、使用超大规模视觉图像挑战赛中的图片所训练的权重，通过该权重对卷积网络进行初始化，使用自然图片的权重初始化在一定程度上能够辅助模型进行训练，以收敛到一个相对较好的结果。

步骤6、将步骤2中采样出的样本通过模型的输入层送入模型，利用反向传播算法对模型进行优化训练；

步骤7、通过反向传播算法对模型权重进行训练。

步骤8、将模型训练后的权重保存，使用该权重对皮肤病变图片进行分类诊断。

本发明的有益效果是：

1)本发明提出一种自动化皮肤病变检测分类算法，可以辅助医生或者自主检测皮肤病变属于那种皮肤疾病。

2)本发明提出的基于深度学习的皮肤病变图片检测分类算法在划分出来的测试集上测试，测试结果显示算法诊断准确率超过人类医生的诊断准确率。