【文献阅读2020】像素级自适应学习的超分辨率PixelLevelSelfPacedLearningForSuperResolution

原文链接
github

摘要&＃xff1a;近年来&＃xff0c;基于深度网络的预测超分辨率图像在多个图像领域得到了广泛的应用。然而&＃xff0c;随着这些网络的构建越来越深入&＃xff0c;它们的训练时间也越来越长&＃xff0c;这可能会引导学习者进行局部优化。为了解决这一问题&＃xff0c;本文设计了一种像素级自适应学习(PSPL,PIXEL-LEVEL SELF-PACED LEARNING)训练策略来加速SISR模型的收敛速度。PSPL模拟自定步长学习&＃xff0c;将预测的SR图像中的每个像素及其对应的ground truth中的像素赋予一个注意权值&＃xff0c;引导模型在参数空间中找到一个更好的区域。大量的实验证明&＃xff0c;PSPL可以加速SISR模型的训练&＃xff0c;并促使现有的几个模型获得新的更好的结果。

总结&＃xff1a; 这篇文章不是提出一个崭新的单张图像超分辨率模型&＃xff0c;而是提出一个加速训练原有单张图像超分辨率模型的手段。 文章认为&＃xff0c;之前的SRCNN、SRResNet、EDSR等模型确实有很好的表现&＃xff0c;但是它们太大、太复杂&＃xff0c;无法有效的训练&＃xff08;图1&＃xff09;他们可能会在训练上花费更多的时间&＃xff1b;而且&＃xff0c;有些模型即使训练了很长时间&＃xff0c;也可能没有产生更好的结果。

博主认为&＃xff0c;如果说其他超分辨率模型致力于前向传播的网络结构搭建&＃xff0c;也就是如何学习从LR到SR的映射&＃xff1b;那么&＃xff0c;这篇文章关注点在于根据其他模型训练好的SR&＃xff0c;如何利用SR和HR的关系来进行决定下一步的训练&＃xff0c;即训练策略的问题。

不同于以往根据SR和HR&＃xff0c;往往直接计算loss&＃xff0c;然后进行反向传播&＃xff0c;作者设计一个训练策略来加速这些模型的训练速度&＃xff0c;进一步增强它们的反向传播性能 。

这篇文章提出了一个叫做 像素级自适应学习(PSPL) 的训练策略。其中每个样本都有一个学习权值&＃xff0c;引导学习者学习更有效的特征&＃xff0c;更快的收敛&＃xff0c;更好的泛化。实际上&＃xff0c;本文介绍的就是每次训练之后得到的SR和HR&＃xff0c;先进行了一系列处理&＃xff0c;然后再计算loss。 博主认为&＃xff0c;这篇文章讲的是一种训练策略&＃xff0c;这个过程并不在前向传播中。换句话说&＃xff0c;文章是通过改变每次训练后的HR和SR&＃xff0c;来改变loss的值&＃xff0c;进一步影响反向传播中模型参数的变化的。

图2说明了PSPL如何切断和重新安排数据流。为了便于学习&＃xff0c;每个训练阶段分为以下四个步骤。在给定LR图像和HR图像的情况下&＃xff1a;
1.首先通过模型预测SR图像。
2.然后根据SR和HR图像生成相似度图&＃xff08;similarity map&＃xff09;。
3.再次&＃xff0c;在相似度图的基础上生成注意图&＃xff08;attention map&＃xff09;&＃xff0c;注意图对这些差异较大的像素对给予了更多的注意。
4.最后&＃xff0c;通过注意图与原始SR(HR)图像的逐项相乘得到新的SR(HR)图像&＃xff0c;并在计算损耗时将原始SR(HR)图像替换为新的SR(HR)图像&＃xff0c;从而影响SISR模型的优化。此外&＃xff0c;在整个训练过程中&＃xff0c;随着训练步骤的增加&＃xff0c;注意图中的所有值都将趋近于一个常数&＃xff0c;这意味着PSPL在整个训练过程中的效用逐渐降低。

PSPL的名称来自两个方面。一是随着训练时间的延长&＃xff0c;所有注意力的重量逐渐降低&＃xff0c;这与自主学习增加学习对象难度的过程相似;另一种是PSPL为图像中的每个像素分配注意权值&＃xff0c;这与传统的样本级自定步长学习中为样本分配权值不同。

设SR代表超分辨率结果&＃xff0c;和HR同尺寸。
1.学习LR到HR的映射F&＃xff0c;得到SR&＃xff08;也就是前人提出的SISR方法&＃xff09;&＃xff1b;
2.根据SR和HR图像生成相似度图&＃xff08;similarity map&＃xff09;。
(1)SR和HR对应的 patches(psp_sps​&＃xff0c;php_hph​)分别和圆对称高斯权重矩阵&＃xff08;a circular-symmetric Gaussian weighting matrix&＃xff09;做点乘&＃xff08;element-wise multiplication&＃xff09;&＃xff0c;得到 ps^\hat{p_s}ps​^​ , ph^\hat{p_h}ph​^​ &＃xff1b;

(2)根据ps^\hat{p_s}ps​^​ , ph^\hat{p_h}ph​^​ 的均值、方差&＃xff0c;根据SSIM计算公式得到每个patch的 msm_sms​&＃xff0c;然后得到整张图片的 MsM_sMs​ .
$C$里的两个参数k1k_1k1​ k2k_2k2​是人工设置的
使用SSIM的好处&＃xff1a;

首先&＃xff0c;SSIM是一种基于感知的准则&＃xff0c;它在空间上是平稳的。 其次&＃xff0c;SSIM能够利用其相邻像素值&＃xff0c;使其与绝对差异相比较更稳定

。

3.生成注意力图Attention Map

由 MsM_sMs​经过一个高斯函数G&＃xff0c;产生 MaM_aMa​ &＃xff1a;
这个高斯函数的三个参数 $\gamma$ $\mu$ $\delta$分别代表峰值大小 峰值的位置 和 宽度。 在PSPL中&＃xff0c;G更多地关注具有较小相似性的这些像素对. 这些参数是人工选择的。这些参数在训练过程中都是常数。

下面是某个patch的输入SSIM指数&＃xff08;msm_sms​&＃xff09;和输出Attention Map&＃xff08;mam_ama​&＃xff09;以及训练阶段的变化图:

4.Loss function

将Attention Map&＃xff08;MaM_aMa​&＃xff09;分别和SR HR做点乘&＃xff08;Fig2 绿色实线&＃xff09;&＃xff0c;对得到的两个结果做Loss。

由于MsM_sMs​到 MaM_aMa​ 只是为了得到新的SR&＃xff0c;HR来计算Loss&＃xff0c;不在SISR网络内部&＃xff0c;因此这一过程不参与反向传播&＃xff0c;只是用于训练中。毕竟前面已经强调了&＃xff0c;这篇文章讲的是一种训练策略&＃xff0c;这个过程并不在前向传播中。换句话说&＃xff0c;文章是通过改变HR和SR&＃xff0c;来改变loss的值&＃xff0c;进一步影响反向传播中模型参数的变化的。

结果

不同模型下&＃xff0c;是否应用该训练策略的效果对比&＃xff1a;