ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LDCT图像重建论文——Eformer: Edge Enhancement based Transformer for Medical Image Denoising

2022-01-07 12:59:07  阅读:254  来源: 互联网

标签:采样 Transformer based Eformer Sobel 卷积 损失 Convolution


Eformer是进行低剂量CT图像重建的工作,它第一次将Transformer用在医学图像去噪上,ICCV2021的paper,值得一读。

知乎同名账号同步发布。

目录

一、架构和贡献

在这里插入图片描述
和Uformer一样,将LeWin Transformer加入了UNet中,不一样的是将经过Sobel Convolution处理的图片concat到UNet中不同阶段的encode和decode过程中。采用了残差学习,原图减残差为预测去噪结果。注意下采样和上采样是用卷积和反卷积(注意避免棋盘效应,后文会讲)。

二、主要细节

需要了解的主要就是Sobel Convolution和LeWin Transformer,后者我在Uformer笔记中已经记录,简单放上公式:
在这里插入图片描述

2.1,Sobel Convolution

Sobel Convolution的功能是获得edge-enhanced images,效果如下图所示:
在这里插入图片描述
细节没看,文章中对其的引用放下面:

[19] Tengfei Liang, Yi Jin, Yidong Li, and Tao Wang. Edcnn: Edge enhancement-based densely connected network with compound loss for low-dose ct denoising. 2020 15th IEEE International Conference on Signal Processing (ICSP), Dec 2020.
[24] Irwin Sobel. An isotropic 3x3 image gradient operator. Presentation at Stanford A.I. Project 1968, 02 2014.

Sobel Convolution之后跟的激活函数是GeLU。

2.2, 下采样和上采样

采用3×3的卷积进行下采样,stride为2,padding为1.作者说不用pooling的原因是可能会丢失细节,所以采用strided conv来下采样。

采用反卷积进行上采样,作者提到了棋盘效应:

棋盘效应,源头就是反卷积过程中,当卷积核大小不能被步长整除时,反卷积就会出现重叠问题,插零的时候,输出结果会出现一些数值效应,就像棋盘一样。

所以卷积核大小应当能被步长整除,作者采用4×4的卷积核,stride为2.

2.3,损失函数

采用两个损失函数,一个是MSE,一个是感知损失。

MSE:
在这里插入图片描述
MSE损失会造成过度平滑和图像模糊,所以作者没有只用它。

ResNet based MSP(Multi-scale Perceptual):
在这里插入图片描述
x i − R ( x i ) x_i-R(x_i) xi​−R(xi​)是网络的预测结果, y i y_i yi​是ground-truth, ϕ s \phi_s ϕs​表示ResNet,不同的s表示不同尺度的ResNet,一共有C种s,所以是multi-scale。

最终损失函数如下所示:
在这里插入图片描述

作者也对比了不同方法采用的不同损失:
在这里插入图片描述

三、实验

在Mayo上做的,先用Pydicom将像素从Dicom文件提取到Numpy array中,然后将像素数值scale到0-1之间。其他数据集和训练细节在paper第7页。

和其他方法的对比:
在这里插入图片描述
作者放的效果图:
在这里插入图片描述
在这里插入图片描述

标签:采样,Transformer,based,Eformer,Sobel,卷积,损失,Convolution
来源: https://blog.csdn.net/umbrellalalalala/article/details/122361552

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有