Nothing Special   »   [go: up one dir, main page]

CN112446891A - 基于U-Net网络脑胶质瘤的医学图像分割方法 - Google Patents

基于U-Net网络脑胶质瘤的医学图像分割方法 Download PDF

Info

Publication number
CN112446891A
CN112446891A CN202011144947.1A CN202011144947A CN112446891A CN 112446891 A CN112446891 A CN 112446891A CN 202011144947 A CN202011144947 A CN 202011144947A CN 112446891 A CN112446891 A CN 112446891A
Authority
CN
China
Prior art keywords
convolution
net network
layer
network model
data
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.)
Granted
Application number
CN202011144947.1A
Other languages
English (en)
Other versions
CN112446891B (zh
Inventor
张聚
陈德臣
牛彦
施超
潘玮栋
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Zhejiang University of Technology ZJUT
Original Assignee
Zhejiang University of Technology ZJUT
Priority date (The priority date 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 date listed.)
Filing date
Publication date
Application filed by Zhejiang University of Technology ZJUT filed Critical Zhejiang University of Technology ZJUT
Priority to CN202011144947.1A priority Critical patent/CN112446891B/zh
Publication of CN112446891A publication Critical patent/CN112446891A/zh
Application granted granted Critical
Publication of CN112446891B publication Critical patent/CN112446891B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/10Segmentation; Edge detection
    • G06T7/11Region-based segmentation
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/214Generating training patterns; Bootstrap methods, e.g. bagging or boosting
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods
    • G06N3/084Backpropagation, e.g. using gradient descent
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/0002Inspection of images, e.g. flaw detection
    • G06T7/0012Biomedical image inspection
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/10Image acquisition modality
    • G06T2207/10072Tomographic images
    • G06T2207/10088Magnetic resonance imaging [MRI]
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20081Training; Learning
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20084Artificial neural networks [ANN]
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/30Subject of image; Context of image processing
    • G06T2207/30004Biomedical image processing
    • G06T2207/30016Brain
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/30Subject of image; Context of image processing
    • G06T2207/30004Biomedical image processing
    • G06T2207/30096Tumor; Lesion

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • General Physics & Mathematics (AREA)
  • Data Mining & Analysis (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • General Health & Medical Sciences (AREA)
  • Evolutionary Computation (AREA)
  • Health & Medical Sciences (AREA)
  • General Engineering & Computer Science (AREA)
  • Artificial Intelligence (AREA)
  • Computational Linguistics (AREA)
  • Computing Systems (AREA)
  • Biomedical Technology (AREA)
  • Biophysics (AREA)
  • Software Systems (AREA)
  • Mathematical Physics (AREA)
  • Molecular Biology (AREA)
  • Bioinformatics & Computational Biology (AREA)
  • Evolutionary Biology (AREA)
  • Bioinformatics & Cheminformatics (AREA)
  • Medical Informatics (AREA)
  • Nuclear Medicine, Radiotherapy & Molecular Imaging (AREA)
  • Radiology & Medical Imaging (AREA)
  • Quality & Reliability (AREA)
  • Image Analysis (AREA)

Abstract

基于U‑Net网络脑胶质瘤的医学图像分割方法,包括:步骤1)对所选数据集进行预处理;步骤2)构建网络模型;步骤3)设置训练策略;步骤4)训练网络并更新参数;5)评估模型;6)将验证集送入训练好的模型。本发明先是对U‑Net的整体架构进行了调整,并在此基础上加入了在本文中称之为refine的模块,refine模块融合了1x1卷积、空洞卷积和全局平均池化的模块。加入的空洞卷积能使得网络的感受野在不增加卷积核参数的情况下得到有效的提升,1x1卷积可以提取更多的细节特征,通过这两部分的结合,能有效地集成粗糙的全局特征和细节的局部特征。全局平均池化能得到各个通道间的相关性,能得到模型在空间上更多的特征信息。

Description

基于U-Net网络脑胶质瘤的医学图像分割方法
技术领域
本发明属于深度学习技术领域,应用于医学图像分割方面,具体使用了一种新的基于U-Net网络脑胶质瘤的医学图像分割方法。
背景技术及意义
医学图像分割是从医学图像中识别器官或病变的像素点,是医学图像诊断和计算机辅助诊断系统中不可缺少的步骤,也是医学图像分析中最具挑战性的任务之一。其中脑肿瘤的医学图像分割是脑肿瘤研究的关键步骤,对脑肿瘤的精确分割能更加准确地对脑肿瘤进行分类并有效地帮助后续的治疗、也是准确设计放疗或手术治疗肿瘤的关键步骤。传统的医学图像分割算法包括基于阀值的分割算法、基于边缘的分割算法、基于区域的分割算法、基于图论的分割算法、基于聚类的分割算法等。
近几年来,深度学习的方法在计算机视觉领域取得了巨大的突然。其中卷积神经网络在医学图像分割任务中的表现均超过了传统的算法。在处理医学图像领域中,最常见的分割网络是U-Net网络,U-Net的主要架构是以相同数量的上采样层和下采样层组合,上采样层和下采样层为对称结构,并以跳跃连接的方式融合上下采样层中高级和低级特征信息,从而得到更加精确的分割结果。
随着深度学习的不断发展,如何更好地优化之前的模型并将其应用到医学图像中来,使其分割的医学图像能更加地准确成了目前研究的重要领域。
发明内容
本发明要克服现有技术的上述缺点,提供一种基于U-Net网络脑胶质瘤的医学图像分割方法。
本发明在U-Net网络模型的大体框架上加入了在本文中称之为refine的模块。refine模块融合了1x1卷积、空洞卷积和全局平均池化的模块。加入的空洞卷积能使得网络的感受野在不增加卷积核参数的情况下得到有效的提升,有利于在卷积核中建立全局视野,提取整体特征,1x1卷积可以提取更多的细节特征,通过这两部分的结合,能有效地集成粗糙的全局特征和细节的局部特征。全局平均池化能得到各个通道间的相关性,将通过全局平均池化得到的结果与前两个模块的所得到的结果相乘。能得到在空间上更多的特征信息。
为使本发明的目的、技术方案和优点更加清晰,下面就对本发明的技术方案作进一步描述。本发明的基于U-Net网络脑胶质瘤的医学图像分割方法,包括如下步骤:
步骤1)对所选的数据集进行图像预处理;
首先读取含有脑胶质瘤患者的MR(核磁共振)图像并将其转换成灰度图,然后通过对原数据集进行裁剪、翻转、旋转、缩放、移位等方式来扩大数据集,这个操作称之为数据增强,数据增强可以增加训练的数据量,提高U-Net网络模型的泛化能力,同时也可以增加噪音数据,提升U-Net网络模型的鲁棒性。最后对数据进行归一化处理。
步骤2)构建U-Net网络模型;
2.1.本发明所提出的refine模块:
refine模块一共由两条路径组成,第一条路径是将输入的数据经过全局平均池化层,全局平均池化是将每一层特征图求平均值,然后通过全局平均池化层之后再通过两层全连接层,最后得到表示通道之间相关性的数据。第二条路径是由1x1卷积和空洞卷积组成,其中空洞卷积的膨胀率每一层都设置不同的数值,从上而下空洞卷积的膨胀率分别为12、6、3、3。数据分别经过1x1卷积和空洞卷积,然后将得到的结果进行想加,1x1卷积能够提取更多的细节特征,空洞卷积能使网络的感受野在卷积核参数不变的情况下得到有效的提升,通过这两部分的结合,能有效地集成粗糙的全局特征和细节的局部特征。最后将两条路径得到的结果进行相乘。
2.2.以U-Net网络模型为基础的框架:
U-Net网络模型包含4组对称结构的上采样和下采样,分为编码部分和解码部分,自上而下一共分为6层。除了首尾两层以外,每一层都包含了两个卷积层,每一层经过第一个卷积操作之后会通过ReLU激活函数,然后经过第二个卷积操作,并通过批标准化和ReLU激活函数,接着在每次下采样部分都会通过Dropout操作,Dropout可以使某个神经元的激活值以一定的概率停止工作,这样可以使U-Net网络模型的泛化性更强,可以比较有效地缓解过拟合的发送,在一定程度上达到正则化的效果。
2.3.将refine模块加入到调整后的U-Net网络中:
这个小节介绍本发明所提出的U-Net网络模型的具体流程:先是编码部分,输入的图片是256x256x3尺寸,接着进入第二层,在第二层中先是经过了两个卷积操作,第一个卷积核是7x7尺寸并且添加了Relu激活函数;第二个卷积核是5x5尺寸并且加入了BN(batchnormalization批标准化)和Relu激活函数。BN就是通过一定的规范化手段,把每层神经网络任意神经元这个输入值的分布强行拉回到均值为0方差为取得标准正态分布,这样能避免梯度消失问题的产生并且大大加快训练速度。然后加入Dropout操作,最后进行最大池化操作,池化操作之后表示第一层结束。接下来的三、四、五层中,操作和第二层类似,只是卷积核的不同,第六层,采用1x1的卷积核是为了提取图像中更多的细节信息。然后是解码部分,先是将编码部分得到的结果进行上采样操作,然后将第五层中下采样得到的结果通过refine模块,最后将这两部的数据进行连接起来,通过concatenate操作后再进行两次卷积,接着进行以上的操作依次向上最终得到输出图。
步骤3)设置训练策略;
将经过预处理的数据集以6:2:2的比例依次划分为训练集、测试集和验证集。采用随机初始化和Adam优化算法。根据电脑配置等设置BatchSize(一次训练所选取的样本数)以及epoch(表示回合的意思,训练完所有的数据表示一回合)。设置合适的初始学习率,一阶矩估计的指数衰减率、二阶矩估计的指数衰减率、每次更新时学习率下降的值。
步骤4)训练U-Net网络模型和更新参数;
根据设置好的训练策略对U-Net网络模型进行训练。更新参数:U-Net网络模型中采用反向传播算法(BP)算法对网络中的权重和偏置进行更新。训练迭代过程中利用损失函数(Dice Loss)进行参数地更新。
损失函数Dice Loss公式为:
Figure BDA0002739356030000031
y代表真实值,p代表预测值。
步骤5)评估U-Net网络模型;
建立包括精确度Precision、召回率Recall、准确度ACC、骰子系数DC、F1分数的评价指标,通过计算评价指标将我们的U-Net网络模型与最先进的用于医学图像的深度学习模型进行比较从而评估我们方法的性能。
评价标准的公示如下所示:
Figure BDA0002739356030000041
表示在所有预测为正样本的数据中,有多少是真正的正样本;
Figure BDA0002739356030000042
表示所有被正确预测为证样本的样本数量与真实的正样本的总数量的比值;
Figure BDA0002739356030000043
表示所有的样本数据中,有多少是被正确预测的;
Figure BDA0002739356030000044
表示正样本和预测样本间的相似度;
Figure BDA0002739356030000045
表示为精确率和召回率的调和平均数;
其中TP为真阳性,TN为真阴性,FP为假阳性,FN为假阴性。
步骤6)将验证集送入训练好的U-Net网络模型;
将之前分割好的验证集送入本发明中所训练好的U-Net网络模型,输出的结果会随机地从验证集中挑选50张患者的MR图像进行分割效果的可视化,这50张患者的MR图像中包含了不同类型的肿瘤。分割效果的可视化图片中一共包含了3张图片。从左往右第一张图片表示含有低级神经胶质瘤的脑部MR图像,第二张表示专家分割出的肿瘤图像,第三张表示使用本发明中提出的U-Net网络模型所分割出的肿瘤图像。通过对比专家分割出的肿瘤图像和本发明中训练好的U-Net网络模型所分割出的图像来对模型进行评估。
本发明具有以下优点:
1.本发明加入refine模块,refine模块融合了1x1卷积、空洞卷积和全局平均池化的模块。加入的空洞卷积能使得网络的感受野在不增加卷积核参数的情况下得到有效的提升,有利于在卷积核中建立全局视野,提取整体特征,1x1卷积可以提取更多的细节特征,通过这两部分的结合,能有效地集成粗糙的全局特征和细节的局部特征。全局平均池化能得到各个通道间的相关性,将通过全局平均池化得到的结果与前两个模块的所得到的结果相乘。能得到在空间上更多的特征信息。
2.本发明在U-Net网络模型的基础上加入了Dropout,Dropout可以使某个神经元的激活值以一定的概率停止工作,这样可以使U-Net网络模型的泛化性更强,可以比较有效地缓解过拟合的发送,在一定程度上达到正则化的效果。
附图说明
图1是本发明所提出的U-Net网络结构图;
图2是本发明所提出的refine模块结构图;
图3是本发明所提出网络的具体的流程图;
图4a~图4c是使用本发明的网络对脑胶质瘤进行分割的示意图,图4a表示含有低级神经胶质瘤的脑部MR图像,图4b表示专家分割出的肿瘤图像,图4c表示使用本发明中提出的网络模型所分割出的肿瘤图像。
具体实施方式
以下结合附图对本发明做进一步说明:
本发明的一种新的基于U-Net网络脑胶质瘤的医学图像分割方法具体包括以下步骤:
步骤1)对所选的数据集进行图像预处理;
首先读取含有脑胶质瘤患者的MR(核磁共振)图像并将其转换成灰度图,然后通过对原数据集进行裁剪、翻转、旋转、缩放、移位等方式来扩大数据集,这个操作称之为数据增强,数据增强可以增加训练的数据量,提高U-Net网络模型的泛化能力,同时也可以增加噪音数据,提升U-Net网络模型的鲁棒性。最后对数据进行归一化处理。
步骤2)构建U-Net网络模型;
2.1本发明所提出的模块:
如图2所示。这个模块一共由两条路径组成,第一条路径是将输入的数据先经过全局平均池化层(图中用GAP表示),全局平均池化是将每一层特征图求平均值,然后通过2层全连接层(图中用Dense表示),最后得到表示通道之间相关性的数据。第二条路径是由1x1卷积(图中用Conv1x1表示)和空洞卷积(图中用Dilated conv表示)组成,其中空洞卷积的膨胀率每一层都设置不同的数值,从上而下空洞卷积的膨胀率分别为12、6、3、3。数据分别经过1x1卷积和空洞卷积,最后将得到的结果进行想加,1x1卷积能够提取更多的细节特征,空洞卷积能使网络的感受野在卷积核参数不变的情况下得到有效的提升,通过这两条部分的结合,能有效地集成粗糙的全局特征和细节的局部特征。最后将两条路径得到的结果进行相乘。
2.2.以U-Net网络模型为基础的框架:
如图1所示,U-Net网络模型包含4组对称结构的上采样和下采样,分为编码部分和解码部分,自上而下一共分为6层。除首尾两层外每一层都包含了两个卷积层,每一层经过第一个卷积操作之后会通过ReLU激活函数,然后经过第二个卷积操作,并通过批标准化和ReLU激活函数,接着在下采样部分之前会通过Dropout操作,Dropout可以使某个神经元的激活值以一定的概率停止工作,这样可以使U-Net网络模型的泛化性更强,可以比较有效地缓解过拟合的发送,在一定程度上达到正则化的效果。
2.3.将refine模块加入到调整后的U-Net网络模型中:
在图3中,详细地介绍了本发明所提出的U-Net网络模型的具体流程:先是编码部分,输入的图片是256x256x3尺寸,接着进入第二层,在第二层中先是经过了两个卷积操作,第一个卷积核是7x7尺寸并且添加了Relu激活函数;第二个卷积核是5x5尺寸并且加入了BN(batch normalization批标准化)和Relu激活函数。BN就是通过一定的规范化手段,把每层神经网络任意神经元这个输入值的分布强行拉回到均值为0方差为取得标准正态分布,这样能避免梯度消失问题的产生并且大大加快训练速度。然后加入Dropout操作,最后进行最大池化操作(图中用Maxpol表示),池化操作之后表示第一层结束。接下来的三、四、五层中,操作和第二层类似,只是卷积核的不同,具体的数据如图3所示。第六层,采用1x1的卷积核是为了提取图像中更多的细节信息。然后是解码部分,先是将编码部分得到的结果进行上采样操作(图中用Deconv表示),然后将第五层中下采样得到的结果通过refine模块,最后将这两部的数据进行连接起来(图中用concatenate表示),通过concatenate操作后再进行两次卷积,如流程图所示进行以上的操作依次向上最终得到输出图。
步骤3)设置训练策略;
将经过预处理的数据集以6:2:2的比例依次划分为训练集、测试集和验证集。采用随机初始化和Adam优化算法。根据电脑配置等设置BatchSize(一次训练所选取的样本数)以及epoch(表示回合的意思,训练完所有的数据表示一回合)。设置合适的初始学习率,一阶矩估计的指数衰减率、二阶矩估计的指数衰减率、每次更新时学习率下降的值。
步骤4)训练网络和更新参数;
根据设置好的训练策略对U-Net网络模型进行训练。更新参数:U-Net网络模型中采用反向传播算法(BP)算法对网络中的权重和偏置进行更新。训练迭代过程中利用损失函数(Dice Loss)进行参数地更新。
损失函数Dice Loss公式为:
Figure BDA0002739356030000071
y代表真实值,p代表预测值。
步骤5)评估U-Net网络模型;
建立包括精确度Precision、召回率Recall、准确度ACC、骰子系数DC、F1分数的评价指标,通过计算评价指标将我们的U-Net网络模型与最先进的用于医学图像分割的深度学习模型进行比较从而评估我们方法的性能。
评价标准的公示如下所示:
Figure BDA0002739356030000072
表示在所有预测为正样本的数据中,有多少是真正的正样本;
Figure BDA0002739356030000073
表示所有被正确预测为证样本的样本数量与真实的正样本的总数量的比值;
Figure BDA0002739356030000074
表示所有的样本数据中,有多少是被正确预测的;
Figure BDA0002739356030000075
表示正样本和预测样本间的相似度;
Figure BDA0002739356030000076
表示为精确率和召回率的调和平均数;
其中TP为真阳性,TN为真阴性,FP为假阳性,FN为假阴性。
步骤6)将验证集送入训练好的U-Net网络模型;
将之前分割好的验证集送入本发明中所训练好的U-Net网络模型,输出的结果会随机地从验证集中挑选50张患者的MR图像进行分割效果的可视化,这50张患者的MR图像中包含了不同类型的肿瘤。分割效果的可视化图片中一共包含了3张图片。如图4a~图4c所示,图4a表示含有低级神经胶质瘤的脑部MR图像,图4b表示专家分割出的肿瘤图像,其中白色的部分表示神经胶质瘤,图4c表示使用本发明中提出的U-Net网络模型所分割出的肿瘤图像。通过对比专家分割出的肿瘤图像和本发明中训练好的U-Net网络模型所分割出的图像来对模型进行评估。

Claims (2)

1.基于U-Net网络脑胶质瘤的医学图像分割方法,具体步骤如下:
步骤1)对所选的数据集进行图像预处理;
数据增强,首先读取含有脑胶质瘤患者的MR(核磁共振)图像并将其转换成灰度图,然后通过对原数据集进行裁剪、翻转、旋转、缩放、移位等方式来扩大数据集;最后对数据进行归一化处理;
步骤2)构建U-Net网络模型;
2.1.构建refine模块:
refine模块由两条路径组成,第一条路径是将输入的数据先经过全局平均池化层,全局平均池化是将每一层特征图求平均值,然后通过2层全连接层,最后得到表示通道之间相关性的数据;第二条路径是由1x1卷积和空洞卷积组成,其中空洞卷积的膨胀率每一层都设置不同的数值,从上而下空洞卷积的膨胀率分别为12、6、3、3;数据分别经过1x1卷积和空洞卷积,最后将得到的结果进行想加,1x1卷积能够提取更多的细节特征,空洞卷积能使网络的感受野在卷积核参数不变的情况下得到有效的提升,通过这两条部分的结合,能有效地集成粗糙的全局特征和细节的局部特征;最后将两条路径得到的结果进行相乘;
2.2.以U-Net网络模型为基础的框架:
U-Net网络模型包含4组对称结构的上采样和下采样,分为编码部分和解码部分,自上而下一共分为6层;除了首尾两层以外,每一层都包含了两个卷积层,每一层经过第一个卷积操作之后会通过ReLU激活函数,然后经过第二个卷积操作,并通过批标准化和ReLU激活函数,接着在下采样部分之前通过Dropout操作;
2.3.将refine模块加入到U-Net网络模型中:
先是编码部分,输入的图片是256x256x3尺寸,接着进入第二层,在第二层中先是经过了两个卷积操作,第一个卷积核是7x7尺寸并且添加了Relu激活函数;第二个卷积核是5x5尺寸并且加入了BN(batch normalization批标准化)和Relu激活函数;BN就是通过一定的规范化手段,把每层神经网络任意神经元这个输入值的分布强行拉回到均值为0方差为取得标准正态分布;然后加入Dropout操作,最后进行最大池化操作,池化操作之后表示第一层结束;接下来的三、四、五层中,操作和第二层类似,只是卷积核的不同,第六层,采用1x1的卷积核是为了提取图像中更多的细节信息;然后是解码部分,先是将编码部分得到的结果进行上采样操作,然后将第五层中下采样得到的结果通过refine模块,最后将这两部的数据进行连接起来,通过concatenate操作后再进行两次卷积,接着进行以上的操作依次向上最终得到输出图;
步骤3)设置训练策略;
将经过预处理的数据集以6:2:2的比例依次划分为训练集、测试集和验证集;采用随机初始化和Adam优化算法;根据电脑配置等设置一次训练所选取的样本数BatchSize以及回合数epoch;设置合适的初始学习率,一阶矩估计的指数衰减率、二阶矩估计的指数衰减率、每次更新时学习率下降的值;
步骤4)训练网络和更新参数;
根据设置好的训练策略对U-Net网络模型进行训练;更新参数:U-Net网络网络模型采用反向传播算法(BP)算法对网络中的权重和偏置进行更新;训练迭代过程中利用损失函数(Dice Loss)进行参数地更新;
损失函数Dice Loss公式为:
Figure FDA0002739356020000021
y代表真实值,p代表预测值;
步骤5);评估U-Net网络模型;
建立包括精确度Precision、召回率Recall、准确度ACC、骰子系数DC、F1分数的评价指标,通过计算评价指标将所述的U-Net网络模型与最先进的用于分割医学图像的深度学习模型进行比较;
评价标准的公示如下所示:
Figure FDA0002739356020000022
表示在所有预测为正样本的数据中,有多少是真正的正样本;
Figure FDA0002739356020000023
表示所有被正确预测为证样本的样本数量与真实的正样本的总数量的比值;
Figure FDA0002739356020000024
表示所有的样本数据中,有多少是被正确预测的;
Figure FDA0002739356020000025
表示正样本和预测样本间的相似度;
Figure FDA0002739356020000026
表示为精确率和召回率的调和平均数;
其中TP为真阳性,TN为真阴性,FP为假阳性,FN为假阴性;
步骤6);将验证集送入训练好的U-Net网络模型;
将之前分割好的验证集送入训练好的U-Net网络模型,输出的结果会随机地从验证集中挑选若干张患者的MR图像进行分割效果的可视化,这些患者的MR图像中包含了不同类型的肿瘤;分割效果的可视化图片中一共包含了表示含有低级神经胶质瘤的脑部MR图像、表示专家分割出的肿瘤图像、表示使用本发明中提出的U-Net网络模型所分割出的肿瘤图像;通过对比专家分割出的肿瘤图像和本发明中训练好的U-Net网络模型所分割出的图像来对模型进行评估。
2.如权利要求1所述的基于U-Net网络脑胶质瘤的医学图像分割方法,其特征在于:步骤6)所述的患者的MR图像选用50张。
CN202011144947.1A 2020-10-23 2020-10-23 基于U-Net网络脑胶质瘤的医学图像分割方法 Active CN112446891B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202011144947.1A CN112446891B (zh) 2020-10-23 2020-10-23 基于U-Net网络脑胶质瘤的医学图像分割方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202011144947.1A CN112446891B (zh) 2020-10-23 2020-10-23 基于U-Net网络脑胶质瘤的医学图像分割方法

Publications (2)

Publication Number Publication Date
CN112446891A true CN112446891A (zh) 2021-03-05
CN112446891B CN112446891B (zh) 2024-04-02

Family

ID=74735554

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202011144947.1A Active CN112446891B (zh) 2020-10-23 2020-10-23 基于U-Net网络脑胶质瘤的医学图像分割方法

Country Status (1)

Country Link
CN (1) CN112446891B (zh)

Cited By (9)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN113177938A (zh) * 2021-05-25 2021-07-27 深圳大学 基于圆形卷积核的脑胶质瘤的分割方法、装置及相关组件
CN113240677A (zh) * 2021-05-06 2021-08-10 浙江医院 一种基于深度学习的视网膜视盘分割方法
CN113384261A (zh) * 2021-05-28 2021-09-14 华南理工大学 一种基于深度学习的椎体压缩性骨折多模态智能诊断系统
CN113658142A (zh) * 2021-08-19 2021-11-16 江苏金马扬名信息技术股份有限公司 一种基于改进U-Net神经网络的髋关节股骨近端分割方法
CN114004836A (zh) * 2022-01-04 2022-02-01 中科曙光南京研究院有限公司 一种基于深度学习的自适应生物医学影像分割方法
CN114283153A (zh) * 2021-11-15 2022-04-05 天津大学 基于神经网络的肺部病理切片癌细胞分割算法
CN114359292A (zh) * 2021-12-10 2022-04-15 南昌大学 一种基于多尺度和注意力的医学图像分割方法
CN115406815A (zh) * 2022-11-02 2022-11-29 杭州华得森生物技术有限公司 基于多源数据融合的肿瘤细胞检测设备及其方法
US11580646B2 (en) 2021-03-26 2023-02-14 Nanjing University Of Posts And Telecommunications Medical image segmentation method based on U-Net

Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN110675419A (zh) * 2019-10-11 2020-01-10 上海海事大学 一种自适应注意门的多模态脑胶质瘤影像分割方法
CN111046921A (zh) * 2019-11-25 2020-04-21 天津大学 基于U-Net网络和多视角融合的脑肿瘤分割方法
CN111192245A (zh) * 2019-12-26 2020-05-22 河南工业大学 一种基于U-Net网络的脑肿瘤分割网络及分割方法
CN111325751A (zh) * 2020-03-18 2020-06-23 重庆理工大学 基于注意力卷积神经网络的ct图像分割系统

Patent Citations (4)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN110675419A (zh) * 2019-10-11 2020-01-10 上海海事大学 一种自适应注意门的多模态脑胶质瘤影像分割方法
CN111046921A (zh) * 2019-11-25 2020-04-21 天津大学 基于U-Net网络和多视角融合的脑肿瘤分割方法
CN111192245A (zh) * 2019-12-26 2020-05-22 河南工业大学 一种基于U-Net网络的脑肿瘤分割网络及分割方法
CN111325751A (zh) * 2020-03-18 2020-06-23 重庆理工大学 基于注意力卷积神经网络的ct图像分割系统

Non-Patent Citations (1)

* Cited by examiner, † Cited by third party
Title
YING CHEN等: ""Hybrid-attention densely connected U-Net with GAP for extracting livers from CT volumes"", 《MEDICAL PHYSICS》 *

Cited By (11)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
US11580646B2 (en) 2021-03-26 2023-02-14 Nanjing University Of Posts And Telecommunications Medical image segmentation method based on U-Net
CN113240677A (zh) * 2021-05-06 2021-08-10 浙江医院 一种基于深度学习的视网膜视盘分割方法
CN113240677B (zh) * 2021-05-06 2022-08-02 浙江医院 一种基于深度学习的视网膜视盘分割方法
CN113177938A (zh) * 2021-05-25 2021-07-27 深圳大学 基于圆形卷积核的脑胶质瘤的分割方法、装置及相关组件
CN113384261A (zh) * 2021-05-28 2021-09-14 华南理工大学 一种基于深度学习的椎体压缩性骨折多模态智能诊断系统
CN113658142A (zh) * 2021-08-19 2021-11-16 江苏金马扬名信息技术股份有限公司 一种基于改进U-Net神经网络的髋关节股骨近端分割方法
CN113658142B (zh) * 2021-08-19 2024-03-12 江苏金马扬名信息技术股份有限公司 一种基于改进U-Net神经网络的髋关节股骨近端分割方法
CN114283153A (zh) * 2021-11-15 2022-04-05 天津大学 基于神经网络的肺部病理切片癌细胞分割算法
CN114359292A (zh) * 2021-12-10 2022-04-15 南昌大学 一种基于多尺度和注意力的医学图像分割方法
CN114004836A (zh) * 2022-01-04 2022-02-01 中科曙光南京研究院有限公司 一种基于深度学习的自适应生物医学影像分割方法
CN115406815A (zh) * 2022-11-02 2022-11-29 杭州华得森生物技术有限公司 基于多源数据融合的肿瘤细胞检测设备及其方法

Also Published As

Publication number Publication date
CN112446891B (zh) 2024-04-02

Similar Documents

Publication Publication Date Title
CN112446891B (zh) 基于U-Net网络脑胶质瘤的医学图像分割方法
CN111192245B (zh) 一种基于U-Net网络的脑肿瘤分割网络及分割方法
CN111738363B (zh) 基于改进的3d cnn网络的阿尔茨海默病分类方法
CN110969626B (zh) 基于3d神经网络的人脑核磁共振影像的海马体提取方法
CN109272048B (zh) 一种基于深度卷积神经网络的模式识别方法
CN106228185B (zh) 一种基于神经网络的通用图像分类识别系统及方法
CN105139004B (zh) 基于视频序列的人脸表情识别方法
CN112634211B (zh) 基于多个神经网络的mri图像分割方法、装置及设备
CN111009324A (zh) 脑网络多特征分析的轻度认知障碍辅助诊断系统及方法
CN115496720A (zh) 基于ViT机制模型的胃肠癌病理图像分割方法及相关设备
CN112288749A (zh) 一种基于深度迭代融合深度学习模型的颅骨图像分割方法
CN112529886A (zh) 一种基于Attention DenseUNet的MRI神经胶质瘤分割方法
CN112750137A (zh) 基于深度学习的肝脏肿瘤分割方法及系统
CN115035127A (zh) 一种基于生成式对抗网络的视网膜血管分割方法
CN117036288A (zh) 一种面向全切片病理图像的肿瘤亚型诊断方法
CN117611599B (zh) 融合中心线图和增强对比度网络的血管分割方法及其系统
CN118094371A (zh) 基于多尺度残差网络和改进gru的滚动轴承故障诊断方法
Erwin et al. BVU-Net: A U-net modification by VGG-batch normalization for retinal blood vessel segmentation
CN112861881A (zh) 一种基于改进MobileNet模型的蜂窝肺识别方法
Vinta et al. Segmentation and Classification of Interstitial Lung Diseases Based on Hybrid Deep Learning Network Model
CN114862868B (zh) 基于ct灌注源数据的脑卒中最终梗死区分割方法
CN116228731A (zh) 一种多对比学习冠状动脉高危斑块检测方法、系统及终端
CN116503593A (zh) 一种基于深度学习的视网膜oct图像积液分割方法
CN115131628A (zh) 一种基于分型辅助信息的乳腺图像分类方法及设备
Li et al. A Multi-Category Brain Tumor Classification Method Bases on Improved ResNet50.

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
GR01 Patent grant
GR01 Patent grant