CN111489364B - 基于轻量级全卷积神经网络的医学图像分割方法 - Google Patents

Abstract

本发明请求保护一种基于轻量级全卷积神经网络的医学图像分割方法。首先对数据集进行灰度化、归一化、对比度受限自适应直方图均衡(CLAHE)、伽马校正等预处理；然后，对训练集进行随机的提取patch和测试集顺序提取patch图以完成数据增强；接着，搭建由收缩路径(左侧)和扩张路径(右侧)组成的全卷积神经网络架构，针对图像数量较少的数据集设计留一法(leave‑one‑out)训练方法；最后，通过通道稀疏正则化训练、裁剪比例因子小于设定阈值的通道以及微调裁剪后的网络完成BN通道模型裁剪，得到轻量级全卷积神经网络，并将测试数据输入到该网络中快速测试完成图像分割。轻量级全卷积神经网络即保证了深度网络的高分割精度优势，又提高了图像分割网络的测试速度。

Description

基于轻量级全卷积神经网络的医学图像分割方法

技术领域

本发明属于图像特征提取与分割方法技术领域，特别是针对轻量级全卷积神经网络的医学图像语义分割方法。

背景技术

全卷积神经网络(FCN)已经在图像分割尤其是语义分割领域占据了主导地位，由于其具有可以接受任意大小的输入图像、运算速率高效等特点，并能取得像素级别的图像分割效果。

当前对于全卷积神经网络已经做了大量的实验，自2015年FCN起，产生了一批经典的模型，如UNet及UNet-like等结构，这些模型基于FCN不断的加深网络层数以提高网络的分割精度的。但也因此带来了一系列问题，如网络训练环境需求高，运算速度慢，参数过多等以至于难以将模型落实到实际应用中。

为将深度网络应用到实际中，近几年，学者们轻量级的卷积神经网络做了大量的研究，也因此产生了不少优秀的网络结构，如MobileNet，XceptionNet等，它们巧妙的结果设计，使深度网络在移动设备上的应用成为可能。轻量级网络的诞生，进一步推动了业界对模型压缩算法的关注，如量化权重、结构稀疏化、模型裁剪等。这些模型压缩算法都在一定程度上压缩模型的空间体积使得网络更加的轻量化。其在卷积神经网络中应用广泛，但针对全卷积神经网络的模型裁剪研究应用还是比较少。而全卷积神经网络，比如Unet，在医学图像分割中有着卓越的优势。因此，如果能够针对全卷积网络结构设计一种模型裁剪方法，使得全卷积网络更加的轻量化，那么将使全卷积网络在医学图像分割领域中的进一步应用成为可能。

发明内容

本发明旨在解决以上现有技术的问题。提出了一种通过BN通道裁剪全卷积神经网络以使模型更轻量化的基于轻量级全卷积神经网络的医学图像分割方法。本发明的技术方案如下：

一种基于轻量级全卷积神经网络的医学图像分割方法，其包括以下步骤：

步骤1：对医学图像数据集进行灰度化、归一化、对比度受限自适应直方图均衡、伽马校正在内的预处理；

步骤2：针对训练集随机提取patch图，首先设定patch图的宽和高，然后随机地选取像素点作为patch的中心；

步骤3：搭建由左侧收缩路径和右侧扩张路径组成的全卷积神经网络；

步骤4：针对步骤3中搭建好的全卷积神经网络，使用dice_loss计算预测值y和期望值y'的相似度，其中dice_loss的定义如下：

步骤5：结合步骤4中的损失函数，针对图像数量较少的数据集使用留一法(leave-one-out)进行交叉验证；

步骤6：加载步骤5中得到的预训练网络模型，通过通道稀疏正则化训练、裁剪比例因子小于设定阈值的通道，以及微调裁剪后的网络完成BN通道模型裁剪，得到轻量级全卷积神经网络；最后，加载步骤1中的测试图像数据到该网络中，快速测试完成图像分割。

进一步的，所述步骤1对医学图像数据集进行灰度化、归一化、对比度受限自适应直方图均衡、伽马校正在内的预处理，具体为：首先采用的是对R、G、B分量进行加权平均的算法，其表达式如下，其中Gray(i,j)表示图像上的像素点，i表示像素点的横坐标，j表示像素点的纵坐标：

Gray(i,j)＝0.299*R(i,j)+0.578*G(i,j)+0.114*B(i,j)

然后采用z-score标准化方法，对上述灰度化后的图像进行标准化并缩放图像数据，表达为：

其中,μ表示的是每一维度像素点自身的均值，σ表示的是每一维度像素点自身的均方差；

接着，对标准化后的图像，在保证能够将直方图分布变成近似均匀分布的前提下，设定一个限制直方图分布的阈值δ以限制性的增强图像对比度，同时，在像素的映射过程中使用双线性插值加快直方图均衡化的速度；然后继续对对比度受限自适应直方图均衡(CLAHE)后的图像V_in的伽玛曲线进行非线性色调编辑，识别图像信号中的前景和背景部分，并增大两者的比例；最后把图片上的像素点值归一化映射到0-1之间。

进一步的，所述步骤2针对训练集随机提取patch图，首先设定patch图的宽和高，然后随机地选取像素点作为patch的中心，具体为：

针对训练集，首先设定patch图的宽、高大小分别为patch_w和patch_h，然后随机地选取像素点作为patch的中心，该像素点的坐标需满足如下条件：

其中，x为横坐标，y为纵坐标，img_w是原图的宽度大小，img_h是原图的高度大小；最后，以该像素点为中心，从原图中截取宽度为patch_w，高度为patch_h的训练图patch，针对测试集，同样先设定patch_w和patch_h，然后判断原图的大小是否和设定的patch图的大小成比例关系，如果没有，则对原图进行像素填充；最后，以输入原图的左上角为起点，按照“从上到下从左到右”原则循环提取patch图。

进一步的，所述步骤3搭建由左侧收缩路径和右侧扩张路径组成的全卷积神经网络架构，具体为：

收缩路径遵循典型的卷积网络模式，每个阶段均由两个3×3卷积模块进行特征提取，步长为2的下采样操作跟在卷积之后，在每次下采样之后，特征图尺寸的大小减少一倍，特征通道的数量增加一倍；在扩展路径中，在卷积模块后使用双线性插值上采样操作，其步长也为2，在每次上采样之后跳跃连接收缩路径每阶段的输出；在最后一层，使用1x1卷积代替全连接层将每个64分量特征向量映射到所需的类数。

进一步的，所述步骤4针对步骤3中搭建好的网络，使用dice_loss评估预测值y和期望值y'的相似度，其中dice_loss的定义如下：

具体为：

其首先计算手工分割图y和预测图y'之间的交集y∩y'，近似为手工分割图和预测图之间的点乘y·y'，|y|+|y'|表示的是逐元素相加。

进一步的，所述步骤5结合步骤4中的损失函数，针对图像数量较少的数据集使用留一法(leave-one-out)进行交叉验证；具体为：在面对数量较小的数据集时，留一法在划分数据集的时候仅留一验证，即数据集S的大小为N，那么划分N-1条数据为训练集，剩下的一条为验证集，在训练过程中循环遍历数据集N次，直到所有的数据都作过验证集，最后，对验证误差求平均。

进一步的，所述步骤6加载步骤5中得到的预训练网络模型，通过通道稀疏正则化训练、裁剪比例因子小于设定阈值的通道以及微调裁剪后的网络完成BN通道模型裁剪，得到轻量级全卷积神经网络，具体为：

首先引进比例因子γ，比例因子通过和通道的输出进行相乘操作使得每个通道获得一个重要权重值w，在训练的过程中需要执行L1正则化操作，具体表达为：

其中，L表示目标函数，f(x,W)表示的是预测图，(x,y)表示输入原图和其对应的手工分割图，W表示训练过程中的网络权重，l(·)代表的是训练过程中使用的损失函数，g(·)是稀疏性对比例因子γ的惩罚，参数λ保证网络的平衡性。其次针对比例因子γ，将批标准化BN层中为保证输入数据的表达能力而引进的可学习参数γ视作网络裁剪的比例因子。

进一步的，批标准化BN层的原理可用公式表达为：

其中x为输入图像，μ为输入像素点的均值，σ为输入像素点的方差，γ和β为可训练参数；最后进行网络通道裁剪，遍历每个批标准化BN层对应通道[c₁,c₂,c₃,...c_n]上的比例因子γ，根据γ对网络通道进行裁剪，其遵循的原则是，如果γ接近于0，那么可认为该通道对网络的贡献小，将其裁剪对网络训练效果影响不大，得到轻量级全卷积神经网络，输入待测试图像数据到该轻量级网络完成图像分割。

本发明的优点及有益效果如下：

本发明利用BN通道裁剪的思想对全卷积神经网络结构进行轻量化，通过对模型进行压缩，减少了模型的尺寸大小。相比原始的结构，一方面，能有效裁剪网络层之间冗余的通道数量，从而有效的减少了模型参数，提升分割网络的运算质量。另一方面，由于裁剪了无效的通道，避免了某些无用特征信息带来的噪声干扰，也在一定程度保证了图像分割效果的质量。

本发明针对图像数量较少的数据集采用留一法交叉验证，使得模型训练几乎是使用所有的样本图像，这样评估所得的结果最接近样本，比其他的交叉验证方法得到的模型更加的可靠。除此之外，留一法还有随机性小，可重复性强等优点。

本发明的其他优点还有通过区分训练集和测试集的提取patch方法，既保证了数据能够得到充分的训练，又确保了网络输出的可视效果。另外，经过BN通道裁剪后的轻量化全卷积神经网络使得该模型在医学图像分割应用中的进一步推广成为可能。

附图说明

图1是本发明提供优选实施例基于轻量级全卷积神经网络的医学图像分割方法流程图。

图2为全卷积神经网络结构的原始架构图。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、详细地描述。所描述的实施例仅仅是本发明的一部分实施例。

本发明解决上述技术问题的技术方案是：

一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，包括以下步骤：

步骤1：对数据集进行灰度化、归一化、对比度受限自适应直方图均衡(CLAHE)、伽马校正(Gamma adjustment)等预处理，以提高医学图像的分割质量；

步骤2：由于步骤1中预处理完的数据数量过少，为有效地避免深度网络训练时过拟合陷入局部最优解等，本方法通过对训练集进行随机的提取patch和测试集顺序提取patch图来增大数据集的数量；

步骤3：搭建由收缩路径(左侧)和扩张路径(右侧)组成的全卷积神经网络架构；

步骤4：针对步骤3中搭建好的网络，使用dice_loss评估预测值y和期望值y'的相似度，以更好的评估网络的分割效果，其中dice_loss的定义如下：

步骤5：结合步骤4中的损失函数，针对图像数量较少的数据集使用留一法(leave-one-out)进行交叉验证；

步骤6：加载步骤5中得到的预训练网络模型，通过通道稀疏正则化训练、裁剪比例因子小于设定阈值的通道以及微调裁剪后的网络完成BN通道模型裁剪，得到轻量级全卷积神经网络。最后，加载步骤1中的测试图像数据到该网络中，快速测试完成图像分割。

进一步的，所述步骤1：对数据集进行灰度化、归一化、对比度受限自适应直方图均衡(CLAHE)、伽马校正(Gamma adjustment)等预处理，以提高医学图像的分割质量。具体为：首先采用的是对R、G、B分量进行加权平均的算法，其表达式如下，其中Gray(i,j)表示图像上的像素点，i表示像素点的横坐标，j表示像素点的纵坐标：

Gray(i,j)＝0.299*R(i,j)+0.578*G(i,j)+0.114*B(i,j)

Gray(i,j)在保证输入图V_RGB亮度能够保留的同时，消除了图像色调和饱和度信息。经过灰度化的图片可以减少后续图像计算的数量并完全反映整个图像。然后采用z-score标准化方法，对上述灰度化后的图像进行标准化并缩放图像数据，以便度量标准可以参与评估计算。该方法可表达为：

其中,μ表示的是每一维度像素点自身的均值，σ表示的是每一维度像素点自身的均方差。采取这样的操作可以保证图像特征分布均匀，从而提高收敛速度避免局部最优解。接着，对标准化后的图像，在保证能够将直方图分布变成近似均匀分布的前提下，设定一个限制直方图分布的阈值δ以限制性的增强图像对比度，从而避免过度增强放大的噪声并消除块边缘过渡的不平衡。同时，在像素的映射过程中使用双线性插值加快直方图均衡化的速度。然后继续对对比度受限自适应直方图均衡(CLAHE)后的图像V_in的伽玛曲线进行非线性色调编辑，识别图像信号中的前景和背景部分，并增大两者的比例，从而提高图像的对比度。最后为避免网络训练时loss值出现负值、过大、nan等现象，把图片上的像素点值归一化映射到0-1之间。

进一步的，所述步骤2：由于步骤1中预处理完的数据数量过少，为有效地避免深度网络训练时过拟合陷入局部最优解等，本方法通过对训练集进行随机的提取patch和测试集顺序提取patch图来增大数据集的数量。其具体为：

针对训练集，首先设定patch图的宽高大小patch_w和patch_h，然后随机地选取像素点作为patch的中心，该像素点的坐标需满足如下条件：

其中，x为横坐标，y为纵坐标，img_w是原图的宽度大小，img_h是原图的高度大小。最后，以该像素点为中心，从原图中截取宽度为patch_w，高度为patch_h的训练图patch。针对测试集，同样先设定patch_w和patch_h。然后判断原图的大小是否和设定的patch图的大小成比例关系，如果没有，则对原图进行像素填充。最后，以输入原图的左上角为起点，按照“从上到下从左到右”原则循环提取patch图。测试集的做法是为了在训练结果中能够还原图片。

进一步的，所述步骤3：搭建由收缩路径(左侧)和扩张路径(右侧)组成的全卷积神经网络架构。具体为：

收缩路径遵循典型的卷积网络模式，每个阶段均由两个3×3卷积模块进行特征提取，步长为2的下采样操作跟在卷积之后。在每次下采样之后，特征图尺寸的大小减少一倍，特征通道的数量增加一倍。在扩展路径中，与收缩路径相同的是，每个步骤都对特征图进行卷积。不同的是，为了能够还原特征图实现端到端图像分割任务，在卷积模块后使用双线性插值上采样操作，其步长也为2。同时，为了能够结合低语义特征图中的精确位置信息和高语义特征图中的特征，在每次上采样之后跳跃连接收缩路径每阶段的输出。在最后一层，使用1x1卷积代替全连接层将每个64分量特征向量映射到所需的类数，这样做既保证了网络能够实现分类任务又极大的降低了网络的参数计算量。

进一步的，所述步骤4：针对步骤3中搭建好的网络，使用dice_loss评估预测值y和期望值y'的相似度，以更好的评估网络的分割效果。其中dice_loss的定义如下：

具体为：

首先计算手工分割图y和预测图y'之间的交集y∩y'，其通常近似为手工分割图和预测图之间的点乘y·y'，|y|+|y'|表示的是逐元素相加。对于二分类而言，由于其预测图上的值只有0和1，在进行y·y'运算时可以有效的对手工分割图中未被激活的像素清零，而对那些激活的像素，通过点乘可以获得相应的置信度，在训练过程中，网络会对置信度低的点进行严厉的惩罚，而给予高置信度像素点较高的dice。

进一步的，所述步骤5：结合步骤4中的损失函数，针对图像数量较少的数据集使用留一法(leave-one-out)进行交叉验证。具体为：与一般的训练方法将数据集按照七三分划分训练集和测试集不同，在面对图像数量较小的数据集时，留一法在划分数据集的时候仅留一验证。即数据集S的大小为N，那么划分N-1条数据为训练集，剩下的一条为验证集。在训练过程中循环遍历数据集N次，直到所有的数据都作过验证集，最后，对验证误差求平均。

进一步的，所述步骤6：加载步骤5中得到的预训练网络模型，通过通道稀疏正则化训练、裁剪比例因子小于设定阈值的通道以及微调裁剪后的网络完成BN通道模型裁剪，得到轻量级全卷积神经网络。最后，加载步骤1中的测试图像数据到该网络中，快速测试完成图像分割。具体为：

首先引进比例因子γ，比例因子通过和通道的输出进行相乘操作使得每个通道获得一个重要权重值w，该权重值越大表示通道的重要性越强，反之则代表该通道为无效通道。γ和w均可在网络训练中不断更新。同时，为了防止网络过拟合，在训练的过程中需要在其后执行L1正则化操作，具体可表达为：

其中，L表示目标函数，f(x,W)表示的是预测图，(x,y)表示输入原图和其对应的手工分割图。W表示训练过程中的网络权重，l(·)代表的是训练过程中使用的损失函数Entropy Cross。g(·)是稀疏性对比例因子γ的惩罚，参数λ保证网络的平衡性。其次针对比例因子γ，为了保证它的唯一性和可学习性，本方法将批标准化BN层中为保证输入数据的表达能力而引进的可学习参数γ视作网络裁剪的比例因子。批标准化BN层的原理可用公式表达为：

其中x为输入图像，μ为输入像素点的均值，σ为输入像素点的方差，γ和β为可训练参数。最后进行网络通道裁剪，遍历每个批标准化BN层对应通道[c₁,c₂,c₃,...c_n]上的比例因子γ，根据γ对网络通道进行裁剪。其遵循的原则是，如果γ接近于0，那么可认为该通道对网络的贡献小，将其裁剪对网络训练效果影响不大。对完成通道裁剪后的轻量级网络步骤。

具体的，优选实施例为：

如图1所示，本实施例提供的轻量级全卷积神经网络的医学图像分割方法，包括以下步骤：

步骤1：首先对图像的R、G、B分量进行加权平均的算法；然后采用z-score标准化方法，对上述灰度化后的图像进行标准化并缩放图像数据；接着，对标准化后的图像，在保证能够将直方图分布变成近似均匀分布的前提下，设定一个限制直方图分布的阈值δ以限制性的增强图像对比度，从而避免过度增强放大的噪声并消除块边缘过渡的不平衡。同时，在像素的映射过程中使用双线性插值加快直方图均衡化的速度。接着继续对图像的伽玛曲线进行非线性色调编辑，识别图像信号中的前景和背景部分，并增大两者的比例，从而提高图像的对比度。最后为避免网络训练时loss值出现负值、过大、nan等现象，把图片上的像素点值归一化映射到0-1之间。其中，加权平均算法的公式为：

Gray(i,j)＝0.299*R(i,j)+0.578*G(i,j)+0.114*B(i,j) (1)

z-score的公式：

步骤2：针对训练集随机提取patch图，首先设定patch图的宽高，然后随机地选取像素点作为patch的中心，该像素点的坐标需满足如下条件：

其中，x为横坐标，y为纵坐标，img_w是原图的宽度大小，img_h是原图的高度大小。最后，以该像素点为中心，从原图中截取patch。针对测试集顺序提取patch图，同样先设定宽高。然后判断原图的大小是否和设定的patch图的大小成比例关系，如果没有，则对原图进行像素填充。最后，以输入原图的左上角为起点，按照“从上到下从左到右”原则循环提取patch图。

步骤3：搭建由收缩路径(左侧)和扩张路径(右侧)组成的全卷积神经网络架构。具体为：

如图2所示，收缩路径中每个阶段的特征提取任务，均由两个3×3卷积模块完成，步长为2的下采样操作则负责降低图像的尺寸扩大卷积感受野。在每次下采样之后，特征图尺寸的大小减少半，特征通道的数量增倍。在扩展路径中，使用双线性插值上采样操作，还原特征图以实现端到端图像分割任务。同时，通过跳跃连接机制结合不同图像尺寸大小的空间信息、语义特征信息。在最后一层，使用1×1卷积代替全连接层进行分类。

步骤4：为更好的评估网络的分割任务，针对步骤3构建的全卷积网络，采用dice_loss作为网络的评估指标，其中dice_loss的定义如下：

步骤5：加载步骤3构建的网络，针对图像数量较少的数据集使用留一法(leave-one-out)进行交叉验证；

步骤6：加载步骤5中得到的预训练网络模型，首先引进比例因子γ，将比例因子和通道的输出相乘操作得到通道权重w。γ和w均可在网络训练中不断更新。并且在训练的过程中加入L1正则化操作以防止网络过拟合，具体可表达为：

其中，L表示目标函数，f(x,W)表示的是预测图，(x,y)表示输入原图和其对应的手工分割图。W表示训练过程中的网络权重，l(·)代表的是训练过程中使用的损失函数Entropy Cross。g(·)是稀疏性对比例因子γ的惩罚，参数λ保证网络的平衡性。其次，由公式(5)可知，批标准化BN层引入了γ和β两个可学习参数以保证输入数据的表达能力，因此为了保证比例因子的唯一性和可训练性，本方法将γ视作网络裁剪的比例因子。

对于公式(5)，x为输入图像，μ为输入像素点的均值，σ为输入像素点的方差，γ和β为可训练参数。

最后遍历每个批标准化BN层对应通道上的比例因子γ，如果γ接近于0，那么可认为该通道对网络的贡献小，将其裁剪，得到轻量级全卷积神经网络。输入待测试图像数据到该轻量级网络可以更快速高效地完成图像分割。

以上这些实施例应理解为仅用于说明本发明而不用于限制本发明的保护范围。在阅读了本发明的记载的内容之后，技术人员可以对本发明作各种改动或修改，这些等效变化和修饰同样落入本发明权利要求所限定的范围。

Claims (8)

1.一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，包括以下步骤：

步骤1：对医学图像数据集进行灰度化、归一化、对比度受限自适应直方图均衡、伽马校正在内的预处理；

步骤2：针对训练集随机提取patch图，首先设定patch图的宽和高，然后随机地选取像素点作为patch的中心；

步骤3：搭建由左侧收缩路径和右侧扩张路径组成的全卷积神经网络；

步骤4：针对步骤3中搭建好的全卷积神经网络，使用dice_loss计算预测值y和期望值y'的相似度，其中dice_loss的定义如下：

步骤5：结合步骤4中的损失函数，针对图像数量较少的数据集使用留一法进行交叉验证；

步骤6：加载步骤5中得到的预训练网络模型，通过通道稀疏正则化训练、裁剪比例因子小于设定阈值的通道，以及微调裁剪后的网络完成BN通道模型裁剪，得到轻量级全卷积神经网络；最后，加载步骤1中的测试图像数据到该网络中，快速测试完成图像分割。

2.根据权利要求1所述的一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，所述步骤1对医学图像数据集进行灰度化、归一化、对比度受限自适应直方图均衡、伽马校正在内的预处理，具体为：首先采用的是对R、G、B分量进行加权平均的算法，其表达式为如下，其中Gray(i,j)表示图像上的像素点，i表示像素点的横坐标，j表示像素点的纵坐标：

Gray(i,j)＝0.299*R(i,j)+0.578*G(i,j)+0.114*B(i,j)

然后采用z-score标准化方法，对上述灰度化后的图像进行标准化并缩放图像数据，表达为：

其中,μ表示的是每一维度像素点自身的均值，σ表示的是每一维度像素点自身的均方差；

接着，对标准化后的图像，在保证能够将直方图分布变成近似均匀分布的前提下，设定一个限制直方图分布的阈值δ以限制性的增强图像对比度，同时，在像素的映射过程中使用双线性插值加快直方图均衡化的速度；然后继续对对比度受限自适应直方图均衡(CLAHE)后的图像V_in的伽玛曲线进行非线性色调编辑，识别图像信号中的前景和背景部分，并增大两者的比例；最后把图片上的像素点值归一化映射到0-1之间。

3.根据权利要求1所述的一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，所述步骤2针对训练集随机提取patch图，首先设定patch图的宽和高，然后随机地选取像素点作为patch的中心，具体为：

针对训练集，首先设定patch图的宽、高大小分别为patch_w和patch_h，然后随机地选取像素点作为patch的中心，该像素点的坐标需满足如下条件：

其中，x为横坐标，y为纵坐标，img_w是原图的宽度大小，img_h是原图的高度大小；最后，以该像素点为中心，从原图中截取宽度为patch_w，高度为patch_h的训练图patch，针对测试集，同样先设定patch_w和patch_h，然后判断原图的大小是否和设定的patch图的大小成比例关系，如果没有，则对原图进行像素填充；最后，以输入原图的左上角为起点，按照“从上到下从左到右”原则循环提取patch图。

4.根据权利要求1所述的一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，所述步骤3搭建由左侧收缩路径和右侧扩张路径组成的全卷积神经网络架构，具体为：

收缩路径遵循典型的卷积网络模式，每个阶段均由两个3×3卷积模块进行特征提取，步长为2的下采样操作跟在卷积之后，在每次下采样之后，特征图尺寸的大小减少一倍，特征通道的数量增加一倍；在扩展路径中，在卷积模块后使用双线性插值上采样操作，其步长也为2，在每次上采样之后跳跃连接收缩路径每阶段的输出；在最后一层，使用1x1卷积代替全连接层将每个64分量特征向量映射到所需的类数。

5.根据权利要求1所述的一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，所述步骤4针对步骤3中搭建好的网络，使用dice_loss评估预测值y和期望值y'的相似度，其中dice_loss的定义如下：

具体为：其首先计算手工分割图y和预测图y'之间的交集y∩y'，近似为手工分割图和预测图之间的点乘y·y'，|y|+|y'|表示的是逐元素相加。

6.根据权利要求1所述的一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，所述步骤5：结合步骤4中的损失函数，针对图像数量较少的数据集使用留一法进行交叉验证；

具体为：在面对数量较小的数据集时，留一法在划分数据集的时候仅留一验证，即数据集S的大小为N，那么划分N-1条数据为训练集，剩下的一条为验证集，在训练过程中循环遍历数据集N次，直到所有的数据都作过验证集，最后，对验证误差求平均。

7.根据权利要求1所述的一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，所述步骤6加载步骤5中得到的预训练网络模型，通过通道稀疏正则化训练、裁剪比例因子小于设定阈值的通道以及微调裁剪后的网络完成BN通道模型裁剪，得到轻量级全卷积神经网络，具体为：

首先引进比例因子γ，比例因子通过和通道的输出进行相乘操作使得每个通道获得一个重要权重值w，在训练的过程中需要执行L1正则化操作，具体表达为：

其中，L表示目标函数，f(x,W)表示的是预测图，(x,y)表示输入原图和其对应的手工分割图，W表示训练过程中的网络权重，l(·)代表的是训练过程中使用的损失函数，g(·)是稀疏性对比例因子γ的惩罚，参数λ保证网络的平衡性； 其次针对比例因子γ，将批标准化BN层中为保证输入数据的表达能力而引进的可学习参数γ视作网络裁剪的比例因子。

8.根据权利要求1所述的一种基于轻量级全卷积神经网络的医学图像分割方法，其特征在于，批标准化BN层的原理可用公式表达为：

其中x为输入图像，μ为输入像素点的均值，σ为输入像素点的方差，γ和β为可训练参数；最后进行网络通道裁剪，遍历每个批标准化BN层对应通道[c₁,c₂,c₃,...c_n]上的比例因子γ，根据γ对网络通道进行裁剪，其遵循的原则是，如果γ接近于0，那么可认为该通道对网络的贡献小，将其裁剪对网络训练效果影响不大，得到轻量级全卷积神经网络，输入待测试图像数据到该轻量级网络完成图像分割。

Images