基于梯度正则卷积神经网络的低剂量CT图像去噪方法与流程

文档序号:15350392发布日期:2018-09-04 23:15阅读:209来源:国知局

本发明属于图像处理技术领域,特别涉及一种ct图像的去噪方法,可用于ct设备成像时辐射量为传统ct成像的1/6到1/4的低剂量ct图像进行增强,提高图像质量和成像器官、组织的可视效果。



背景技术:

传统的ct成像由于使用了较高的x射线辐射,根据成像器官或组织的不同,一般为1毫希弗到12毫希弗,对病人具有一定的辐射伤害,而且辐射可能导致的癌变和基因突变也引起了人们的担忧,因此降低成像时x射线剂量,一般为传统ct成像使用剂量的1/6到1/4,进行低剂量ct成像成为了一种新的选择。但是,这种做法会导致成像的结果质量很差,图像中会有大量的噪声和伪影,严重影响了医生对病变区域和重要器官、组织的诊断和分析。所以,急需一种有效的方法以提高低剂量ct图像的质量。

根据处理数据类型的不同,低剂量ct图像去噪主要有两种方式。其一,是对ct成像过程中成像光子的正弦信号进行预处理,滤除噪声,然后再使用反向投影算法进行成像。这种方法利用了ct成像过程中噪声的生成机制,即成像光子在统计上服从复合泊松分布,代表算法包括双边滤波、惩罚加权最小二乘法、静态小波变换和最大似然估计等等。这类算法简单高效,但对原始信号的依赖较大,空间分辨率不高。其二,是对反向投影算法重建后的图像进行处理,直接在图像域滤除噪声,对原始信号依赖较低,但无法有效地确定噪声的分布。常见方法包括小波分解、非局部均值bm3d、稀疏表示与字典学习ksvd等等。

近年来,深度学习在图像处理和分析领域掀起了一阵热浪,从低层次的图像去噪、超分辨任务到高层次的图像检测与分类任务都有涉及。这种类脑方式的信息处理机制模仿了人类视觉神经系统,对图像处理和分析十分有效,已经有一些研究工作将深度学习方法应用到了图像去噪方面,取得了不错的结果。在低剂量ct图像去噪的应用上,kang等人首先对低剂量ct图像进行小波分解,利用得到的小波系数作为输入,然后通过卷积神经网络去除噪声,重建ct图像。chen等人提出了一个对称的卷积和转置卷积神经网络,先对输入图像进行特征提取和编码,再经过网络对编码信号进行解码重建图像,在编解码的过程中滤除噪声。wolterink等人率先使用了生成对抗网络进行去噪,通过生成网络来模拟低剂量ct图像的噪声生成机制,结果发现使用卷积神经网络对低剂量ct图像去噪是可行的,使用生成对抗策略可以生成更接近于全剂量的ct图像。但这类算法在重建图像的过程中,仅仅考虑了对图像灰度信息的恢复,没有过多地考虑图像的其他信息,很容易导致恢复的图像边缘模糊,细节损失严重。



技术实现要素:

本发明的目的在于针对上述深度学习在图像重建过程中的不足,提出了一种基于梯度正则卷积神经网络的低剂量ct图像去噪方法,以保留更完整的图像边缘和细节,提高恢复图像的质量。

为实现上述目的,本发明在图像恢复的过程中,同时保持灰度信息的和梯度信息,实现恢复图像细节的保留,减小模糊边缘和损失细节的现象,其实现方案包括如下:

(1)使用ct设备同时对人体同一部位进行全剂量和低剂量成像,得到多张成对的低剂量ct图像x和全剂量ct图像y,记为{x,y},其中,全剂量ct成像时x射线管电压为120千伏,管电流为200毫安,辐射剂量约为3毫希弗;低剂量ct图像时x射线管电压为120千伏,管电流为50毫安,辐射剂量为0.75毫希弗;采集的ct图像数据集记为d;

(2)对采集的ct图像数据集d进行扩充,得到扩充ct图像数据集da;

(3)对da中每张低剂量ct和全剂量ct图像进行归一化操作,将灰度值映射到区间[0,1];

(4)对数据集da中每对ct图像进行取块,建立图像块数据集dp;

(5)根据得到的图像块数据集dp,搭建一个全卷积去噪神经网络n,网络层数通过交叉验证设为12,网络的输入为低剂量ct图像,输出为去噪后的ct图像,使用msra方法初始化网络的权值w,网络的偏置b均初始化为数值0;

(6)使用图像块数据集dp,训练全卷积去噪网络n:

(6a)打乱dp中图像块对的顺序,依次从dp中选择256对ct图像块数据,记为一个训练批次{xpb,ypb},其中低剂量图像块xpb作为网络的输入,全剂量图像块ypb用来计算网络的损失;

(6b)将xpb输入到网络n中,经过前向传播,得到网络的输出apb,计算apb和ypb之间的均方误差,得到网络n的灰度损失lv;

(6c)使用梯度算子g分别卷积apb和ypb,得到对应的梯度图ga和gy,计算ga和gy之间的均方误差,得到网络n的梯度损失lg;

(6d)对网络n的灰度损失lv和梯度损失lg进行线性加权,得到网络的整体损失l=lv+λ×lg,其中λ为加权系数;

(6e)使用带动量项的批量梯度下降算法优化整体损失l,初始学习速率为0.0001,动量项参数为0.9,权值衰减因子为0.0001,对网络的权值w和偏置b进行更新;

(6f)重复以上步骤(6a)到(6e),直到达到网络训练的最大迭代次数t=100000,得到训练好的去噪网络nt;

(7)将一张完整的低剂量ct图像xt输入到训练好的去噪网络nt中,经过该去噪网络nt的前向传播,得到去噪后的ct图像at。

本发明与现有技术相比具有以下优点:

1、本发明在图像恢复的过程中既考虑了图像灰度信息的重建,也考虑了图像梯度的保持,使得网络既有好的去噪效果,也能保留图像的大部分细节。

2、本发明是直接在图像域进行去噪,独立于具体的噪声分布和ct仪器参数,应用范围更广泛。

附图说明

图1是本发明的实现总流程图;

图2是本发明中所使用的低剂量ct图像;

图3是本发明中所使用的全剂量ct图像;

图4是本发明中所使用的低剂量ct图像扩充样例图;

图5是本发明中所使用的全剂量ct图像扩充样例图;

图6是本发明中的图像取块示例图;

图7是本发明中所构建的全卷积神经网络结构图;

图8是用本发明对低剂量ct图像去噪后的结果图。

具体实施方式

以下结合附图,对本发明的具体实施方案和效果作进一步的解释和说明:

参照图1,本发明基于梯度正则化的低剂量ct图像去噪方法,其实现步骤如下:

步骤1:数据准备。

1a)使用ct设备同时对人体同一部位进行全剂量和低剂量成像,其中,全剂量ct成像时x射线管电压为120千伏,管电流为200毫安,辐射剂量约为3毫希弗;低剂量ct图像时x射线管电压为120千伏,管电流为50毫安,辐射剂量为0.75毫希弗;

1b)将得到成对的低剂量ct图像x和全剂量ct图像y,记为{x,y},其中低剂量ct图像x,如图2所示,全剂量ct图像y如图3所示,两种图像的大小均为512*512,将采集的ct图像数据集记为d;

1c)对采集的ct图像数据集d进行扩充,得到扩充ct图像数据集da:

1c1)对图像对{x,y}中的低剂量ct图像x进行扩充:

参照图4,本步骤的具体实现如下:首先,对低剂量ct图像,如图4(a)所示,沿其垂直中轴线左右翻转,得到第2张低剂量ct图像,如图4(e)所示;然后,对这2张低剂量ct图像分别沿逆时针方向旋转90度,180度和270度,得到6张新的不同角度的低剂量ct图像,如图4(b)~4(d)和4(f)~4(h)所示,即1张低剂量ct图像经扩充后共计得到8张低剂量ct图像;

1c2)对图像对{x,y}中的全剂量ct图像y进行扩充:

参照图5,本步骤的具体实现为:首先,对全剂量ct图像如图5(a)所示,沿其垂直中轴线左右翻转,得到第2张全剂量ct图像,如图5(e)所示;然后,对这2张全剂量ct图像分别沿逆时针方向旋转90度,180度和270度,得到6张新的不同角度的全剂量ct图像,如图5(b)~5(d)和5(f)~5(h)所示,即1张全剂量ct图像经扩充后共计得到8张全剂量ct图像;

1c3)对ct图像数据集d中每一对ct图像对{x,y},都进行上述步骤1c1)和1c2)的扩充操作,得到扩充ct图像数据集da;

1d)对da中每张低剂量ct和全剂量ct图像进行归一化操作,将灰度值映射到区间[0,1],归一化公式如下:

其中,x(r,c)表示ct图像x的在第r行第c列的像素灰度值,max(x)表示图像中所有像素的最大值,min(x)表示图像中所有像素的最小值,表示归一化后的ct图像在第r行第c列的数值;

1e)对数据集da中归一化后的每对ct图像进行取块,建立图像块数据集dp;

参照图6,本步骤的具体实现如下:

1e1)将ct图像的中心区域定义为距ct图像四周像素距离至少为48的图像区域;

1e2)对图像对{x,y}中的低剂量ct图像x进行取块:

在低剂量ct图像上的中心区域随机选择位置,记录此位置,并以此位置从左上角向右下方截取尺寸为48*48的图像块,记为xp,在一张低剂量ct图像上共计取32个图像块;

1e3)对图像对{x,y}中的全剂量ct图像y进行取块:

在全剂量ct图像上,根据在低剂量ct图像上记录的取块位置,并以此位置从左上角向右下方截取尺寸为48*48的图像块yp,在一张全剂量ct图像上共计取32个图像块;

1e4)将截取的一一对应的图像块xp和yp记为ct图像块对{xp,yp},对数据集da中每一对ct图像对{x,y}都进行上述步骤1e2)和1e3)的取块操作,得到ct图像块数据集dp。

步骤2:根据得到的图像块数据集dp,搭建一个全卷积去噪神经网络n。

参照图7,本步骤的具体实现如下:

2a)将ct图像作为网络n的输入,网络n的输出结果为去噪后的ct图像,输入输出图像大小保持不变;

2b)通过交叉验证,将网络n的层数设为12,包括输入层、中间层和输出层;

2c)设第1层为输入层,其包含一个1个卷积层conv和1个修正线性单元激活层relu,该卷积层含有64个卷积核,每个卷积核大小为3*3,如图7中黄色模块所示;

2d)设第2层至第11层为中间层,该中间层的每一层都由同一种模块构造,该模块按连接顺序依次含有1个卷积层、1个relu层和1个批规范化层bn,该卷积层含有64个卷积核,每个卷积核大小为3*3,如图7中绿色模块所示;

2e)设第12层为输出层,其包含1个卷积层,该卷积层只包含1个卷积核,卷积核大小为3*3,如图7中蓝色模块所示。

上述网络n中的卷积层,其数学形式如下:

其中,fil表示网络第l层的第i个特征图,当l=0时,f0表示网络输入的低剂量ct图像块,表示网络第l层的第i个卷积核的权值,表示网络第l层的第i个卷积核的偏置,nl表示网络第l层的卷积核数量,∑表示求和运算,表示图像卷积运算,该卷积运算采用“same”方式,以保持卷积前后图像大小不变。

上述网络n中的修正线性单元激活层relu,其数学形式如下:

其中,x表示输入数据;

上述网络n中的批规范化层bn,其数学形式如下:

其中,x表示输入数据,表示其均值,σ表示其标准差,γ表示bn层的斜率参数,β表示bn层的偏置参数。

步骤3,对网络n的权值w和偏置b进行初始化。

3a)使用msra方法对网络n的权值w进行初始化,其公式如下:

其中,w表示网络的权值,n(·,·)表示高斯分布,即网络的权值w服从均值为0,标准差为的高斯分布;

3b)将网络n的偏置b均初始化为数值0。

步骤4:使用图像块数据集dp,训练全卷积去噪网络n。

4a)打乱dp中图像块对的顺序,依次从dp中选择m对ct图像块数据,记为一个训练批次{xpb,ypb},其中低剂量图像块作为网络的输入,全剂量图像块用来计算网络的损失,i=1,2,...,m,m表示批量大小,本实例中取值为m=256;

4b)将xpb输入到网络n中,经过前向传播,得到网络的输出apb,计算apb和ypb之间的均方误差,得到网络n的灰度损失lv:

其中,row和col分别表示输入的图像块的高度和宽度,取值均为48,表示第i个全剂量ct图像块的第r行第c列的像素灰度值,表示第i个网络输出去噪图像块的第r行第c列的像素灰度值,||·||2表示平方运算,∑表示求和运算;

4c)选择梯度卷积算子g,本实例中选取如下:

4d)使用梯度算子g分别卷积apb和ypb,得到对应的梯度图ga和gy:

其中,表示图像卷积运算,该卷积运算采用“same”方式,以保持卷积前后图像大小不变。

4e)计算ga和gy之间的均方误差,得到网络n的梯度损失lg:

其中,表示第i个全剂量ct图像块对应的梯度图在第r行第c列的数值,表示第i个网络输出去噪图像块对应的梯度图在第r行第c列的数值;

4f)对网络n的灰度损失lv和梯度损失lg进行线性加权,得到网络的总体损失l:

l=lv+λ×lg,

其中,λ为加权系数,通过交叉验证取值为2;

4g)使用带动量项的批量梯度下降算法优化整体损失l,初始学习速率为0.0001,动量项参数为0.9,权值衰减因子为0.0001,对网络的权值w进行更新;

4h)重复以上步骤(4a)到(4g),直到达到网络训练的最大迭代次数t=100000,得到训练好的去噪网络nt。

步骤5:使用训练好的去噪网络nt对低剂量ct图像进行去噪。

将一张完整的低剂量ct图像xt,如图8(a)所示,输入到训练好的去噪网络nt中,经过该去噪网络nt的前向传播,得到去噪后的ct图像at,如图8(b)所示。

去噪网络虽然是利用图像块去训练的,但其是一个全卷积网络,不受输入图像尺寸的影响,因此可以直接输入完整的ct图像进行去噪。使用图像块进行训练是因为完整的ct图像数量有限,可能导致训练的网络过拟合,且使用图像块训练可以加快网络的训练速度。

从图8可以看出,在图像恢复的过程中同时考虑图像灰度信息的重建和图像梯度的保持,得到的去噪的后图像细节更丰富、清晰,可视性更好。

以上描述仅是本发明的一个具体实例,不构成对本发明的任何限制,显然对于本领域的专业人员来说,在了解了本发明内容和原理后,都可能在不背离本发明原理、结构的情况下,进行形式和细节上的各种修正和改变,但是这些基于本发明思想的修正和改变仍在本发明的权利要求保护范围之内。

当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1