首页> 中国专利> 基于噪声先验的CNN医学CT图像去噪方法

基于噪声先验的CNN医学CT图像去噪方法

摘要

基于噪声先验的CNN医学CT图像去噪,具体步骤如下:步骤1)创建医学CT图像模型;步骤2)构建噪声先验信息提取网络;步骤3)构建去噪网络;步骤4)训练噪声先验信息提取网络并更新参数;步骤5)训练去噪网络的训练并更新参数;步骤6)医学CT图像去噪;本发明具有以下优点和创新:提出了使用医学CT图像的噪声先验信息进行去噪;使用两个网络进行去噪;分别是:噪声先验信息提取网络、去噪网络;噪声先验信息提取网络对CT医学图像进行噪声信息提取,以获取更多图像信息;将噪声先验信息提取网络预测出的噪声水平图与噪声图串行拼接,并输入到去噪网络;在网络中加入了BN层,提高了网络的泛化能力,加快网络收敛速度,提高网络去噪性能。

著录项

  • 公开/公告号CN112419169A

    专利类型发明专利

  • 公开/公告日2021-02-26

    原文格式PDF

  • 申请/专利权人 浙江工业大学;

    申请/专利号CN202011094605.3

  • 发明设计人 张聚;牛彦;施超;潘玮栋;陈德臣;

    申请日2020-10-14

  • 分类号G06T5/00(20060101);G06N3/04(20060101);G06N3/08(20060101);

  • 代理机构33201 杭州天正专利事务所有限公司;

  • 代理人王兵

  • 地址 310014 浙江省杭州市下城区潮王路18号

  • 入库时间 2023-06-19 10:00:31

说明书

技术领域

本发明涉及医学图像去噪领域,主要涉及医学CT图像,具体涉及一种适用于医学CT图像的基于噪声先验的CNN医学CT图像去噪方法。

技术背景

由于医学CT图像可以简单便捷的获取病人的数据信息,医学图像处理逐渐得到人们的重视。通过利用来自各种技术手段获得的医学CT图像已经成为广大医务人员工作和生活中不可缺少的一部分,帮助医生对疾病进行准确的分析和全面的诊断。随着当前获取的病人数据量的不断增加,其病症也逐步呈现多样性的趋势,给病人做出科学的医学诊断和准确的治疗带来的巨大的挑战。

在数字化采集数据和传输数据的过程中,医学CT图像往往会在这个过程中受到噪声的干扰。大多数真实的医学图像都是噪声图像。这对医学图像的分析有很大的影响。医学图像去噪是图像预处理阶段的重要任务之一。噪声的类型也不同,如椒盐噪声、高斯噪声等。在医学CT图像中,噪声会直接影响医生对诊断结果的判断。

在传统的非机器学习去噪方法,例如线性滤波法、中值滤波法、维纳滤波法等,会弱化图像的细节特征,丢失重要信息。随着计算机的发展,深度卷积神经网络在处理自然图像方面取得了巨大的发展。该方法通过最小化损失函数来优化深度神经网络中权重参数。其损失函数是通过累计计算噪声图像和干净图像对应位置像素的差异值获得,通过PSNR、SSIM等评级指标判断去噪能力。然而在进行降噪处理后的图像中会损失一些细节图像,在相对较高的噪声图像中会损失更多的细节纹理也很难做到真正的多级去噪,这就不符合去噪的目的。

发明内容

本发明要克服现有技术弊端,提供一种基于噪声先验的CNN医学CT图像去噪方法。

本发明的目的在于提高医学CT图像的去噪效果,在以往采用的医学图像去噪方法中往往是直接将单一噪声级图像或者多噪声级图像混合作为训练数据,使得到的去噪模型仅仅可以处理有限范围内的噪声图片,这会导致神经网络模型不能充分泛化到有更加宽泛噪声级的图像中,并且在模型训练时神经网络不能充分获取图片信息。为了提高整个网络的泛化能力,本发明利用连续多个含跳跃连接的卷积神经网络对含噪声的CT图像进行噪声信息的提取,将提取出来的数据与噪声图像串行连接作为去噪网络的输入,使用提取的信息和噪声图像来优化神经网络中的参数。

本发明的创新和优点在于:本发明充分利用卷积神经网络学习到图片噪声信息,将噪声信息与噪声图像相结合,使得去噪网络可以充分利用噪声信息进行训练,增强的去噪网络的泛化能力。为了神经网络的训练速度,本发明在去噪模型中采用BN层。最后通过仿真验证了方法的可行性也证明了对医学CT图像去噪具有一定的效果。

为使本发明的目的、技术方案和优点更加清晰,下面就对本发明的技术方案进行详细的描述,基于噪声先验的CNN医学CT图像去噪方法,具体步骤如下:

步骤1)创建医学CT图像模型:

创建医学CT图像模型采用高斯噪声模型,其数学表达式为:

Y=X+V (1)

其中X是为不含噪声的干净图像,Y为实际含噪声图像,V为噪声;V的噪声分布服从高斯分布,高斯噪声是指它的概率密度函数服从高斯分布(即正态分布)的一类噪声,即为高斯随机变量z概率密度函数,其数学表达式为:

其中μ表示为数学期望,σ表示为标准差;

步骤2)构建噪声先验信息提取网络:

噪声先验信息提取网络将噪声图像作为输入,构建三个网络层,分别为输入层、隐藏层和输出层;输入层采用3×3大小的卷积核,通道数量设置为32,卷积步长设为1,padding为1;隐藏层设置为5层并且每一层之间采用短跳跃连接,均采用大小为3×3的卷积核,通道数量设置为32,卷积步长设为1,padding为1;输出层输入层采用3×3大小的卷积核,通道数量设置为1,卷积步长设为1,padding为1;噪声先验信息提取网络将噪声图像作为噪声先验信息输入,输出图像尺寸与输入图像尺寸相同,通道数设为32;

步骤3)构建去噪网络:

去噪网络分为三部分:

第一部分将噪声先验信息提取网络的输出和噪声原图作为输入,先经过一个Conv+BN+PReLU层,其中卷积核设置为3×3,步长为1,padding为1,然后通过一个卷积,其卷积核设置为3×3,步长为2,padding为1;

第二部分使用三个相同结构的子网络并联接入;每一个子网络有13层;其中第2层使用Conv+PReLU结构且卷积核设置为3×3,步长为1,padding为1;第4层使用Conv结构且卷积核设置为3×3,步长为1,padding为1;第9、10、11、12、13层使用Conv+BN+PReLU结构且卷积核设置为3×3,步长为1,padding为1;第1、3层使用Conv+PReLU结构且卷积核设置为3×3,步长为2,padding为1;第5、7、8层使用Sub-pixelconvolution;第1、2和3、4采用跳跃连接;子网络第3层的输出分别串行加入第6层中;最后将三个子网络的输出串联拼接;

第三部分使用Conv+PReLU的结构,使用7个卷积核为3×3,步长为1,padding为1的卷积层;在第1—5层中每层之间均使用跳跃连接;在第1层和第4层之间使用跳跃连接;在第二层和第五层之间只用跳跃连接;

步骤4)训练噪声先验信息提取网络并更新参数:

在图像预处理阶段将训练集和验证集裁剪为32×32的尺寸并加入噪声,去噪网络中的权重参数设施为随机初始化,使用高斯加性白噪声替代医学CT图像中的噪声,对原始干净图像分别添加均值为0,标准差为5、10、15、20、25、30、35、40、45、50的高斯白噪声得到去噪网络训练数据,将标准差张成32×32尺寸的张量,该张量为噪声图像的噪声水平图,由此得到先验信息提取网络的标签数据;

为了约束生成的特征图符合噪声信息,将模型输出特征图和标签对应位置的像素做均方误差和计算,求出一个Batch的输出图像和干净图像的平均均方误差;其损失函数公式为:

θ

优化参数,具体包括:

CNN网络使用Adam优化器对噪声先验信息提取网络的参数进行更新,不添加偏置,只对噪声先验信息提取网络中权重进行参数优化更新;

步骤5)训练去噪网络并更新参数:

采用医学脑部CT图像作为去噪目标,在图像预处理阶段将训练集和验证集裁剪为32×32的尺寸并加入噪声,去噪网络中的权重参数设施为随机初始化,使用高斯加性白噪声替代医学CT图像中的噪声,对原始干净图像分别添加均值为0,标准差为15、30、45的高斯白噪声,将标准差张成32×32尺寸的噪声水平图并将其与噪声图片串行拼接,由此得到去噪网络训练数据;

优化参数,具体包括:

该网络使用Adam优化器对整个去噪网络的参数进行更新,不添加偏置,只对网络中权重进行参数优化更新;其损失函数公式为:

θ

步骤6)医学CT图像去噪;

本发明使用两个网络进行去噪,分别是:噪声先验信息提取网络、去噪网络;首先向噪声先验信息提取网络中输入测试集中含噪声的医学CT图像,将噪声先验信息提取网络预测出的噪声水平图与噪声图串行拼接,并输入去噪网络进行预测,最后去噪网络输出去除噪声的医学CT图像。

本发明的网络模型实现了盲去噪处理噪声图片,不需要专门针对某一个噪声级图片进行训练。

本发明具有以下优点:

1.提出了使用噪声先验信息对医学CT图像去噪,增强了网络的鲁棒性,可以实现盲去噪。

2.在CNN网络模型中加使用PReLU激活函数,提高了网络的去噪性能

3.使用跳跃连接的方式,可以使网络可以训练更深的结构。

4.采用了BN层,可以加快网络训练的速度。

附图说明

图1是本发明的含有高斯噪声的医学CT图像示意图;

图2是本发明的PReLU激活函数示意图;

图3是本发明的噪声先验信息提取网络网络结构;

图4(1)和图4(2)是本发明的去噪网络结构,其中图4(1)是去噪网络3个部分中第1、2部分,图4(2)是去噪网络3个部分中的第3部分;

图5是本发明的医学CT图像去噪示意图

具体实施方式:

以下结合附图对本发明做具体的解释说明

本发明的基于噪声先验的CNN医学CT图像去噪算法具体步骤如下:

步骤1)创建医学CT图像模型:

创建医学CT图像模型采用高斯噪声模型。其数学表达式为:

Y=X+V (1)

其中X是为不含噪声的干净图像,Y为实际含噪声图像,V为噪声;V的噪声分布服从高斯分布,高斯噪声是指它的概率密度函数服从高斯分布(即正态分布)的一类噪声,噪声图像如图1所示,即为高斯随机变量z概率密度函数,其数学表达式为:

其中μ表示为数学期望,σ表示为标准差;

步骤2)构建噪声先验信息提取网络:

噪声先验信息提取网络将噪声图像作为输入,构建三个网络层,分别为输入层、隐藏层和输出层;输入层采用3×3大小的卷积核,通道数量设置为32,卷积步长设为1,padding为1;隐藏层设置为5层并且每一层之间采用短跳跃连接,均采用大小为3×3的卷积核,通道数量设置为32,卷积步长设为1,padding为1;输出层输入层采用3×3大小的卷积核,通道数量设置为1,卷积步长设为1,padding为1;噪声先验信息提取网络将噪声图像作为噪声先验信息输入,输出图像尺寸与输入图像尺寸相同,通道数设为32,如图2所示;

步骤3)构建去噪网络:

去噪网络分为三部分:

第一部分将噪声先验信息提取网络的输出和噪声原图作为输入,先经过一个Conv+BN+PReLU层(PReLU如图3所示),其中卷积核设置为3×3,步长为1,padding为1,然后通过一个卷积,其卷积核设置为3×3,步长为2,padding为1;

第二部分使用三个相同结构的子网络并联接入;每一个子网络有13层;其中第2层使用Conv+PReLU结构且卷积核设置为3×3,步长为1,padding为1;第4层使用Conv结构且卷积核设置为3×3,步长为1,padding为1;第9、10、11、12、13层使用Conv+BN+PReLU结构且卷积核设置为3×3,步长为1,padding为1;第1、3层使用Conv+PReLU结构且卷积核设置为3×3,步长为2,padding为1;第5、7、8层使用Sub-pixelconvolution;第1、2和3、4采用跳跃连接;子网络第3层的输出分别串行加入第6层中;最后将三个子网络的输出串联拼接;

第三部分使用Conv+PReLU的结构,使用7个卷积核为3×3,步长为1,padding为1的卷积层;在第1—5层中每层之间均使用跳跃连接;在第1层和第4层之间使用跳跃连接;在第二层和第五层之间只用跳跃连接,如图4所示;

步骤4)训练噪声先验信息提取网络并更新参数:

在图像预处理阶段将训练集和验证集裁剪为32×32的尺寸并加入噪声,去噪网络中的权重参数设施为随机初始化,使用高斯加性白噪声替代医学CT图像中的噪声,对原始干净图像分别添加均值为0,标准差为5、10、15、20、25、30、35、40、45、50的高斯白噪声得到去噪网络训练数据,将标准差张成32×32尺寸的张量,该张量为噪声图像的噪声水平图,由此得到先验信息提取网络的标签数据;

为了约束生成的特征图符合噪声信息,将模型输出特征图和标签对应位置的像素做均方误差和计算,求出一个Batch的输出图像和干净图像的平均均方误差;其损失函数公式为:

θ

优化参数,具体包括:

CNN网络使用Adam优化器对噪声先验信息提取网络的参数进行更新,不添加偏置,只对噪声先验信息提取网络中权重进行参数优化更新;

步骤5)训练去噪网络的训练并更新参数:

采用医学脑部CT图像作为去噪目标,在图像预处理阶段将训练集和验证集裁剪为32×32的尺寸并加入噪声,去噪网络中的权重参数设施为随机初始化,使用高斯加性白噪声替代医学CT图像中的噪声,对原始干净图像分别添加均值为0,标准差为15、30、45的高斯白噪声,将标准差张成32×32尺寸的噪声水平图并将其与噪声图片串行拼接,由此得到去噪网络训练数据;

优化参数,具体包括:

该网络使用Adam优化器对整个去噪网络的参数进行更新,不添加偏置,只对网络中权重进行参数优化更新;其损失函数公式为:

θ

步骤6)医学CT图像去噪;

本发明使用两个网络进行去噪,分别是:噪声先验信息提取网络、去噪网络;首先向噪声先验信息提取网络中输入测试集中含噪声的医学CT图像,将噪声先验信息提取网络预测出的噪声水平图与噪声图串行拼接,并输入去噪网络进行预测,最后去噪网络输出去除噪声的医学CT图像,如图5所示。

去获取专利,查看全文>

相似文献

  • 专利
  • 中文文献
  • 外文文献
获取专利

客服邮箱:kefu@zhangqiaokeyan.com

京公网安备:11010802029741号 ICP备案号:京ICP备15016152号-6 六维联合信息科技 (北京) 有限公司©版权所有
  • 客服微信

  • 服务号