BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

这篇具有很好参考价值的文章主要介绍了BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

目录

摘要:

网络架构:

具体流程:

BSConv:

ESDB——Efficient Separable Distillation Block:

 Blueprint Shallow Residual Block (BSRB):

ESA和CCA:

 实验:

BSRN的实现细节:

BSRN-S(比赛)的实现细节:

不同卷积分解的效果:

ESA和CCA的有效性:

 不同激活函数对比:

BSRN的有效性:

 和SOTA方法的对比:

视觉效果比较:

 BSRN-S for NTIRE2022 Challenge:

总结:


摘要:

单图像超分辨率(SISR)的最新进展已经取得了非凡的性能,但计算成本太高,无法应用于边缘设备。为了缓解这个问题,一些全新的并且有效的方案已经被提出。具有注意机制的卷积神经网络(CNN)因其效率和有效性而引起越来越多的关注。然而,在卷积操作中仍然存在冗余的部分

在本文中,作者提出了一种轻量级的单图像超分辨率网络,称为蓝图可分离残差网络BSRN——Blueprint Separable Residual Network,BSRN的设计灵感来自于残差特征蒸馏网络(RFDN)和蓝图可分离卷积(BSConv)。

它采用了与RFDN类似的架构,同时引入了一种更高效的蓝图浅残差块(blueprint shallow residual block, BSRB),即在RFDN的浅层残差块(shallow residual block, SRB)中使用BSConv替换标准卷积。此外,作者使用有效的ECA块和CCA块来增强模型的表示性能力。

大量的实验结果表明,与现有的高效SR方法相比,该方法在较少的参数量下取得了最佳的性能。此外,该方案在NTIRE 2022高效超分辨率挑战的子赛道1(model complexity track)上取得了第一名。

网络架构:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 显然,它继承了RFDN的结构,同样,它也是由四个阶段组成,浅层特征提取、深层特征提取、多层特征融合和重构输出。

具体流程:

ILR 输入、ISR输出,在预处理阶段,输入图像首先会被复制n次,然后concat这些图像:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 concat表示沿着通道维度进行操作,n是concat的图像数量,接下来,浅层特征提取部分映射输入图像到更高维特征空间:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 HSF 表示浅层特征提取模块,具体来说,使用BSConv实现浅层特征提取,BSConv由一个1x1卷积和一个逐通道卷积(DW-Conv)组成,然后,一系列堆叠的ESDB使用F0进行深度特征提取,逐步细化提取的特征,这个过程公式如下:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 Hk(·) 表示第k个ESDB块, Fk-1 and Fk 表示第KESDB块的输入和输出特征。

为了充分利用所有深度的特征,通过1 × 1卷积和GELU激活,将不同深度产生的特征进行融合与映射。然后,利用BSConv对特征进行细化。多层特征融合表示为

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 Hfusion(·) 表示融合模块并且Ffused 是聚合特征。

为了充分利用残差学习,利用了长跳跃连接,重构阶段的公式:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

Hrec(·) 表示重构模块,它是由一个标准的3x3卷积和一个亚像素卷积组成

利用L1损失:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 concat表示沿着通道维度进行操作,可能是为了便于后边的BSConv。

BSConv:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

这也是出自于一篇轻量级网络的研究,如MobileNet中,会有深度可分离卷积Depthwise Separable Convolution(DSCs),由depthwise(DW)和pointwise(PW)两个部分结合起来,用来提取特征图,虽然相比常规的卷积操作,其参数数量和运算成本比较低,但是深度可分卷积实际上执行的是跨内核的相关性,而不是内核内部的相关性。而研究发现内核内部的相关性占主导地位,因此有更大的潜力进行有效分离。所以,这个叫做“蓝图”可分的卷积就针对内核内部的相关性做了研究。

所谓“蓝图”,其实就是作者可视化了三个基础网络VGG-19、Inception V2、ResNet-50的卷积核,最后发现很多卷积核呈现相同的“视觉结构”。就好比这些卷积核是照着一个模子,然后加点线性变换产生的。然后作者就将这个模子称为“蓝图”(高大上的名字)。

具体而言,BSConvU是将一个标准的卷积分解为1x1卷积(PW)和一个逐通道卷积。它是深度可分离卷积(DSConv—逐通道、逐点)的逆向版本,BSConvU在许多情况下,在高效的标准卷积分离方面表现得更好。此外,作者还对BSConvU进行了变体操作——BSConvS。后续的实验也会验证不同卷积分解的效果。

关于“蓝图”,此处不进行过多的讲解,感兴趣可以找原文来看。

Rethinking depthwise separable convolutions: How intra-kernel correlations lead to improved mobilenets

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

ESDB——Efficient Separable Distillation Block:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读 

受到RFDNRFDB的启发,作者设计了一个ESDB它和RFDB的结构很像但是更加高效。ESDB通常由三个阶段组成。

信息蒸馏操作是通过卷积以一定比例压缩特征通道实现

在第一阶段,对于输入特征Fin, 特征蒸馏公式如下:
BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

DL 表示蒸馏层,它生成蒸馏特征,RL表示精修层(借鉴IMDB-R,它会逐步修复粗粒度特征。在特征压缩阶段,蒸馏的特征Fdistilled 1, Fdistilled 2, Fdistilled 3, Fdistilled 4 会进行concat,然后由一个1x1卷积进行压缩, 

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

Concat表示只沿着通道维度进行操作 Fcondensed 是压缩后的特征,Hlinear(·) 表示1x1的卷积。

最后一阶段,为了提高模型的表征能力并且保持它的高效性,引进了轻量的增强空间注意力块(ESA)和对比通道块CCA,二者分别从空间和通道的角度增强模型的能力。

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 Blueprint Shallow Residual Block (BSRB):

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 ESDB一个基本的模块就是BSRB,它是由一个BSConv、一个GELU激活函数以及一个跳跃连接组成。激活函数GELU被视为ReLU的一个更平滑的变体。在该网络中,作者也发现GELU性能要比ReLU和LReLU更好。 后续的实验会证明。

ESA和CCA:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

ESA和CCA的高效性在以往的方法中已经被证明过了。作者引入这两个模块。

ESA开始于一个1x1卷积去减少输入特征的通道维度,然后使用strided convolution and a strided max-pooling layerThe combination of strided convolution and max-pooling is widely used in image classification to quickly reduce the spatial dimensions at the beginning of the network. 去减少空间尺寸,根据组卷积去提取特征,使用基于插值的上采样恢复空间尺寸,值得注意的是,ESA中的卷积也是BSConvs(而传统的ESA中:先用1x1再用stride为2的stride convolution、再使用大小为7x7stride为3的max-pooling)以获得更好的效率。结合残差连接,提取的特征会进一步被1x1卷积处理去恢复通道尺寸,最后,特征矩阵会经由一个sigmoid函数生成,并且和原始输入特征相乘

CCA块是之前SR任务中通道注意力模块的升级版。与传统的利用每个信道方向特征的均值计算的信道注意不同,CCA利用包括均值和标准差之和在内的对比度信息来计算信道注意权重。

 实验:

BSRN的实现细节:

BSRN是由8个ESDB组成,通道数是64,所有深度卷积核设置为3,mini-batch大小是64并且LR输入图像的patch为48x48,1 × 106 (6是指数)iterations.

使用了两个GeForce RTX 3090gpu和训练约30小时。

BSRN-S(比赛)的实现细节:

NTIRE2022挑战的实施细节。BSRN-s是为挑战而设计的BSRN的一个小变体,它要求参与者设计一个高效的网络,同时在DIV2K验证数据集上保持PSNR为29.00 dB。

具体来说,将ESDB的数量减少到5个,并将特征数减少到48个。CCA块被可学习的通道级权所取代。在训练过程中,输入patch大小设置为64×64,mini-batch为256。训练迭代次数增加到1.5×106(6是指数),有四个GeForce RTX 2080Tigpu用于训练。

不同卷积分解的效果:

作者展现了基于RFDN的不同卷积分解方法的影响,DSConv代表了原始的深度可分离卷积, 

可以观察到,当执行卷积分解时,性能下降是存在的,不过计算量和参数量也显著下降。为了平衡性能和参数量、计算量的关系,尽量在效果差的不多的情况下,选择了BSConvU 。

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

ESA和CCA的有效性:

没有ESA,虽然减少了9%的参数量,但有一个明显的性能降低。

没有CCA,BSRN性能也降低了,所以ESA和CCA有效。

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 不同激活函数对比:

之前的大部分工作都会选择ReLU 或者LeakyReLU GELU 是最近才开始流行的。作者们也探索了不同的激活函数选择最适合该模型的。同时也对比了MobileNet V3(轻量)中提出的新的激活函数h-swish。结果表明,不同的激活函数有着很显著的差别,在这些激活函数中,GELU 相比于其他的函数,在Urban100数据集上有着很好的表现

 BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

BSRN的有效性:

作者们设计了BSRN的两个变体进而证明提出架构的有效性,设置BSRN的深度和宽度都和原始的RFDN相同叫做BSRN-1,然后将模型容量扩大到与RFDN相似的计算量叫BSRN-2。

BSRN-1凭借更少的计算超过了RFDN此外,BSRN-2相比于RFDN获得了一个更有意义的性能提升,特别是在Manga109数据集。

实验结果表明提出的架构是更优的。

 BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 和SOTA方法的对比:

 BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

视觉效果比较:

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读 

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 

 BSRN-S for NTIRE2022 Challenge:

’Val PSNR’ and ’Test PSNR’ are PSNR results tested on the DIV2K validation and test sets。与其他方法相比,作者的方法参数数最少,FLOPs第二少。

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

就运行时间而言,这与代码的优化和针对不同操作人员的具体测试平台的计算有关。经过优化后,BSRN-S-opt在相同GPU上获得了与IMDN和RFDN相似的运行时长。 然而,由于GPU计算深度卷积的不友好性,该方法的运行时长依旧较大。

 BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 

总结:

BSRN通过使用蓝图可分离卷积BSConv构造基本构建块,减少冗余。BSConv是深度可分离卷积DSConv的升级版本,它更好地利用内核内部相关性来实现高效分离(深度可分卷积实际上执行的是跨内核的相关性,而不是内核内部的相关性。而研究发现内核内部的相关性占主导地位,因此有更大的潜力进行有效分离。该网络表明BSConv 是有益于有效SR的。其次,合适的注意力模块已经被证明可以提升高效SR网络的性能。受到以往工作的鼓舞,作者也引进了两个有效的注意力块去增强模型的能力,增强的空间注意力(ESA)和对比通道注意力(CCA)。图1展示了BSRN方法取得了SOTA性能。

BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读

 本文的主要贡献:

1引进了BSConv去构建基本的构建块并且展示它对SR的有效性。

2利用两个有效的注意力模块和有限的额外计算来增强模型的能力。文章来源地址https://www.toymoban.com/news/detail-452513.html

到了这里,关于BSRN网络——《Blueprint Separable Residual Network for Efficient Image Super-Resolution》论文解读的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包