一种乳腺癌图像深度学习识别模型训练方法及系统

1.本发明涉及乳腺癌图像处理技术领域，尤其涉及一种乳腺癌图像深度学习识别模型训练方法及系统。


背景技术：

2.乳腺癌是乳腺上皮细胞在多种致癌因子的作用下，发生增殖失控的现象。疾病早期常表现为乳房肿块、乳头溢液、腋窝淋巴结肿大等症状，晚期可因癌细胞发生远处转移，出现多器官病变，直接威胁患者的生命，乳腺mri检查因其无创、信息量丰富和软组织分辨率高的优势，成为乳腺癌影像诊断及评价的重要检查，临床上，通过对乳腺mri图像的计算机处理和分析来辅助医师对乳腺病变进行良恶性分类，以达到提高诊断准确率和为制定个体化诊疗方案提供重要参考依据的目的。
3.其中深度学习技术可以通过构建的深层次网络直接从网络中得到数据的特征,实现了特征的自适应提取,并广泛地应用于医学图像的疾病诊断中。卷积神经网络由输入层、卷积层、池化层、全连接层和输出层构成，并广泛地应用于计算机视觉领域(包括mri图像)，在卷积层中，会定义一个卷积核，将卷积核认为是一种“感受野”，在卷积运算时，这个“感受野”会按照从左到右、从上到下的顺序在输入图像上进行滑动卷积运算，通过这样的操作，就可以得到输入图像的特征图像。
4.通过迁移学习方法，在已训练的神经网络模型vgg16中，进行医学图像的特征提取，结合模型的学习大量数据分析乳腺肿瘤的mri图像信息，对乳腺疾病的影像诊断及评价，尤其是提高诊断准确率和为制定个体化诊疗方案提供重要参考依据具有积极意义。因此，我们提出一种乳腺癌图像深度学习识别模型训练方法及系统。


技术实现要素：

5.本发明的目的是为了解决现有技术中存在的缺点，而提出的一种乳腺癌图像深度学习识别模型训练方法及系统。
6.为达到以上目的，本发明采用的技术方案为：一种乳腺癌图像深度学习识别模型训练方法，包括如下步骤：
7.s1：获取乳腺癌mri图像；
8.s2：进行乳腺癌mri图像的特征提取，选取第6卷积层、第9卷积层和第12卷积层的特征，进一步的选择第6卷积层、第9卷积层和第12卷积层的各层的下一层进行最大池化后的特征进行分类；
9.s3：选择svm分类器，将图像数据集按照7∶3的比例分为训练集和测试集，再分别将第6卷积层、第9卷积层和第12卷积层通过vgg16网络模型提取的乳腺癌mri特征图像输入至svm分类器中进行分类，得出每一层的分类准确率；
10.s4：以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练；
11.s5：选择预训练后的乳腺癌mri图像作为第二训练数据，对预训练后的vgg16网络模型进行训练，得到用于乳腺癌mri图像识别的深度学习模型。
12.优选的，在s1中，乳腺癌mri图像首先进行归一化操作，使乳腺癌mri图像大小统一为224像素
×
224像素。
13.优选的，在s4中，最大池化是在每次对数据操作时，将池化覆盖区域的数据取最大值，表示为：
[0014][0015]
其中，为上一层的特征图，h为数据的行数，w为数据的列数。
[0016]
优选的，在s5中，其中准确率公式如下：
[0017][0018]
通过上述的准确率公式得出不同层的分类准确率。
[0019]
优选的，在s2中，采用med3d模型训练框架，以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练，得到模型参数。
[0020]
优选的，采用monai模型训练框架，选择预训练后的模型参数作为第二训练数据，对vgg16网络模型进行训练。
[0021]
优选的，将预训练后的训练数据作为初始模型参数，对预训练后的vgg16网络模型进行训练。
[0022]
一种乳腺癌图像深度学习识别模型训练系统，包括有获取单元、预处理单元、预训练单元以及训练单元；
[0023]
所述获取单元，用以乳腺癌mri图像的采集获取；
[0024]
所述预处理单元，用以进行乳腺癌mri图像的的预处理，包括特征提取、特征降维以及利用svm分类器进行分类；
[0025]
所述预训练单元，用以获取乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据进行预训练；
[0026]
所述训练单元，通过vgg16网络模型以预训练后的乳腺癌mri图像作为第二训练数据进行训练，并得到用于乳腺癌图像识别的深度学习模型。
[0027]
与现有技术相比，本发明具有以下有益效果：
[0028]
通过深度学习识别模型训练，对乳腺肿瘤图像的识别、训练，实现了乳腺肿瘤图像的病变识别判断，且摒弃了传统的人工筛查方法的低效率方式，对乳腺癌治疗过程中制定可靠方案具有积极意义。
附图说明
[0029]
图1为本发明的流程示意图；
[0030]
图2为本发明的模型训练方法的流程框图。
具体实施方式
[0031]
以下描述用于揭露本发明以使本领域技术人员能够实现本发明。以下描述中的优选实施例只作为举例，本领域技术人员可以想到其他显而易见的变型。
[0032]
实施例1
[0033]
如附图1-2所示的一种乳腺癌图像深度学习识别模型训练方法，包括如下步骤：
[0034]
s1：获取乳腺癌mri图像；
[0035]
s2：进行乳腺癌mri图像的特征提取，选取第6卷积层、第9卷积层和第12卷积层的特征，进一步的选择第6卷积层、第9卷积层和第12卷积层的各层的下一层进行最大池化后的特征进行分类；
[0036]
s3：选择svm分类器，将图像数据集按照7∶3的比例分为训练集和测试集，再分别将第6卷积层、第9卷积层和第12卷积层通过vgg16网络模型提取的乳腺癌mri特征图像输入至svm分类器中进行分类，得出每一层的分类准确率；
[0037]
s4：以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练；
[0038]
s5：选择预训练后的乳腺癌mri图像作为第二训练数据，对预训练后的vgg16网络模型进行训练，得到用于乳腺癌mri图像识别的深度学习模型。
[0039]
一种乳腺癌图像深度学习识别模型训练系统，包括有获取单元、预处理单元、预训练单元以及训练单元；
[0040]
所述获取单元，用以乳腺癌mri图像的采集获取；
[0041]
所述预处理单元，用以进行乳腺癌mri图像的的预处理，包括特征提取、特征降维以及利用svm分类器进行分类；
[0042]
所述预训练单元，用以获取乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据进行预训练；
[0043]
所述训练单元，通过vgg16网络模型以预训练后的乳腺癌mri图像作为第二训练数据进行训练，并得到用于乳腺癌图像识别的深度学习模型。
[0044]
工作时，乳腺癌mri图像首先进行归一化操作，使乳腺癌mri图像大小统一为224像素
×
224像素，进行乳腺癌mri图像的特征提取，选取第6卷积层、第9卷积层和第12卷积层的特征，进一步的选择第6卷积层、第9卷积层和第12卷积层的各层的下一层进行最大池化后的特征进行分类；选择svm分类器，将图像数据集按照7∶3的比例分为训练集和测试集，再分别将第6卷积层、第9卷积层和第12卷积层通过vgg16网络模型提取的乳腺癌mri特征图像输入至svm分类器中进行分类，得出每一层的分类准确率，入至svm分类器中进行分类，得出每一层的分类准确率，的计算公式，可得出不同层的分类准确率；随即以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练；选择预训练后的乳腺癌mri图像作为第二训练数据，对预训练后的vgg16网络模型进行训练，得到用于乳腺癌mri图像识别的深度学习模型，其中，采用med3d模型训练框架，以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练，得到模型参数；采用monai模型训练框架，选择预训练后的模型参数作为第二训练数据，对vgg16网络模型进行训练，将预训练后的训练数据作为初始模
型参数，对预训练后的vgg16网络模型进行训练，最终得到深度学习识别模型。
[0045]
在s1中，乳腺癌mri图像首先进行归一化操作，使乳腺癌mri图像大小统一为224像素
×
224像素，在s4中，最大池化是在每次对数据操作时，将池化覆盖区域的数据取最大值，表示为：
[0046][0047]
其中，为上一层的特征图，h为数据的行数，w为数据的列数。
[0048]
在s5中，其中准确率公式如下：
[0049][0050]
通过上述的准确率公式得出不同层的分类准确率。
[0051]
在s2中，采用med3d模型训练框架，以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练，得到模型参数。
[0052]
采用monai模型训练框架，选择预训练后的模型参数作为第二训练数据，对vgg16网络模型进行训练，将预训练后的训练数据作为初始模型参数，对预训练后的vgg16网络模型进行训练。
[0053]
实施例2
[0054]
通过上述的实施例1，结合实验进行结果验证，首先于mri图像数据库内获取相关乳腺肿瘤图像，并通过归一化设置为224*224像素的大小，同时配置vgg16网络参数，硬件采用第八代i7处理器、32gb内存，geforce1080ti显卡以及tensorflow1.4深度学习开源模型；
[0055]
本发明的使用流程及工作原理：
[0056]
参照说明书附图1-2所示，在进行乳腺癌图像的识别模型训练时，首先，获取乳腺癌mri图像；进行乳腺癌mri图像的特征提取，选取第6卷积层、第9卷积层和第12卷积层的特征，进一步的选择第6卷积层、第9卷积层和第12卷积层的各层的下一层进行最大池化后的特征进行分类；选择svm分类器，将图像数据集按照7∶3的比例分为训练集和测试集，再分别将第6卷积层、第9卷积层和第12卷积层通过vgg16网络模型提取的乳腺癌mri特征图像输入至svm分类器中进行分类，得出每一层的分类准确率，至svm分类器中进行分类，得出每一层的分类准确率，的计算公式，可得出不同层的分类准确率；
[0057]
随即以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练；选择预训练后的乳腺癌mri图像作为第二训练数据，对预训练后的vgg16网络模型进行训练，得到用于乳腺癌mri图像识别的深度学习模型，其中，采用med3d模型训练框架，以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练，得到模型参数；
[0058]
采用monai模型训练框架，选择预训练后的模型参数作为第二训练数据，对vgg16网络模型进行训练，将预训练后的训练数据作为初始模型参数，对预训练后的vgg16网络模
型进行训练，最终得到深度学习识别模型，此种模型训练方法能够有效的进行乳腺肿瘤图像的识别、训练，且能够结合乳腺肿瘤图像的病变进行识别判断，有利于在获取病人乳腺癌mri图像后的进行后续诊断或制定较佳的治疗方案，进一步的有利于乳腺癌的治疗；
[0059]
综上：通过深度学习识别模型训练，对乳腺肿瘤图像的识别、训练，实现了乳腺肿瘤图像的病变识别判断，且摒弃了传统的人工筛查方法的低效率方式，对乳腺癌治疗过程中制定可靠方案具有积极意义。
[0060]
以上显示和描述了本发明的基本原理、主要特征和本发明的优点。本行业的技术人员应该了解，本发明不受上述实施例的限制，上述实施例和说明书中描述的只是本发明的原理，在不脱离本发明精神和范围的前提下本发明还会有各种变化和改进，这些变化和改进都落入要求保护的本发明的范围内。本发明要求的保护范围由所附的权利要求书及其等同物界定。

技术特征：
1.一种乳腺癌图像深度学习识别模型训练方法，其特征在于：包括如下步骤：s1：获取乳腺癌mri图像；s2：进行乳腺癌mri图像的特征提取，选取第6卷积层、第9卷积层和第12卷积层的特征，进一步的选择第6卷积层、第9卷积层和第12卷积层的各层的下一层进行最大池化后的特征进行分类；s3：选择svm分类器，将图像数据集按照7∶3的比例分为训练集和测试集，再分别将第6卷积层、第9卷积层和第12卷积层通过vgg16网络模型提取的乳腺癌mri特征图像输入至svm分类器中进行分类，得出每一层的分类准确率；s4：以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练；s5：选择预训练后的乳腺癌mri图像作为第二训练数据，对预训练后的vgg16网络模型进行训练，得到用于乳腺癌mri图像识别的深度学习模型。2.根据权利要求1所述的一种乳腺癌图像深度学习识别模型训练方法，其特征在于，在s1中，乳腺癌mri图像首先进行归一化操作，使乳腺癌mri图像大小统一为224像素
×
224像素。3.根据权利要求1所述的一种乳腺癌图像深度学习识别模型训练方法，其特征在于，在s4中，最大池化是在每次对数据操作时，将池化覆盖区域的数据取最大值，表示为：其中，为上一层的特征图，h为数据的行数，w为数据的列数。4.根据权利要求1所述的一种乳腺癌图像深度学习识别模型训练方法，其特征在于，在s5中，其中准确率公式如下：通过上述的准确率公式得出不同层的分类准确率。5.根据权利要求1所述的一种乳腺癌图像深度学习识别模型训练方法，其特征在于，在s2中，采用med3d模型训练框架，以乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的vgg16网络模型进行乳腺癌mri图像进行预训练，得到模型参数。6.根据权利要求5所述的一种乳腺癌图像深度学习识别模型训练方法，其特征在于，采用monai模型训练框架，选择预训练后的模型参数作为第二训练数据，对vgg16网络模型进行训练。7.根据权利要求6所述的一种乳腺癌图像深度学习识别模型训练方法，其特征在于，将预训练后的训练数据作为初始模型参数，对预训练后的vgg16网络模型进行训练。8.一种乳腺癌图像深度学习识别模型训练系统，其特征在于，包括有获取单元、预处理单元、预训练单元以及训练单元；所述获取单元，用以乳腺癌mri图像的采集获取；所述预处理单元，用以进行乳腺癌mri图像的的预处理，包括特征提取、特征降维以及利用svm分类器进行分类；
所述预训练单元，用以获取乳腺癌mri图像数据库中的乳腺癌图像作为第一训练数据进行预训练；所述训练单元，通过vgg16网络模型以预训练后的乳腺癌mri图像作为第二训练数据进行训练，并得到用于乳腺癌图像识别的深度学习模型。

技术总结
本发明涉及乳腺癌图像处理技术领域，尤其是一种乳腺癌图像深度学习识别模型训练方法及系统，包括获取乳腺癌MRI图像；进行乳腺癌MRI图像的特征提取；通过VGG16网络模型提取的乳腺癌MRI特征图像输入至SVM分类器中进行分类；以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练；选择预训练后的乳腺癌MRI图像作为第二训练数据，对预训练后的VGG16网络模型进行训练，得到用于乳腺癌MRI图像识别的深度学习模型。通过深度学习识别模型训练，对乳腺肿瘤图像的识别、训练，实现了乳腺肿瘤图像的病变识别判断，且摒弃了传统的人工筛查方法的低效率方式，对乳腺疾病的影像诊断及评价具有积极意义。影像诊断及评价具有积极意义。影像诊断及评价具有积极意义。


技术研发人员：王金花 刘于宝
受保护的技术使用者：南方医科大学深圳医院
技术研发日：2023.06.28
技术公布日：2023/9/22