一种基于迁移学习的深度神经网络脑肿瘤图像分类方法技术

一种基于迁移学习的深度神经网络脑肿瘤图像分类方法，包括4个步骤：S1数据预处理：从原始图像数据集生成两种数据集；S2构建预训练模型：预训练模型中应用的是ImageNet数据集，并将收敛后的预训练模型保存下来；S3构建改进的CNN架构模型：选定带有预训练的InceptionV3的迁移学习的深度卷积神经网络，微调网络，使整个模型进一步地适应目标任务，最终提升分类效果；S4脑肿瘤图像分类评估：分别验证已裁剪和未裁剪的数据集，通过精确度、召回率等，评价分类性能。本发明专利技术提供可以通过脑肿瘤MR图像数据集对患不同脑肿瘤的病人进行分类，利用分类的脑肿瘤MR图像辅助医生诊断，帮助病人提前发现病情和提高医生的诊断效率。现病情和提高医生的诊断效率。现病情和提高医生的诊断效率。

【技术实现步骤摘要】
一种基于迁移学习的深度神经网络脑肿瘤图像分类方法


[0001]专利技术涉及深度学习和医学相关领域，特别涉及一种基于迁移学习的深度神经网络脑肿瘤图像分类方法。

技术介绍

[0002]根据CA(A Cancer Journal for Clinicians)杂志发布的全球癌症数据，2020年全球新发癌症病例1929万例。癌症的产生可以通过早期检测来预防，但脑癌的临床诊断非常困难,与癌症不同，肿瘤可以是良性、癌前或恶性,良性肿瘤可以手术切除，不影响其他器官和组织。然而，脑肿瘤包括脑膜瘤、胶质瘤和垂体。脑膜瘤肿瘤发生于膜(保护大脑和脊髓的区域)，胶质瘤发生于脑组织，垂体瘤是位于脑下垂体区域的颅骨内的肿块。脑肿瘤由于前期患者并未感觉到自己已经患病、未能动态观察和分析病情，这就耽误了治疗的最佳时间。
[0003]随着计算机人工智能和图像处理的发展，基于深度学习和神经网络的高级机器学习模型被创造以在各种医疗诊断中辅助医生诊断，这些模型为许多医学成像提供了有用的帮助，几种用于分类图像和检测MR图像中部分区域。但由于网络上公开的大多数据集数量都非常少，且训练得出的模型都是在小数据集中得到的，而基于深度学习和神经网络的高级机器学习模型需要大量的数据才能实现最佳性能。

技术实现思路

[0004]本专利技术目的在于提供一种基于迁移学习的深度神经网络脑肿瘤图像分类方法，对三种脑肿瘤数据进行训练；通过准确性、损失、精确度、召回率等，对不同种类的脑肿瘤图像进行分类。
[0005]为了实现以上目的，本专利技术所采用的技术方案包括以下步骤：
[0006]步骤S1.数据预处理：从原始图像数据集生成两种数据集；并将两种数据集都归一化调整为256x256像素。
[0007]步骤S2.构建预训练模型：预训练模型中应用的是ImageNet数据集，并将收敛后的预训练模型保存下来。
[0008]步骤S3.构建改进的CNN架构模型：选定带有预训练的InceptionV3的迁移学习的深度卷积神经网络(DCNN)，通过脑肿瘤数据集来微调网络，使整个模型进一步地适应目标任务，最终提升分类效果。
[0009]步骤S4.脑肿瘤图像分类评估：分别验证已裁剪和未裁剪的数据集，通过准确性、损失、精确度、召回率等，评价分类性能。
[0010]所述步骤S1数据预处理如下：
[0011]步骤S1：从原始图像数据集生成两种数据集，其中一个围绕大脑视图做剪裁处理，另一个不进行剪裁；
[0012]步骤S12：将两种数据集都归一化调整为256x256像素。
[0013]所述步骤S2构建预训练模型步骤如下：
[0014]步骤S21：使用ImageNet数据集，将其放入深度卷积神经网络模型。
[0015]步骤S22：设置网络参数。
[0016]步骤S23：训练InceptionV3网络模型。
[0017]步骤S24：保存预训练模型。
[0018]所述步骤S3构建改进的CNN架构模型包括以下步骤：
[0019]步骤S31：准备脑肿瘤MR图像数据集，将其作为目标数据集。
[0020]步骤S32：将脑肿瘤MR图像数据集输入预训练的InceptionV3网络模型(调用预训练模型)。
[0021]步骤S33：设置预训练参数，对InceptionV3预训练得出的权重进行微调，使提取到的分类特征更贴近于脑肿瘤数据集。
[0022]步骤S34：训练改进的CNN架构模型，根据MR图像数据集中的类别，InceptionV3最后一层替换为具有3个隐藏单元的层。
[0023]步骤S35：保存模型。
[0024]所述步骤S4脑肿瘤图像分类评估步骤如下：
[0025]步骤S41：将测试数据导入训练好的网络模型，进行测试，生成模型分类准确性、损失、精确度、召回率等。分类准确性：分类的主要性能评估指标是准确性。其公式(1)如下：
[0026][0027]其中，TP表示真阳性，TN表示真阴性，FP表示假阳性，FN表示假阴性。
[0028]精确度：特定于类的性能度量需要进行验证，精确度是此类指标之一。其公式(2)如下：
[0029][0030]召回率：这是另一个基本度量，定义为模型正确预测的类中观察点的分数。其公式(3)如下：
[0031][0032]损失函数：实验中,通过构建经全连接层输出的分类概率与输入图像的真实标签之间的交叉熵函数作为网络的目标函数。其公式(4)如下：
[0033][0034]其中p
k
(X)是输入细胞衍射图像X后经网络得到的分成k类的概率分布。
[0035]步骤S42：根据准确性、精确度、损失和召回率判断基于迁移学习的深度卷积神经网络在脑肿瘤分类中是最佳的。
[0036]本专利技术具有以下有益效果及优点：
[0037](1)相较于传统的图像分类，本专利技术基于迁移学习的深度卷积神经网络对脑肿瘤图像进行分类，采用预训练的InceptionV3网络模型，并在调用其训练脑肿瘤数据集时对参数微调，保证提取更好的图像特征；
[0038](2)本专利技术使用剪裁和未剪裁的数据集，最终表示该网络结构模型不需要对脑肿
瘤图像进行分类分割。本专利技术的网络还具有非常好的执行速度，每个epoch 15s，提高了图像分类的效率。
附图说明
[0039]图1为本专利技术基于迁移学习的深度神经网络脑肿瘤图像分类方法步骤图。
[0040]图2为本专利技术迁移学习的深度神经网络脑肿瘤图像分类方法流程图。
具体实施方式
[0041]实施例：
[0042]如图1所示，本专利技术技术方案包含四个方面的步骤：数据预处理、构建预训练模型、构建改进的CNN架构模型、脑肿瘤图像分类评估。
[0043]所述步骤S1.数据预处理：从原始图像数据集生成两种数据集；并将两种数据集都归一化调整为256x256像素；
[0044]所述步骤S2.构建预训练模型：预训练模型中应用的是ImageNet数据集，并将收敛后的预训练模型保存下来；
[0045]所述步骤S3.构建改进的CNN架构模型：选定带有预训练的InceptionV3的迁移学习的深度卷积神经网络(DCNN)，通过脑肿瘤数据集来微调网络，使整个模型进一步地适应目标任务，最终提升分类效果；
[0046]所述步骤S4.脑肿瘤图像分类评估：分别验证已裁剪和未裁剪的数据集，通过准确性、损失、精确度、召回率等，评价分类性能；
[0047]如图2所示，本专利技术迁移学习的深度神经网络脑肿瘤图像分类的方法的工作过程主要分为以下几步：
[0048]首先从原始图像数据集生成两种数据集，其中一个围绕大脑视图做剪裁处理，另一个不进行剪裁。将两种数据集都归一化调整为256x256像素，发现256x256像素和512x512像素的分类准确率没有太大区别，且前者在内存和处理时间方面优于后者，所以选择256x256像素的数据集。
...

【技术保护点】

【技术特征摘要】
1.一种基于迁移学习的深度神经网络脑肿瘤图像分类方法，其特征在于，包括以下步骤：步骤S1：从原始图像数据集生成两种数据集，并将两种数据集都归一化调整为256x 256像素；步骤S2：预训练模型中应用的是ImageNet数据集，并将收敛后的预训练模型保存下来；步骤S3：选定带有预训练的InceptionV3的迁移学习的深度卷积神经网络(DCNN)，通过脑肿瘤数据集来微调网络，使整个模型进一步地适应目标任务，最终提升分类效果；步骤S4：分别验证已裁剪和未裁剪的数据集，通过准确性、损失、精确度、召回率等，评价分类性能。2.根据权利要求1所述的一种基于迁移学习的深度神经网络脑肿瘤图像分类方法，其特征在于，所述步骤S1中的预处理方法包括以下步骤：S11：从原始图像数据集生成两种数据集，其中一个围绕大脑视图做剪裁处理，另一个不进行剪裁；S12：将两种数据集都归一化调整为256x 256像素。3.根据权利要求1所述的一种基于迁移学习的深度神经网络脑肿瘤图像分类方法，其特征在于，所述步骤S2中...

【专利技术属性】
技术研发人员：邓酩，侯立宪，柳庆龙，谢才龙，
申请(专利权)人：桂林理工大学，
类型：发明
国别省市：