一种医学影像小病灶分割方法技术

本发明专利技术公开了一种医学影像小病灶分割方法,包括由粗略分割第一阶段、细化第二阶段和分割错误区域的注意力模块构成分割网络，在第一阶段的训练中使用五折交叉验证；在交叉验证期间，训练集的每个样本都将包含在验证集中，所有样本都有机会被视为验证集的数据并在相应训练集上训练的模型上进行测试；然后将第一阶段预测的结果与真实分割的结果进行比较，得到的差异就反应了模型难以预测的部分，将不匹配信息作为第二阶段的监督；在第二阶段将信息增强特征输入到DA模块中，使用注意力机制，从而加强网络的分割精度。

【技术实现步骤摘要】
一种医学影像小病灶分割方法
本专利技术属于计算机视觉的语义分割领域，涉及一种医学影像小病灶分割方法。
技术介绍
语义分割算法：语义分割算法是针对图像上每一个像素的分类，是像素级别的问题，因此在训练集中需要对图像的每个像素附加一个标签。用公式来表示即为：从标签空间L＝{l1,l2,l3,…,lk}表示一组随机变量X＝{x1,x2,x3,…,xN}。每个标签l表示不同的类或对象，例如，飞机、汽车、交通标志等。这个标记空间具有k个可能的状态，这些状态通常扩展到k+1个，并将l和0作为背景或空类。x表示图像的像素，像素个数为N。目前应用广泛的语义分割网络都是基于FCN来进行改进的。FCN网络利用CNNs在图像上的强大学习能力提出了一个全卷积化概念，将现有的一些常用分类深度网络模型如VGG16、GoogLeNet等网络的全连接层全部用卷积层来替代，这样做的好处最终输出的结果是一张图片而不是一维向量，实现了端对端的语义分割；其次，通过去除全连接层能实现任意大小图片的输入，从而保证输入与输出图片大小相等。由于卷积层后接有池化层，池化层又称下采样层，会对图片的分辨率大小产生影响。为了保证输入图片与输出图片的大小相等，FCN网络使用反卷积的方式进行上采样以维持图片的分辨率。注意力机制：注意力机制(AttentionMechanism)源于对人类视觉的研究。在认知科学中，由于信息处理的瓶颈，人类会选择性地关注所有信息的一部分，同时忽略其他可见的信息。上述机制通常被称为注意力机制。人类视网膜不同的部位具有不同程度的信息处理能力，即敏锐度(Acuity)，只有视网膜中央凹部位具有最强的敏锐度。为了合理利用有限的视觉信息处理资源，人类需要选择视觉区域中的特定部分，然后集中关注它。例如，人们在阅读时，通常只有少量要被读取的词会被关注和处理。综上，注意力机制主要有两个方面：·决定需要关注输入的哪部分。·分配有限的信息处理资源给重要的部分。在计算机视觉领域，注意力机制被引入来进行视觉信息处理。注意力是一种机制，或者方法论，并没有严格的数学定义。比如，传统的局部图像特征提取、显著性检测、滑动窗口方法等都可以看作一种注意力机制。在神经网络中，注意力模块通常是一个额外的神经网络，能够硬性选择输入的某些部分，或者给输入的不同部分分配不同的权重。计算机视觉的研究领域的一大热点研究课题是语义分割，而针对3D图像的语义分割在医疗图像领域取得了长足的发展。3D医疗图像的语义分割一般是将3D的CT或MRI图像进行逐像素的分类，从而达到分割病灶区域或器官的目的。参考文献[1]LongJ,ShelhamerE,DarrellT.Fullyconvolutionalnetworksforsemanticsegmentation[C]//ProceedingsoftheIEEEconferenceoncomputervisionandpatternrecognition.2015:3431-3440.[2]RonnebergerO,FischerP,BroxT.U-net:Convolutionalnetworksforbiomedicalimagesegmentation[C]//InternationalConferenceonMedicalimagecomputingandcomputer-assistedintervention.Springer,Cham,2015:234-241.[3]WangC,MacGillivrayT,MacnaughtG,etal.Atwo-stage3DUnetframeworkformulti-classsegmentationonfullresolutionimage[J].arXivpreprintarXiv:1804.04341,2018.[4]WangY,DengZ,HuX,etal.Deepattentionalfeaturesforprostatesegmentationinultrasound[C]//InternationalConferenceonMedicalImageComputingandComputer-AssistedIntervention.Springer,Cham,2018:523-530.
技术实现思路
为了解决现有语义分割技术中存在的3D小目标分割精度不够的技术问题，本专利技术提出一种结合了分散注意力机制的3D卷积网络DANet，可以更好的分割病变区域。提出的DANet具有两阶段的类U结构，第1阶段进行粗略分割，并在第2阶段完成细化。此外，为了使模型对干扰区域更敏感，这些干扰成分将在第1阶段收集，然后用作第2阶段的监督部分。首先，在第一阶段的训练中使用五折交叉验证。在交叉验证期间，训练集的每个样本都将包含在验证集中。因此，所有样本都有机会被视为验证集的数据并在相应训练集上训练的模型上进行测试；然后将第一阶段预测的结果与真实分割的结果进行比较，得到的差异就反应了模型难以预测的部分，将不匹配信息作为第2阶段的监督；最后，在第二阶段将信息增强特征输入到DA模块中，使用注意力机制，从而加强网络的分割精度。本专利技术的技术方案:一种医学影像小病灶分割方法,包括由粗略分割第一阶段、细化第二阶段和分割错误区域的注意力模块构成分割网络，所述分割网络如下步骤实现医学影像小病灶分割：S1，所述第一阶段网络将采样处理后的原始图像进行五折交叉验证的方式训练获得每一个训练数据的分割结果；S2，将第一阶段网络获得的预测分割结果与真实的分割结果进行对比生成网络预测出错的区域；S3,对所述网络预测错误的区域信息进行增强；使预测错误的区域更容易被学习；S4，所述第二阶段的网络对裁剪后的原始图像进行训练与结合分割错误区域的注意力模块获得分割病变区域的网络模型。进一步，所述分割错误区域的注意力模块通过如下步骤实现分割病变区域的网络模型：2.1将增强后的分割错误区域输入到注意力模块，通过注意力模块后得图像每一个像素的权重矩阵，得到权重矩阵的公式如下：上式中g表示将网络经过跳过连接后的特征和经过增强的错误区域的特征连接起来，φ1代表对操作g后得到的特征进行1*1*1的线性变换，最后是用来得到图像每个像素的权重矩阵；2.2将权重矩阵与解码器中该层的图像相乘，即得到了注意力机制的特征图，将该特征图输入下一层解码器；经过530个左右的迭代获得分割病变区域的网络模型。进一步，所述网络分割错误区域增强的步骤：3.1使用全局平均池化将网络分割错误区域中特征图压缩获得全局特征向量；3.2采用重复模式对全局特征向量进行采样到与原始图像相同的大小区域；最终将网络分割错误区域的信息扩展到整个原始图像。与现有技术相比，本专利技术具有的优点：与现有技术相比，本专利技术提出来的结合注意力机制的3D图像语义分割方法：1)性能提高：经过测试3D-UNet以及其他三个U型网络，与3本文档来自技高网...

【技术保护点】
1.一种医学影像小病灶分割方法,其特征在于：包括由粗略分割第一阶段、细化第二阶段和分割错误区域的注意力模块构成分割网络，所述分割网络如下步骤实现医学影像小病灶分割：/nS1，所述第一阶段网络将采样处理后的原始图像进行五折交叉验证的方式训练获得每一个训练数据的分割结果；/nS2，将第一阶段网络获得的预测分割结果与真实的分割结果进行对比生成网络预测出错的区域；/nS3,对所述网络预测错误的区域信息进行增强；使预测错误的区域更容易被学习；/nS4，所述第二阶段的网络对裁剪后的原始图像进行训练与结合分割错误区域的注意力模块获得分割病变区域的网络模型。/n

【技术特征摘要】
1.一种医学影像小病灶分割方法,其特征在于：包括由粗略分割第一阶段、细化第二阶段和分割错误区域的注意力模块构成分割网络，所述分割网络如下步骤实现医学影像小病灶分割：
S1，所述第一阶段网络将采样处理后的原始图像进行五折交叉验证的方式训练获得每一个训练数据的分割结果；
S2，将第一阶段网络获得的预测分割结果与真实的分割结果进行对比生成网络预测出错的区域；
S3,对所述网络预测错误的区域信息进行增强；使预测错误的区域更容易被学习；
S4，所述第二阶段的网络对裁剪后的原始图像进行训练与结合分割错误区域的注意力模块获得分割病变区域的网络模型。


2.根据权利要求1所述的一种医学影像小病灶分割方法，其特征在于：所述分割错误区域的注意力模块通过如下步骤实现分割病变区域的网络模型：
2.1将增强后的分割错误区域输...

【专利技术属性】
技术研发人员：党萌，万亮，陈峙灏，冯伟，张亚平，
申请(专利权)人：天津大学，
类型：发明
国别省市：天津;12