一种卷积核生成方法和装置、电子设备

1.本发明涉及医学图像处理技术领域，尤其涉及一种卷积核生成方法和装置、电子设备。


背景技术：

2.管状结构在人体结构中无处不在，是维持人类生命体征的重要组织之一。近年来，心脑血管及气管等疾病已经成为全球的主要健康威胁之一。因此，人体管状结构的三维重建对于各类疾病的表征具有重要意义。
3.当前使用的卷积神经网络(convolutional neural network,cnn)在特征学习上具有强大性能，但其无法预测cnn的特征学习方式。除此之外，常见的cnn需要生成大量的通道数或要求具有足够的深度来得到足够的特征，使得网络训练过程中会耗费大量的时间获取冗余参数。复杂的网络结构会造成存储和计算资源压力。
4.目前为解决上述问题，主要采用的方式为：应对管状结构的分割提出轻量化网络模型，但这可能损伤网络的特征提取能力。大量公开研究已经证明了gabor滤波器在管状结构特征提取上的优势，但受限于参数调节的繁琐性。尽管已经有些研究实施了卷积核对于gabor滤波器的调制，但尚没有实现gabor滤波器的学习性。


技术实现要素：

5.本发明实施例的目的是提供一种卷积核生成方法和装置、电子设备，所生成的卷积核可应用于管状结构分割系统，主要解决常规卷积内核对于特征提取的不确定性问题，本发明实施例所生成的卷积核，能够更有目的性和针对性的实现管状结构的特征提取，所提取的特征更加可靠。
6.为解决上述技术问题，本发明提供如下技术方案：
7.一种卷积核生成方法，生成的所述卷积核应用于管状结构分割系统，其中，所述方法包括：
8.根据gabor滤波器原理实施三维gabor滤波器，并构建gabor卷积核；其中，所述gabor卷积核中下一层特征图的单通道结果，基于上一层卷积核的采样值生成；
9.基于反向传播原理，根据每一次迭代得到的损失值传递梯度，更新所述gabor卷积核的权重；
10.基于所述gabor卷积核的cnn模型结构，组合所述gabor卷积核构建的目标模型，其中，所述目标模型包括：特征映射、特征提取和特征融合三部分。
11.其中，根据gabor滤波器原理实施三维gabor滤波器，并构建gabor卷积核的步骤，包括：
12.建立c
in
个gabor卷积核对上层进行采样，并将每一个c
in
中得到的采样值求和，生成下一层特征图m
l+1
的单通道结果，其中，c
in
为上层特征图m
l
的通道数；
13.在第l层结构中，建立c
out
组gabor卷积核其中，每组gabor卷积核包含c
in
个gabor卷积核其中，c
out
为下一层特征图m
l+1
的通道数,生成包含c
out
通道数的下一层特征图m
l+1
。
14.其中，基于反向传播原理，根据每一次迭代得到的损失值传递梯度，更新所述gabor卷积核的权重的步骤，包括：
15.计算每一通道下在l函数下的梯度，并在学习率η下更新g
l
(ω,ψ,σ,θi)，且满足如下条件：
[0016][0017][0018]
其中，所述目标模型的特征映射部分由卷积层m1组成，所述卷积层m1由m0完成一次n通道的卷积并进行3d批规范化和relu激活处理后生成，其中m0为一次训练所选取的样本数建立的单通道特征图；
[0019]
所述特征提取部分引入了gabor卷积核在引入时对m1中的每一通道实施尺度和方向上的特征捕捉，并在通道数合并时将n组特征合并到m2的任一通道中；
[0020]
所述特征融合部分引入了一次卷积实现特征降维以及通道数的合并，使得特征图m
l
从特征空间转移到标记空间中。
[0021]
一种卷积核生成装置，生成的所述卷积核应用于管状结构分割系统，其中，所述装置包括：
[0022]
构建模块，用于根据gabor滤波器原理实施三维gabor滤波器，并构建gabor卷积核；其中，所述gabor卷积核中下一层特征图的单通道结果，基于上一层卷积核的采样值生成；
[0023]
更新模块，用于基于反向传播原理，根据每一次迭代得到的损失值传递梯度，更新所述gabor卷积核的权重；
[0024]
组合模块，用于基于所述gabor卷积核的cnn模型结构，组合所述gabor卷积核构建的目标模型，其中，所述目标模型包括：特征映射、特征提取和特征融合三部分。
[0025]
其中，所述构建模块包括：
[0026]
第一子模块，用于建立c
in
个gabor卷积核对上层进行采样，并将每一个c
in
中得到的采样值求和，生成下一层特征图m
l+1
的单通道结果，其中，c
in
为上层特征图m
l
的通道数；
[0027]
第二子模块，用于在第l层结构中，建立c
out
组gabor卷积核其中，每组gabor卷积核包含c
in
个gabor卷积核其中，c
out
为下一层特征图m
l+1
的通道数,生成包含c
out
通道数的下一层特征图m
l+1
。
[0028]
其中，所述更新模块具体用于：
[0029]
计算每一通道下在l函数下的梯度，并在学习率η下更新g
l
(ω,ψ,σ,θi)，且满足如下条件：
[0030][0031][0032]
其中，所述目标模型的特征映射部分由卷积层m1组成，所述卷积层m1由m0完成一次n通道的卷积并进行3d批规范化和relu激活处理后生成，其中m0为一次训练所选取的样本数建立的单通道特征图；所述特征提取部分引入了gabor卷积核在引入时对m1中的每一通道实施尺度和方向上的特征捕捉，并在通道数合并时将n组特征合并到m2的任一通道中；所述特征融合部分引入了一次卷积实现特征降维以及通道数的合并，使得特征图m
l
从特征空间转移到标记空间中。
[0033]
本发明实施例提供了一种电子设备，该电子设备包括处理器、存储器及存储在所述存储器上并可在所述处理器上运行的程序或指令，所述程序或指令被所述处理器执行时实现上述任意一种卷积核生成方法的步骤。
[0034]
本发明实施例提供了一种可读存储介质，所述可读存储介质上存储程序或指令，所述程序或指令被处理器执行时实现上述任意一种卷积核生成方法的步骤。
[0035]
本发明实施例提供的卷积核生成方法，根据gabor滤波器原理实施三维gabor滤波器，并构建gabor卷积核；基于反向传播原理，根据每一次迭代得到的损失值传递梯度，更新gabor卷积核的权重；基于gabor卷积核的cnn模型结构，组合gabor卷积核构建的目标模型，本发明实施例构建的用于管状结构分割的权重可学习gabor卷积核，可以直接从尺度和方向上增强管状结构的特征，减少cnn的冗余参数，并提取更准确的管状组织。
附图说明
[0036]
图1是表示本技术实施例的一种卷积核生成方法的步骤流程图；
[0037]
图2是表示本技术实施例的一种卷积核生成装置的结构框图；
[0038]
图3是表示本技术实施例的一种电子设备的结构框图。
具体实施方式
[0039]
为使本发明要解决的技术问题、技术方案和优点更加清楚，下面将结合附图及具体实施例进行详细描述。
[0040]
下面结合附图，通过具体的实施例及其应用场景对本技术实施例提供的卷积核生成方案进行详细地说明。
[0041]
本技术实施例生成的卷积核应用于管状结构分割系统，本技术实施例的卷积核生成方法包括以下步骤：
[0042]
步骤101：根据gabor滤波器原理实施三维gabor滤波器，并构建gabor卷积核。
[0043]
其中，gabor卷积核中下一层特征图的单通道结果，基于上一层卷积核的采样值生成。
[0044]
fourier变换是一种信号处理的有力工具，可以将图像从空域转换到频域，并提取到空域上不易提取到的特征。但是fourier变换缺乏时间和位置的局部信息。
[0045]
gabor变换是一种短时加窗fourier变换(简单理解起来就是在特定时间窗内做fourier变换)，是短时傅里叶变换中窗函数取为高斯函数时的一种特殊情况。因此，gabor滤波器可以在频域上不同尺度、不同方向上提取相关的特征。另外，gabor函数与人眼的作用相仿，所以经常用作纹理识别上，并取得了较好的效果。
[0046]
一种可选地构建gabor卷积核的方式可以包括如下流程：
[0047]
首先，建立c
in
个gabor卷积核对上层进行采样，并将每一个c
in
中得到的采样值求和，生成下一层特征图m
l+1
的单通道结果，其中，c
in
为上层特征图m
l
的通道数。表示第l层第n个特征图，ijk表示中心点的位置。l为层数。
[0048]
其次，在第l层结构中，建立c
out
组gabor卷积核其中，每组gabor卷积核包含c
in
个gabor卷积核其中，c
out
为下一层特征图m
l+1
的通道数,生成包含c
out
通道数的下一层特征图m
l+1
。
[0049]
步骤102：基于反向传播原理，根据每一次迭代得到的损失值传递梯度，更新gabor卷积核的权重。
[0050]
更新gabor卷积核的权重时，可通过如下方式实现：
[0051]
计算每一通道下在l函数下的梯度，并在学习率η下更新g
l
(ω,ψ,σ,θi)，且满足如下条件：
[0052][0053][0054]
其中g
l
(ω,ψ,σ,θi)为gabor滤波器函数，ω为正弦波角频，ψ为正弦波的相位偏移，它们确定了三维gabor滤波器中心频率的方向。σ是高斯函数的标准差，为高斯包络在空间信息中的宽度。θi表示角频ω的方向。
[0055]
步骤103：基于gabor卷积核的cnn模型结构，组合gabor卷积核构建的目标模型。
[0056]
其中，目标模型包括：特征映射、特征提取和特征融合三部分。
[0057]
目标模型的特征映射部分由卷积层m1组成，所述卷积层m1由m0完成一次n通道的卷积并进行3d批规范化和relu激活处理后生成，其中m0为一次训练所选取的样本数建立的单通道特征图。
[0058]
特征提取部分引入了gabor卷积核在引入时对m1中的每一通道实施尺度和方向上的特征捕捉，并在通道数合并时将n组特征合并到m2的任一通道中。
[0059]
特征融合部分引入了一次卷积实现特征降维以及通道数的合并，使得特征图m
l
从特征空间转移到标记空间中。
[0060]
本技术实施例提供的卷积核生成方法，第一方面，所创建的gabor卷积核不同于一
般的调制方法，gabor卷积核的参数可以在网络反向传播中自动学习；第二方面，gabor卷积核的结构与常规卷积核完全一致，可以添加到任意cnn的卷积核中；第三方面，针对gabor卷积核的特点，改进了一种基于gabor卷积核的cnn运用到管状结构分割，这种设计理念可以运用到目前任何以cnn为基础的网络结构中；第四方面，在网络的卷积过程中，由于gabor卷积核对于管状结构特征提取具有可控性，无需多余的通道数制造冗余的参数。因此，基于gabor卷积核的cnn的参数能大幅度减少，实现轻量级网络。
[0061]
图2为实现本技术实施例的一种卷积核生成装置的结构框图。
[0062]
本技术实施例的卷积核生成装置生成的所述卷积核应用于管状结构分割系统，本技术实施例的卷积核生成装置包括如下功能模块：
[0063]
构建模块201，用于根据gabor滤波器原理实施三维gabor滤波器，并构建gabor卷积核；其中，所述gabor卷积核中下一层特征图的单通道结果，基于上一层卷积核的采样值生成；
[0064]
更新模块202，用于基于反向传播原理，根据每一次迭代得到的损失值传递梯度，更新所述gabor卷积核的权重；
[0065]
组合模块203，用于基于所述gabor卷积核的cnn模型结构，组合所述gabor卷积核构建的目标模型，其中，所述目标模型包括：特征映射、特征提取和特征融合三部分。
[0066]
可选地，所述构建模块包括：
[0067]
第一子模块，用于建立c
in
个gabor卷积核对上层进行采样，并将每一个c
in
中得到的采样值求和，生成下一层特征图m
l+1
的单通道结果，其中，c
in
为上层特征图m
l
的通道数；
[0068]
第二子模块，用于在第l层结构中，建立c
out
组gabor卷积核其中，每组gabor卷积核包含c
in
个gabor卷积核其中，c
out
为下一层特征图m
l+1
的通道数,生成包含c
out
通道数的下一层特征图m
l+1
。
[0069]
可选地，所述更新模块具体用于：
[0070]
计算每一通道下在l函数下的梯度，并在学习率η下更新g
l
(ω,ψ,σ,θi)，且满足如下条件：
[0071][0072][0073]
可选地，所述目标模型的特征映射部分由卷积层m1组成，所述卷积层m1由m0完成一次n通道的卷积并进行3d批规范化和relu激活处理后生成，其中m0为一次训练所选取的样本数建立的单通道特征图；
[0074]
所述特征提取部分引入了gabor卷积核在引入时对m1中的每一通道实施尺度和方向上的特征捕捉，并在通道数合并时将n组特征合并到m2的任一通道中；
[0075]
所述特征融合部分引入了一次卷积实现特征降维以及通道数的合并，使得特征图ml
从特征空间转移到标记空间中。
[0076]
本技术实施例提供的卷积核生成装置，根据gabor滤波器原理实施三维gabor滤波器，并构建gabor卷积核；基于反向传播原理，根据每一次迭代得到的损失值传递梯度，更新gabor卷积核的权重；基于gabor卷积核的cnn模型结构，组合gabor卷积核构建的目标模型，本发明实施例构建的用于管状结构分割的权重可学习gabor卷积核，可以直接从尺度和方向上增强管状结构的特征，减少cnn的冗余参数，并提取更准确的管状组织。
[0077]
本技术实施例中图2所示的卷积核生成装置可以是装置，也可以是服务器中的部件、集成电路、或芯片。本技术实施例中的图2所示的卷积核生成装置可以为具有操作系统的装置。该操作系统可以为安卓(android)操作系统，可以为ios操作系统，还可以为其他可能的操作系统，本技术实施例不作具体限定。
[0078]
本技术实施例提供的图2所示的卷积核生成装置能够实现图1的方法实施例实现的各个过程，为避免重复，这里不再赘述。
[0079]
可选地，如图3所示，本技术实施例还提供一种电子设备400，包括处理器401，存储器402，存储在存储器402上并可在所述处理器401上运行的程序或指令，该程序或指令被处理器401执行时实现上述卷积核生成方法实施例的各个过程，且能达到相同的技术效果，为避免重复，这里不再赘述。
[0080]
需要注意的是，本技术实施例中的电子设备包括上述所述的服务器。
[0081]
本技术实施例还提供一种可读存储介质，所述可读存储介质上存储有程序或指令，该程序或指令被处理器执行时实现上述卷积核生成方法实施例的各个过程，且能达到相同的技术效果，为避免重复，这里不再赘述。
[0082]
其中，所述处理器为上述实施例中所述的电子设备中的处理器。所述可读存储介质，包括计算机可读存储介质，如计算机只读存储器(read-only memory，rom)、随机存取存储器(random access memory，ram)、磁碟或者光盘等。
[0083]
本技术实施例另提供了一种芯片，所述芯片包括处理器和通信接口，所述通信接口和所述处理器耦合，所述处理器用于运行程序或指令，实现上述卷积核生成方法实施例的各个过程，且能达到相同的技术效果，为避免重复，这里不再赘述。
[0084]
应理解，本技术实施例提到的芯片还可以称为系统级芯片、系统芯片、芯片系统或片上系统芯片等。
[0085]
需要说明的是，在本文中，术语“包括”、“包含”或者其任何其他变体意在涵盖非排他性的包含，从而使得包括一系列要素的过程、方法、物品或者装置不仅包括那些要素，而且还包括没有明确列出的其他要素，或者是还包括为这种过程、方法、物品或者装置所固有的要素。在没有更多限制的情况下，由语句“包括一个
……”
限定的要素，并不排除在包括该要素的过程、方法、物品或者装置中还存在另外的相同要素。
[0086]
以上所述是本发明的优选实施方式，应当指出，对于本技术领域的普通技术人员来说，在不脱离本发明所述原理的前提下，还可以做出若干改进和润饰，这些改进和润饰也应视为本发明的保护范围。