CN116797838A - 一种乳腺癌图像深度学习识别模型训练方法及系统 - Google Patents
一种乳腺癌图像深度学习识别模型训练方法及系统 Download PDFInfo
- Publication number
- CN116797838A CN116797838A CN202310772916.8A CN202310772916A CN116797838A CN 116797838 A CN116797838 A CN 116797838A CN 202310772916 A CN202310772916 A CN 202310772916A CN 116797838 A CN116797838 A CN 116797838A
- Authority
- CN
- China
- Prior art keywords
- breast cancer
- training
- image
- deep learning
- model
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 208000026310 Breast neoplasm Diseases 0.000 title claims abstract description 123
- 206010006187 Breast cancer Diseases 0.000 title claims abstract description 122
- 238000012549 training Methods 0.000 title claims abstract description 101
- 238000013135 deep learning Methods 0.000 title claims abstract description 24
- 238000000034 method Methods 0.000 title claims abstract description 23
- 238000013136 deep learning model Methods 0.000 claims abstract description 9
- 238000011176 pooling Methods 0.000 claims description 12
- 238000007781 pre-processing Methods 0.000 claims description 6
- 238000000605 extraction Methods 0.000 claims description 5
- 238000012360 testing method Methods 0.000 claims description 5
- 230000009467 reduction Effects 0.000 claims description 3
- 238000003745 diagnosis Methods 0.000 abstract description 9
- 230000003902 lesion Effects 0.000 abstract description 6
- 238000011156 evaluation Methods 0.000 abstract description 3
- 238000012545 processing Methods 0.000 abstract description 3
- 238000012216 screening Methods 0.000 abstract description 3
- 208000030270 breast disease Diseases 0.000 abstract description 2
- 230000008901 benefit Effects 0.000 description 3
- 210000000481 breast Anatomy 0.000 description 3
- 230000009286 beneficial effect Effects 0.000 description 2
- 238000004364 calculation method Methods 0.000 description 2
- 201000010099 disease Diseases 0.000 description 2
- 208000037265 diseases, disorders, signs and symptoms Diseases 0.000 description 2
- 238000005516 engineering process Methods 0.000 description 2
- 238000003384 imaging method Methods 0.000 description 2
- 238000012986 modification Methods 0.000 description 2
- 230000004048 modification Effects 0.000 description 2
- 206010006223 Breast discharge Diseases 0.000 description 1
- 206010006272 Breast mass Diseases 0.000 description 1
- 206010028980 Neoplasm Diseases 0.000 description 1
- 230000009471 action Effects 0.000 description 1
- 230000003044 adaptive effect Effects 0.000 description 1
- 238000004458 analytical method Methods 0.000 description 1
- 210000000069 breast epithelial cell Anatomy 0.000 description 1
- 201000011510 cancer Diseases 0.000 description 1
- 231100000357 carcinogen Toxicity 0.000 description 1
- 239000003183 carcinogenic agent Substances 0.000 description 1
- 210000004027 cell Anatomy 0.000 description 1
- 238000013527 convolutional neural network Methods 0.000 description 1
- 238000010586 diagram Methods 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 210000001165 lymph node Anatomy 0.000 description 1
- 230000003211 malignant effect Effects 0.000 description 1
- 238000003062 neural network model Methods 0.000 description 1
- 238000010606 normalization Methods 0.000 description 1
- 210000000056 organ Anatomy 0.000 description 1
- 230000008569 process Effects 0.000 description 1
- 210000004872 soft tissue Anatomy 0.000 description 1
- 208000024891 symptom Diseases 0.000 description 1
- 238000013526 transfer learning Methods 0.000 description 1
Landscapes
- Image Analysis (AREA)
- Image Processing (AREA)
Abstract
本发明涉及乳腺癌图像处理技术领域，尤其是一种乳腺癌图像深度学习识别模型训练方法及系统，包括获取乳腺癌MRI图像；进行乳腺癌MRI图像的特征提取；通过VGG16网络模型提取的乳腺癌MRI特征图像输入至SVM分类器中进行分类；以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练；选择预训练后的乳腺癌MRI图像作为第二训练数据，对预训练后的VGG16网络模型进行训练，得到用于乳腺癌MRI图像识别的深度学习模型。通过深度学习识别模型训练，对乳腺肿瘤图像的识别、训练，实现了乳腺肿瘤图像的病变识别判断，且摒弃了传统的人工筛查方法的低效率方式，对乳腺疾病的影像诊断及评价具有积极意义。
Description
技术领域
本发明涉及乳腺癌图像处理技术领域，尤其涉及一种乳腺癌图像深度学习识别模型训练方法及系统。
背景技术
乳腺癌是乳腺上皮细胞在多种致癌因子的作用下，发生增殖失控的现象。疾病早期常表现为乳房肿块、乳头溢液、腋窝淋巴结肿大等症状，晚期可因癌细胞发生远处转移，出现多器官病变，直接威胁患者的生命，乳腺MRI检查因其无创、信息量丰富和软组织分辨率高的优势，成为乳腺癌影像诊断及评价的重要检查，临床上，通过对乳腺MRI图像的计算机处理和分析来辅助医师对乳腺病变进行良恶性分类，以达到提高诊断准确率和为制定个体化诊疗方案提供重要参考依据的目的。
其中深度学习技术可以通过构建的深层次网络直接从网络中得到数据的特征,实现了特征的自适应提取,并广泛地应用于医学图像的疾病诊断中。卷积神经网络由输入层、卷积层、池化层、全连接层和输出层构成，并广泛地应用于计算机视觉领域(包括MRI图像)，在卷积层中，会定义一个卷积核，将卷积核认为是一种“感受野”，在卷积运算时，这个“感受野”会按照从左到右、从上到下的顺序在输入图像上进行滑动卷积运算，通过这样的操作，就可以得到输入图像的特征图像。
通过迁移学习方法，在已训练的神经网络模型VGG16中，进行医学图像的特征提取，结合模型的学习大量数据分析乳腺肿瘤的MRI图像信息，对乳腺疾病的影像诊断及评价，尤其是提高诊断准确率和为制定个体化诊疗方案提供重要参考依据具有积极意义。因此，我们提出一种乳腺癌图像深度学习识别模型训练方法及系统。
发明内容
本发明的目的是为了解决现有技术中存在的缺点，而提出的一种乳腺癌图像深度学习识别模型训练方法及系统。
为达到以上目的，本发明采用的技术方案为：一种乳腺癌图像深度学习识别模型训练方法，包括如下步骤：
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图像识别的深度学习模型。
优选的，在S1中，乳腺癌MRI图像首先进行归一化操作，使乳腺癌MRI图像大小统一为224像素×224像素。
优选的，在S4中，最大池化是在每次对数据操作时，将池化覆盖区域的数据取最大值，表示为：
其中，为上一层的特征图，H为数据的行数，w为数据的列数。
优选的，在S5中，其中准确率公式如下：
通过上述的准确率公式得出不同层的分类准确率。
优选的，在S2中，采用Med3D模型训练框架，以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练，得到模型参数。
优选的，采用MONAI模型训练框架，选择预训练后的模型参数作为第二训练数据，对VGG16网络模型进行训练。
优选的，将预训练后的训练数据作为初始模型参数，对预训练后的VGG16网络模型进行训练。
一种乳腺癌图像深度学习识别模型训练系统，包括有获取单元、预处理单元、预训练单元以及训练单元；
所述获取单元，用以乳腺癌MRI图像的采集获取；
所述预处理单元，用以进行乳腺癌MRI图像的的预处理，包括特征提取、特征降维以及利用SVM分类器进行分类；
所述预训练单元，用以获取乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据进行预训练；
所述训练单元，通过VGG16网络模型以预训练后的乳腺癌MRI图像作为第二训练数据进行训练，并得到用于乳腺癌图像识别的深度学习模型。
与现有技术相比，本发明具有以下有益效果：
通过深度学习识别模型训练，对乳腺肿瘤图像的识别、训练，实现了乳腺肿瘤图像的病变识别判断，且摒弃了传统的人工筛查方法的低效率方式，对乳腺癌治疗过程中制定可靠方案具有积极意义。
附图说明
图1为本发明的流程示意图；
图2为本发明的模型训练方法的流程框图。
具体实施方式
以下描述用于揭露本发明以使本领域技术人员能够实现本发明。以下描述中的优选实施例只作为举例，本领域技术人员可以想到其他显而易见的变型。
实施例1
如附图1-2所示的一种乳腺癌图像深度学习识别模型训练方法，包括如下步骤：
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图像识别的深度学习模型。
一种乳腺癌图像深度学习识别模型训练系统，包括有获取单元、预处理单元、预训练单元以及训练单元；
所述获取单元，用以乳腺癌MRI图像的采集获取；
所述预处理单元，用以进行乳腺癌MRI图像的的预处理，包括特征提取、特征降维以及利用SVM分类器进行分类；
所述预训练单元，用以获取乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据进行预训练；
所述训练单元，通过VGG16网络模型以预训练后的乳腺癌MRI图像作为第二训练数据进行训练，并得到用于乳腺癌图像识别的深度学习模型。
工作时，乳腺癌MRI图像首先进行归一化操作，使乳腺癌MRI图像大小统一为224像素×224像素，进行乳腺癌MRI图像的特征提取，选取第6卷积层、第9卷积层和第12卷积层的特征，进一步的选择第6卷积层、第9卷积层和第12卷积层的各层的下一层进行最大池化后的特征进行分类；选择SVM分类器，将图像数据集按照7∶3的比例分为训练集和测试集，再分别将第6卷积层、第9卷积层和第12卷积层通过VGG16网络模型提取的乳腺癌MRI特征图像输入至SVM分类器中进行分类，得出每一层的分类准确率， 的计算公式，可得出不同层的分类准确率；随即以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练；选择预训练后的乳腺癌MRI图像作为第二训练数据，对预训练后的VGG16网络模型进行训练，得到用于乳腺癌MRI图像识别的深度学习模型，其中，采用Med3D模型训练框架，以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练，得到模型参数；采用MONAI模型训练框架，选择预训练后的模型参数作为第二训练数据，对VGG16网络模型进行训练，将预训练后的训练数据作为初始模型参数，对预训练后的VGG16网络模型进行训练，最终得到深度学习识别模型。
在S1中，乳腺癌MRI图像首先进行归一化操作，使乳腺癌MRI图像大小统一为224像素×224像素，在S4中，最大池化是在每次对数据操作时，将池化覆盖区域的数据取最大值，表示为：
其中，为上一层的特征图，H为数据的行数，w为数据的列数。
在S5中，其中准确率公式如下：
通过上述的准确率公式得出不同层的分类准确率。
在S2中，采用Med3D模型训练框架，以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练，得到模型参数。
采用MONAI模型训练框架，选择预训练后的模型参数作为第二训练数据，对VGG16网络模型进行训练，将预训练后的训练数据作为初始模型参数，对预训练后的VGG16网络模型进行训练。
实施例2
通过上述的实施例1，结合实验进行结果验证，首先于MRI图像数据库内获取相关乳腺肿瘤图像，并通过归一化设置为224*224像素的大小，同时配置VGG16网络参数，硬件采用第八代i7处理器、32GB内存，GeForce1080Ti显卡以及TensorFlow1.4深度学习开源模型；
本发明的使用流程及工作原理：
参照说明书附图1-2所示，在进行乳腺癌图像的识别模型训练时，首先，获取乳腺癌MRI图像；进行乳腺癌MRI图像的特征提取，选取第6卷积层、第9卷积层和第12卷积层的特征，进一步的选择第6卷积层、第9卷积层和第12卷积层的各层的下一层进行最大池化后的特征进行分类；选择SVM分类器，将图像数据集按照7∶3的比例分为训练集和测试集，再分别将第6卷积层、第9卷积层和第12卷积层通过VGG16网络模型提取的乳腺癌MRI特征图像输入至SVM分类器中进行分类，得出每一层的分类准确率， 的计算公式，可得出不同层的分类准确率；
随即以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练；选择预训练后的乳腺癌MRI图像作为第二训练数据，对预训练后的VGG16网络模型进行训练，得到用于乳腺癌MRI图像识别的深度学习模型，其中，采用Med3D模型训练框架，以乳腺癌MRI图像数据库中的乳腺癌图像作为第一训练数据，并选择已训练的VGG16网络模型进行乳腺癌MRI图像进行预训练，得到模型参数；
采用MONAI模型训练框架，选择预训练后的模型参数作为第二训练数据，对VGG16网络模型进行训练，将预训练后的训练数据作为初始模型参数，对预训练后的VGG16网络模型进行训练，最终得到深度学习识别模型，此种模型训练方法能够有效的进行乳腺肿瘤图像的识别、训练，且能够结合乳腺肿瘤图像的病变进行识别判断，有利于在获取病人乳腺癌MRI图像后的进行后续诊断或制定较佳的治疗方案，进一步的有利于乳腺癌的治疗；
综上：通过深度学习识别模型训练，对乳腺肿瘤图像的识别、训练，实现了乳腺肿瘤图像的病变识别判断，且摒弃了传统的人工筛查方法的低效率方式，对乳腺癌治疗过程中制定可靠方案具有积极意义。
以上显示和描述了本发明的基本原理、主要特征和本发明的优点。本行业的技术人员应该了解，本发明不受上述实施例的限制，上述实施例和说明书中描述的只是本发明的原理，在不脱离本发明精神和范围的前提下本发明还会有各种变化和改进，这些变化和改进都落入要求保护的本发明的范围内。本发明要求的保护范围由所附的权利要求书及其等同物界定。
Claims (8)
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图像作为第二训练数据进行训练，并得到用于乳腺癌图像识别的深度学习模型。
Priority Applications (1)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| CN202310772916.8A CN116797838A (zh) | 2023-06-28 | 2023-06-28 | 一种乳腺癌图像深度学习识别模型训练方法及系统 |
Applications Claiming Priority (1)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| CN202310772916.8A CN116797838A (zh) | 2023-06-28 | 2023-06-28 | 一种乳腺癌图像深度学习识别模型训练方法及系统 |
Publications (1)
| Publication Number | Publication Date |
|---|---|
| CN116797838A true CN116797838A (zh) | 2023-09-22 |
Family
ID=88043436
Family Applications (1)
| Application Number | Title | Priority Date | Filing Date |
|---|---|---|---|
| CN202310772916.8A Pending CN116797838A (zh) | 2023-06-28 | 2023-06-28 | 一种乳腺癌图像深度学习识别模型训练方法及系统 |
Country Status (1)
| Country | Link |
|---|---|
| CN (1) | CN116797838A (zh) |
-
2023
- 2023-06-28 CN CN202310772916.8A patent/CN116797838A/zh active Pending
Similar Documents
| Publication | Publication Date | Title |
|---|---|---|
| Qi et al. | Automated diagnosis of breast ultrasonography images using deep neural networks | |
| CN107748900B (zh) | 基于区分性卷积神经网络的乳腺肿瘤分类装置和存储介质 | |
| Mohapatra et al. | Wavelet transform and deep convolutional neural network-based smart healthcare system for gastrointestinal disease detection | |
| Qian et al. | Skin lesion classification using CNNs with grouping of multi-scale attention and class-specific loss weighting | |
| CN109389584A (zh) | 基于cnn的多尺度鼻咽肿瘤分割方法 | |
| Zhuang et al. | Breast ultrasound lesion classification based on image decomposition and transfer learning | |
| CN116681958B (zh) | 基于机器学习的胎儿肺脏超声影像成熟度预测方法 | |
| Yadav et al. | Deep learning-based CAD system design for thyroid tumor characterization using ultrasound images | |
| Zhuang et al. | Tumor classification in automated breast ultrasound (ABUS) based on a modified extracting feature network | |
| CN116310569A (zh) | 一种基于鲁棒纹理特征的乳腺病变检测与分类装置 | |
| CN114565786A (zh) | 基于通道注意力机制的断层扫描图像分类装置和方法 | |
| Zhou et al. | Deep learning-based breast region extraction of mammographic images combining pre-processing methods and semantic segmentation supported by Deeplab v3+ | |
| CN116843984A (zh) | GLTransNet：一种融合全局特征的钼靶图像分类检测方法 | |
| Zhang et al. | A Lung Cancer Detection and Recognition Method Combining Convolutional Neural Network and Morphological Features | |
| Al-Tam et al. | Breast cancer detection and diagnosis using machine learning: a survey | |
| Li et al. | AttentionNet: Learning where to focus via attention mechanism for anatomical segmentation of whole breast ultrasound images | |
| Hassan et al. | A dilated residual hierarchically fashioned segmentation framework for extracting Gleason tissues and grading prostate cancer from whole slide images | |
| Tenali et al. | Oral Cancer Detection using Deep Learning Techniques | |
| Kumar et al. | A Novel Approach for Breast Cancer Detection by Mammograms | |
| CN111275103A (zh) | 多视角信息协作的肾脏良恶性肿瘤分类方法 | |
| CN109003659A (zh) | 胃幽门螺旋杆菌感染病理诊断支持系统和方法 | |
| Chen et al. | A new classification method in ultrasound images of benign and malignant thyroid nodules based on transfer learning and deep convolutional neural network | |
| CN107463964A (zh) | 一种基于超声图像特征相关性的乳腺肿瘤分类方法、装置 | |
| CN118552563A (zh) | 一种基于窗口注意力语义流对齐的乳腺超声图像分割方法 | |
| Atrey et al. | Integration of ultrasound and mammogram for multimodal classification of breast cancer using hybrid residual neural network and machine learning |
Legal Events
| Date | Code | Title | Description |
|---|---|---|---|
| PB01 | Publication | ||
| PB01 | Publication | ||
| SE01 | Entry into force of request for substantive examination | ||
| SE01 | Entry into force of request for substantive examination |