CrossSR | 新型Cross卷积大幅提升超分结构信息,已开源

AI算法与图像处理

共 4522字,需浏览 10分钟

 ·

2022-01-10 18:51


点击下方AI算法与图像处理”,一起进步!

重磅干货,第一时间送达

https://arxiv.org/abs/2201.01458

https://github.com/yuqing-liu-dut/Cross-SRN

1Abstract

作者设计了一种分层特征挖掘网络,它通过多尺度特征融合方式探测、保持结构信息。首先,基于传统边缘检测器提出了Cross卷积进行边缘特征定位与表达;然后,设计了带特征规范化与通道注意力的CCB模块以对特征固有的相关性进行考量;最后,将Cross卷积嵌入到MFFG模块并以分层方式挖掘结构特征相关性,构建了结构保持网络Cross-SRN

实验结果表明:CrossSRN取得了与其他SOTA方案相当甚至更优的复原性能,同时具有精确且清晰的结构细节。与此同时,我们定义了一种选择具有丰富结构纹理图像的准则。所提CrossSRN在所选基准数据及上取得了更优的性能,说明所提方案具有更好的边缘保持优势。

2Method

上图给出了所提方案整理架构示意图,架构风格与IMDN相似,区别主要有两部分:

  • 核心模块,即MFFG部分,这里也是大部分网络架构的区别所在。
  • body的fuse部分,IMDN采用了类似RDN的多尺度融合,即把各个IMDB的输出都进行了融合;而CrossSRN则是采用类似EDSR仅对最后模块的输出融合。

注:看这里的图示采用了两个卷积应该是类似IMDN的融合方式,但paper提到的是,即只有最后一个模块。

Cross Convolution

上图给出了常规卷积与本文所提Cross卷积的示意图,Cross卷积其实就是两路并行非对称卷积,可以表示如下:

Cross卷积通过两路并行的非对称卷积对水平与垂直梯度信息进行探索以聚焦于边缘信息挖掘。相比常规卷积,Cross卷积具有更少的参数量与计算复杂度、具有相同的感受野,重要的是具有更好的结构保持能力。

上图给出了常规卷积与Cross卷积在结构保持方面的对比示意图,Cross卷积可以保持更多的梯度方向信息。为更好的说明Cross卷积的边缘保持能力,作者还提供了下图的对比图,可以看到:Cross卷积保持更多的边缘信息,验证了其结构纹理探索能力。

Multi-scale Feature Fusion Group

为获得精确的边缘信息,我们构建了CCB模块,见上图。除了Cross卷积外,CCB还包含F-Norm与CA(通道注意力,没什么可说的),两者分别用于空域与通道信息重要性挖掘。F-Norm可表示如下:

更多关于F-Norm的介绍可参考《Iterative Network for Image Super-resolution》一文,为方便理解,笔者在这里摘抄了F-Norm的实现code(是不是简单到家!)如下:

class AdaptiveFM(nn.Module):
    def __init__(self, channels, kernel_size=3):
        super(AdaptiveFM, self).__init__()
        self.conv = nn.Conv2d(channels, channels, kernel_size, padding=kernel_size//2, groups=channels)
    def forward(self, x):
        return self.conv(x)+x

由于边缘信息具有尺度敏感性,故作者将CCB以多尺度特征融合方式构建了MFFG以探索不同尺度特征,见上图(类似IMDN、RFDN的模块)。可描述如下:

分层特征将以残差模块架构进行聚合,故MFFG的输出可表示如下:

也就是说,MFFG即可保持原始信息,亦可以分层方式突出结构信息。

3Experiments

在网络结构方面,CrossSRN有个MFFG模块,通道数为64。训练数据则采用了DIV2K,退化方式为BI。训练图像块尺寸为,数据增广为随机镜像与旋转。优化器选用了Adam,初始学习率为,合计训练1000epoch,每200epoch学习率减半。评估准则选用了常规的PSNR与SSIN,此外还统计了MACs(基于X4超分+输入统计)。

为更好的说明所提方案的结构复原能力,作者基于Urban100与Manga109构建了新的评估基准数据集。首先,采用高斯模糊核对图像进行模糊;然后采用Sobel算子提取图像边缘并通过阈值移除弱响应后计算图像的平均响应。将平均响应高于的图像收录到新的基准数据集中,它包含38个图像,其中18个源自Urban100,20源自Manga109。

Ablation Study

我们先对Cross卷积与多尺度特征融合机制进行消融实验分析。

上表给出了不同卷积形式的性能对比,从中可以看到:Cross卷积具有最佳的PSNR/SSIM指标,同时具有更低的计算量和参数量

上图给出了不同卷积结构的可视化对比,可以看到:经Cross卷积处理后边缘特征得到了进一步增强,验证了所提方案关于结构纹理复原的能力

上表对比了Seq与Cross两种形式不同卷积核尺寸下的性能对比,可以看到:相比Seq形式,Cross形式具有更佳的性能

上表对比了多尺度特征融合方面的性能,可以看到:

  • 相比无MFF形式的模块,所提MFFG特征融合机制具有相当的性能,同时参数量减半
  • 移除CCB模块后,模型性能大幅下降:0.1-0.2dB。

上图给出了MFFG中不同组的特征可视化,可以看到:经CCB处理后,边缘与纹理结构变得更为清晰锐利

上表对比了CA与FN对性能影响,可以看到:移除F-Norm会带来0.06dB@Manga109性能下降;移除CA会带来0.05dB@Urban100性能下降。

Comparison with SOTA Methods

上表给出了所提方案与其他方案的性能对比,可以看到:

  • CrossSRN取得了相当、甚至更优的性能,90%取得了最佳或次佳;
  • 相比DEGREE,CrossSRN具有更佳的性能;
  • 相比SeaNet,CrossSRN在所有尺度与基准数据上取得了90%最佳或次佳,同时参数量仅需43%。这说明:CrossSRN能够以更少参数量、更高效的进行结构复原。

上表对比了不同方法在新构建数据及上的性能对比,可以看到:**相比其他,CrossSRN具有更优的结构信息保持能力。**图像超分怎么可以没有图示呢,看看下面几个图示对比,显然:CrossSRN具有更好的结构纹理复原能力

最后,附上关于模型推理速度方面的对比,见下表。CrossSRN具有最佳指标,更快的速度(慢于CARN,但性能更佳)。

4吐槽时刻

我们先看一下笔者汇总的下面几个表吧,对比了CrossSRN、IMDN以及RFDN的性能(注:原文并没有提供RFDN的指标),很明显:CrossSRN被RFDN“摩擦”了吧,嗯嗯,差不多就是这个意思。

X2Set5Set14B100Urban100Manga109
CrossSRN38.03/0.960633.62/0.918032.19/0.899732.28/0.929038.75/0.9773
IMDN38.00/0.960533.63/0.917732.19/0.899632.17/0.928338.88/0.9774
RFDN38.05/0.960633.68/0.918432.16/0.899432.12/0.927838.88/0.9773
RFDN-L38.08/0.960633.67/0.919032.18/0.899632.24/0.929038.95/0.9773
X3Set5Set14B100Urban100Manga109
CrossSRN34.43/0.927530.33/0.841729.09/0.805028.23/0.853533.65/0.9448
IMDN34.36/0.927030.32/0.841729.09/0.804628.17/0.851933.61/0.9445
RFDN34.41/0.927330.34/0.842029.09/0.805028.21/0.852533.67/0.9449
RFDN-L34.47/0.928030.35/0.842129.11/0.805328.32/0.854733.78/0.9458
X4Set5Set14B100Urban100Manga109
CrossSRN32.24/0.895428.59/0.781727.58/0.736426.16/0.788130.53/0.9081
IMDN32.21/0.894828.58/0.781127.56/0.735326.04/0.783830.47/0.9084
RFDN32.24/0.895228.61/0.781927.57/0.736026.11/0.785830.58/0.9089
RFDN-L32.28/0.895728.61/0.781827.58/0.736326.20/0.788330.61/0.9096

借《飞驰人生》中一句“一顿操作猛如虎,仔细一看原地杵”,嗯嗯,可能说的就是CrossSRN。为啥这么说呢,看看下表几个模型的通道数、模块数等方面的信息对比。


通道数模块数通道注意力F-Norm
CrossSRN6410YY
IMDN646YN
RFDN486YN
RFDN-L526YN

无论是通道数,还是模块数,CrossSRN都要比RFDN多,但性能反而更低(当然,由于CrossSRN用的是Cross卷积,计算量会小一些,可能这也是它使用这么多模块数的原因吧)。美中不足的是,RFDN预训练模型并未全部开源(仅开源了一个参赛用的轻量模型)。

以上纯是吐槽,CrossSRN虽然性能未能真正达到SOTA,不过它提供的Cross卷积思路还是值得借鉴一下滴,对于轻量型超分架构设计还是有点参考价值的,有兴趣的小伙伴可以去尝试一下。

努力分享优质的计算机视觉相关内容,欢迎关注:

交流群


欢迎加入公众号读者群一起和同行交流,目前有美颜、三维视觉计算摄影、检测、分割、识别、医学影像、GAN算法竞赛等微信群


个人微信(如果没有备注不拉群!
请注明:地区+学校/企业+研究方向+昵称



下载1:何恺明顶会分享


AI算法与图像处理」公众号后台回复:何恺明,即可下载。总共有6份PDF,涉及 ResNet、Mask RCNN等经典工作的总结分析


下载2:终身受益的编程指南:Google编程风格指南


AI算法与图像处理」公众号后台回复:c++,即可下载。历经十年考验,最权威的编程规范!



下载3 CVPR2021

AI算法与图像处公众号后台回复:CVPR即可下载1467篇CVPR 2020论文 和 CVPR 2021 最新论文


浏览 43
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报