一种基于图像增强和扩散模型的低剂量CT图像降噪方法技术

本发明专利技术提供了一种基于图像增强和扩散模型的低剂量CT图像降噪方法,S1：构建低剂量CT图像样本集，对进行多步处理以达到的成像水平；S2：将通过滤波器，得到输出；S3：对进行CT图像特征增强，得到输出；S4：通过残差卷积模块对图像进行降噪重建，输出图像；S5：将输入到公开的深度生成模型扩散去噪模型（DDPM）中，将原始数据分布变为正态分布，使用参数化的U

【技术实现步骤摘要】
一种基于图像增强和扩散模型的低剂量CT图像降噪方法


[0001]本专利技术涉及医学图像重建
，具体而言，特别涉及一种基于图像增强和扩散模型的低剂量CT图像降噪方法。

技术介绍

[0002]现代医疗诊断的方法近年来发展迅速，其中最重要的工具是计算机断层扫描CT，其通常用于帮助诊断复杂的骨折、肿瘤、心脏病、肺气肿等，但由于患者暴露于不同时间的放射性波，CT会增加人类从医疗器械中接收的辐射，甚至导致其成为继背景辐射之后影响人类的第二大辐射源。减少CT扫描中射线的剂量已成为一种趋势，但这会导致诸如噪声增加、边缘、角落和锐利特征的对比度降低以及图像过度平滑等负面效果。
[0003]为了解决这个固有的物理问题，已经涌现出很多深度卷积网络方法来提高低剂量CT图像的质量，但现有的卷积网络方法会面临诸如过度平滑、边缘丢失和纹理细节破坏等问题，而卷积神经网络本质上是隐含地将降噪与全局范围内的操作联系起来，而忽略了利用局部视觉信息。视觉Transformer中的局部注意力机制对局部乃至全局的特征增强和扩散模型的高质量图像生成在许多计算机视觉任务中取得了巨大成功，包括图像恢复和重建领域，而在医学图像数据集上的利用目前还处于起步阶段；扩散概率模型应用在核磁共振影像的分割和检测中已经取得了初步的成效，但在CT图像的应用场景中仍有待开发。因此，采用先进技术提高降噪后的低剂量CT图像质量仍然是研究人员需要解决的关键问题。

技术实现思路

[0004]为了弥补现有技术的不足，本专利技术提供了一种基于图像增强和扩散模型的低剂量CT图像降噪方法。
[0005]本专利技术是通过如下技术方案实现的：一种基于图像增强和扩散模型的低剂量CT图像降噪方法，具体包括以下步骤： S1：构建低剂量CT图像样本集，其中为低剂量CT图像，为普通剂量CT图像，对进行多步处理以达到的成像水平；S2：将通过滤波器，得到输出；S3：对进行CT图像特征增强，通过局部注意力模块，在图像空间域应用滑动窗口计算注意力，在特征空间域应用聚类计算注意力，最终得到输出；S4：通过残差卷积模块对图像进行降噪重建，输出图像；对经过纹理增强后的图像进行降噪可以在获得降噪效果的同时不丢失纹理细节；S5：将输入到公开的深度生成模型扩散去噪模型（DDPM）中，模型分为扩散过程和逆扩散过程两个阶段，在扩散阶段，通过不断对添加高斯噪声，将原始数据分布变为正态分布；在逆扩散阶段，使用参数化的U
‑
Net神经网络将从正态分布恢复到原始数据分
布。
[0006]作为优选方案，步骤S2具体包括以下步骤：S2
‑
1：将两种类型包括垂直、水平方向的算子定义为一组，与进行卷积运算，获得一组特征图，以提取边缘信息；S2
‑
2：将上一步输出的特征图与在通道维度上叠加，得到该模块的最终输出；(1)其中，表示利用算子进行卷积运算。
[0007]作为优选方案，步骤S3具体包括以下步骤：S3
‑
1：利用公开的预训练模型ViT中线性嵌入层的操作,对进行变换得到该模块的输入；(2)其中，为线性嵌入层操作，该操作将每个二维图像块展平为一维向量嵌入然后将其线性投影到所需的输入维度中；S3
‑
2：通过局部注意力机制模块，：(3)其中，表示第一阶段局部注意力机制模块；S3
‑
3：依次输入到，，中，共计4阶段，并在相邻局部注意力机制模块中间插入一次卷积层作为特征融合，共计3次，最终得到输出；(7)(8)(9)其中，表示开源的Python机器学习库torch.nn中的Conv2d函数，卷积核为3
×
3大小，步长为2。
[0008]进一步地，步骤S3
‑
2具体包括以下步骤：S3
‑2‑
1：通过正则化层与公开的预训练模型Swin Transformer中图像空间局部注意力模块后在图像空间域进行特征增强得到；(4)其中是图像空间局部注意力模块，是正则化层，过程采用了开源的Python机器学习库torch.nn中的LayerNorm函数，后续提及的正则化层均为当前正则化层的相同操作；S3
‑2‑
2：先后通过正则化层和公开的预训练模型BOAT中图像特征空间局部注
意力模块中，得到；(5)其中，表示图像特征空间局部注意力模块；S3
‑2‑
3：通过正则化层和全连接层后得到第一阶段局部注意力模块的输出，此过程包含残差快速链接；(6)其中，表示全连接层。
[0009]作为优选方案，步骤S4具体包括以下步骤：S4
‑
1：对变形为特征图，输送到包含两个卷积层和一个激活层的ResNet模块中得到输出；(10)S4
‑
2：类似地，通过另一个相同的模块后得到输出； (11)其中，表示torch.nn中的Conv2d函数，卷积核为3
×
3大小，步长为1,表示torch.nn中的ReLU函数，变形操作采用了开源的数值计算扩展库Numpy中的reshape函数。
[0010]作为优选方案，步骤S5具体包括以下步骤：S5
‑
1：执行扩散过程，在扩散过程方差设置为从到线性增加的常数，设置为1000，直到图像信号被完全破坏，通过马尔可夫过程的特性可以推导出任意时间步的解析式：(12)其中，，，为某一时刻t的加噪声图像，是和等维度的隐变量；S5
‑
2：执行逆扩散过程，该过程使用参数化的神经网络来建模, 公式表达如下：(13)其中，表示为到过程的噪声分布预测，遵从 的分布，表示U
‑
Net神经网络中的可训练参数。
[0011]本专利技术由于采用了以上技术方案，与现有技术相比使其具有以下有益效果：一是将CT图像边缘增强和特征增强联合增强以提升降噪效果，同时边缘增强又可辅助特征增强，在降噪前对低剂量CT图像进行增强旨在对降噪后的CT图像依然能够比肩和普通剂量CT图像中真实的纹理和易丢失的细节；边缘检测器Sobel滤波可以提取图像的纹理边缘，同时
与原图像加和能够实现边缘细节增强；利用公开的SwinTransformer和BOAT中的局部注意力机制实现在空间域和特征域进行增强，不同于卷积神经网络框架，局部注意力机制能够完成对图像全局信息的捕获，从而提取细微的图像语义信息；其次，在降噪阶段引入公开的扩散概率模型（DDPM），作为当今前沿的深度生成模型，可以生成质量更高的降噪CT图像。另外，本专利技术整体上采用了端到端的训练方式，减少了工程学习的复杂度。
[0012]本专利技术的附加方面和优点将在下面的描述部分中变得明显，或通过本专利技术的实践了解到。
附图说明
[0013]本专利技术的上述和/或附加的方面和优点从结合下面附图对实施例的描述中将变得明显和容易理解，其中：图1为基于局部注意力机制和扩散概率模型的低剂量CT图像降噪流程图；图2为图1的详细流程图；图3为改进后的Sobel滤波器示例；图4为局部注意力机制流程图。
具体实施本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于图像增强和扩散模型的低剂量CT图像降噪方法，其特征在于,具体包括以下步骤：S1：构建低剂量CT图像样本集，其中为低剂量CT图像，为普通剂量CT图像，对进行多步处理以达到的成像水平；S2：将通过滤波器，得到输出；S3：对进行CT图像特征增强，通过局部注意力模块，在图像空间域应用滑动窗口计算注意力，在特征空间域应用聚类计算注意力，最终得到输出；S4：通过残差卷积模块对图像进行降噪重建，输出图像；对经过纹理增强后的图像进行降噪可以在获得降噪效果的同时不丢失纹理细节；S5：将输入到公开的深度生成模型扩散去噪模型（DDPM）中，模型分为扩散过程和逆扩散过程两个阶段，在扩散阶段，通过不断对添加高斯噪声，将原始数据分布变为正态分布；在逆扩散阶段，使用参数化的U
‑
Net神经网络将从正态分布恢复到原始数据分布。2.根据权利要求1所述的一种基于图像增强和扩散模型的低剂量CT图像降噪方法,其特征在于,所述步骤S2具体包括以下步骤：S2
‑
1：将两种类型包括垂直、水平方向的算子定义为一组，与进行卷积运算，获得一组特征图，以提取边缘信息；S2
‑
2：将上一步输出的特征图与在通道维度上叠加，得到该模块的最终输出；(1)其中，表示利用算子进行卷积运算。3.根据权利要求1所述的一种基于图像增强和扩散模型的低剂量CT图像降噪方法,其特征在于,所述步骤S3具体包括以下步骤：S3
‑
1：利用公开的预训练模型ViT中线性嵌入层的操作,对进行变换得到该模块的输入；(2)其中，为线性嵌入层操作，该操作将每个二维图像块展平为一维向量嵌入然后将其线性投影到所需的输入维度中；S3
‑
2：通过局部注意力机制模块，：(3)其中，表示第一阶段局部注意力机制模块；S3
‑
3：依次输入到，，中，共计4阶段，并在相邻局部注意力机制模块中间插入一次卷积层作为特征融合，共计3次，最终得到输出；(7)(8)
(9)其中，表示开源的Python机器学习库torch.nn中的Conv2d函数，卷积核为3
×
3大小，步长为2。4.根据权利要求3所述的一种基于图像...

【专利技术属性】
技术研发人员：蹇木伟，于小洋，靳悦，王芮，
申请(专利权)人：山东九德智能科技有限公司，
类型：发明
国别省市：