本发明涉及核磁共振技术领域,具体涉及一种基于三维全卷积神经网络的核磁共振图像的分割方法。
背景技术:
核磁共振成像(magneticresonanceimaging,mri)能够提供高分辨率、高品质的图像,对大脑的解剖结构进行准确地描述,因此,对大脑的核磁共振图像的皮下组织区域进行分割一直是一个重要的研究热点之一,也是脑部诊断的重要辅助手段,对脑部的疾病诊断、治疗及手术引导具有重要的意义。为了充分利用核磁共振图像中的解剖信息,为临床诊断提供量化、直观的参考,首先必须精确地对脑皮下组织区域进行分割。然而,由于大脑皮下组织形状多变、结构复杂以及灰度不均匀,不同病人不同模态图像之间差异信息大,使得核磁共振图像分割仍是一个值得深入研究的问题。
传统上对人脑的核磁共振图像进行分割从而确定大脑结构的外形特征和相对位置,主要由医生手工分割或采用标准脑图谱。前者主要基于医生的经验和知识,后者是过去十多年间mri图像分割的主流技术,主要基于标准脑图谱,如加拿大蒙特利尔神经研究所(montrealneurologicalinstitute,mni)于90年代表所建立的mni系列模板,以及更为广泛使用的脑成像国际联盟(internationalconsortiumforbrainmapping,icbm)公认的icbm152模板。然而,无论哪个模板,清晰度都差强人意。而且,模板还需转化成图谱。研究者发展了多种算法进行转换,最初的转换算法是将mni坐标转换成talairach坐标,再使用talairch大脑图谱进行生理结构的标定,另外一个典型算法是lancaster,j.l等于2007年发表的算法。这些转换算法都是非线性的,因此天然存在着计算误差。因此,目前还没有哪种标准图谱被认为是受到所有人认可的黄金标准图谱。
因此,研究者们还研究基于像素的核磁共振图像分割方法,主要采用特征提取、特征选择和分类器设计技术。其中特征提取按方法主要可分为模型方法、统计方法和信号处理方法。统计方法比较简单,易实现,对小图像具有一定的优势,但其对全局信息的利用率不高,与人类视觉模型脱节;模型方法能够注意到纹理局部的随机性和整体上的规律性,不足是模型系数难以求解,参数调节不方便;信号处理方法善于捕捉纹理的细节信息,能同时在空间和频域上体现纹理的特征,然而对高频信息处理不足,非规则纹理特征往往被忽视。mri的不同模态图像能提供不同的纹理边界信息,由于个体差异,不同病人同一模态所表现的信息也大不相同。由上可见,没有一种特征提取方法适合所有mri脑肿瘤的分割。
由yamlecun等人提出的卷积神经网络(convolutionalneuralnetworks,cnns)是有监督深度学习方法的一种,已在众多领域取得巨大的成功,如图像识别、语音识别、自然语言处理。cnns通过对原始输入数据进行循环卷积下采样,以有监督训练的方式获取卷积权重,最终实现直接从原始输入中提取有利于分类的特征,在图像识别中特征表现为纹理、形状、结构等。但是由于cnns需要多次卷积与下采样,计算量巨大,因此输入对象通常是一幅二维的图像。例如专利-一种基于深度学习的脑肿瘤自动分割方法,是对“mri切片进行图像块的提取”再用cnns方法进行训练。但是cnns在卷积和特征提取的迭代过程中往往失掉一些细节特征,只关注显著特征。这在人脸识别和语音识别等应用中并无大碍,但对于细节纹理丰富、变化多样的核磁共振图像来说,很多细腻特征将不被提取和识别。此外,由于核磁共振图像其实是三维数据,所谓的“mri切片”是人为将三维数据分割成二维图像,这样,二维cnns对三维图像的深度信息将不会提取和识别出来,导致最终提取出的脑部组织图像粗糙,尤其深度特征严重缺失。
技术实现要素:
有鉴于此,本发明提供了一种基于三维全卷积神经网络的核磁共振图像的分割方法,解决现有方法中脑皮下组织区域分割不清晰,过多冗余等技术问题,实现清晰准确的图像分割。
本发明通过以下技术手段解决上述问题:
一种基于三维全卷积神经网络的核磁共振图像的分割方法,包括如下步骤:
输入核磁共振图像数据;
设置三维全卷积神经网络结构,将脑部皮下组织的分割任务看作基于核磁共振图像的多特征多分类问题,提取脑部皮下组织的特征;
优化三维全卷积神经网络的参数;
将训练模型应用于核磁共振图像测试集进行测试,产生测试集的分类结果;
将测试集的分类结果画图产生三维脑皮下组织图。
进一步地,设置三维全卷积神经网络结构之前还包括:
预处理核磁共振图像数据。
进一步地,预处理核磁共振图像数据包括去除骨头部分和刚性配准。
进一步地,通过无监督的训练方法提取脑部皮下组织的特征。
进一步地,设置三维全卷积神经网络结构,将脑部皮下组织的分割任务看作基于核磁共振图像的多特征多分类问题,提取脑部皮下组织的特征具体包括:
构造的三维全卷积神经网络结构,包含3组共9层全卷积层,之后是3层全连接层,最后是1层分类层,共13层;
3组全卷积层包含3组,每组包含同样数目的三维卷积核,每层卷积核被应用在上一层的输入,卷积计算得到的结果是卷积核,假设ml是l层网络卷积核的数量,
f是非线性激活函数,使用parametricrectifiedlinearunit作为激活函数,其定义为:
f(xi)=max(0,xi)+αi·min(0,xi)
xi是输入信号,f(xi)代表输出信号,αi是当xi为负数时的倍数值;
全连接层用来建立特征和分类之间的模型,全连接层被设置为3层,每层被设计为一个1×1×1的卷积;
分类层含有m=c个特征映射,其中c为分类的数量,分类层l的输出被softmax函数归一化,因此,分类c∈{1,...,c}的概率为:
进一步地,采用随机梯度下降法和反向转播机制寻找最小化损失函数,优化三维全卷积神经网络网络的参数。
进一步地,采用stochasticgradientdescent和crossentropy方法来定义最小化损失函数:
其中
与现有技术相比,本发明的有益效果如下:
本发明基于三维全卷积神经网络的人脑核磁共振图像的分割方法,解决现有方法中脑皮下组织区域分割不清晰,过多冗余等技术问题,并解决一般二维卷积神经网络对核磁共振图像进行分割只能对切片进行分析,丢失深度信息导致的分割不清晰问题,从而对核磁共振图像实现清晰准确的图像分割。
附图说明
为了更清楚地说明本发明实施例中的技术方案,下面将对实施例描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
图1是本发明基于三维全卷积神经网络的核磁共振图像的分割方法的步骤图;
图2是本发明基于三维全卷积神经网络的核磁共振图像的分割方法的流程图;
图3是本发明三维全卷积神经网络的结构图。
具体实施方式
为使本发明的上述目的、特征和优点能够更加明显易懂,下面将结合附图和具体的实施例对本发明的技术方案进行详细说明。需要指出的是,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例,基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
如图1、图2所示,本发明提供了一种基于三维全卷积神经网络的核磁共振图像的分割方法,包括如下步骤:
1)、核磁共振图像数据输入和预处理。输入的核磁共振图像数据是三维的整个图像数据,而一般深度学习方法仅能处理二维图像。预处理步骤主要包括去除骨头部分和刚性配准。这两步不是必须的,不做处理会增大计算量,但不会影响最终结果。这两步预处理步骤可以通过多种软件实现,例如elastix[kleinet.al.,2010]。
2)、设置三维全卷积神经网络结构。图3示例给出了本发明所构造和使用的三维全卷积神经网络结构,包含3组共9层全卷积层,之后是3层全连接层,最后是1层分类层,共13层。
3组全卷积层包含3组,每组包含同样数目的三维卷积核。每层卷积核被应用在上一层的输入(在第一层时是输入的数据)。卷积计算得到的结果是卷积核。假设ml是l层网络卷积核的数量,
这里
这里f是非线性激活函数。这里我们使用parametricrectifiedlinearunit(prelu)作为激活函数,其定义为:
f(xi)=max(0,xi)+αi·min(0,xi)
这里xi是输入信号,f(xi)代表输出信号,αi是当xi为负数时的倍数值(一般设为0)。
全连接层用来建立特征和分类之间的模型,在一般的卷积神经网络当中被用来提取语义信息,然而在当前技术中为了保证空间信息被尽可能地被保留,全连接层被设置为3层,每层被设计为一个1×1×1的卷积。
分类层含有m=c个特征映射,其中c为分类的数量。分类层l的输出被softmax函数归一化,因此,分类c∈{1,...,c}的概率为:
3)、优化三维全卷积神经网络的参数。在步骤2)中构建了三维全卷积神经网络,将脑部皮下组织的分割任务看作基于核磁共振图像的多特征多分类问题,通过无监督的训练方法提取脑部皮下组织的特征。在这一步需要采用随机梯度下降法和反向转播机制寻找最小化损失函数,优化网络的参数。主要采用stochasticgradientdescent(sgd)和crossentropy(ce)方法来定义最小化损失函数:
其中
4)、将上述训练模型应用于核磁共振图像测试集进行测试,产生测试集的分类结果。
5)、将测试的分类结果画图产生三维脑皮下组织图。
本发明基于三维全卷积神经网络的人脑核磁共振图像的分割方法,解决现有方法中脑皮下组织区域分割不清晰,过多冗余等技术问题,并解决一般二维卷积神经网络对核磁共振图像进行分割只能对切片进行分析,丢失深度信息导致的分割不清晰问题,从而对核磁共振图像实现清晰准确的图像分割。
以上所述实施例仅表达了本发明的几种实施方式,其描述较为具体和详细,但并不能因此而理解为对本发明专利范围的限制。应当指出的是,对于本领域的普通技术人员来说,在不脱离本发明构思的前提下,还可以做出若干变形和改进,这些都属于本发明的保护范围。因此,本发明专利的保护范围应以所附权利要求为准。