《Deep Learning for Image Super-resolution: A Survey》之 Loss Functions

最新推荐文章于 2023-01-09 17:30:25 发布

jinfeng2411

最新推荐文章于 2023-01-09 17:30:25 发布

阅读量1.3k

点赞数 2

分类专栏：论文阅读

本文链接：https://blog.csdn.net/qq_34706871/article/details/103196026

版权

论文阅读专栏收录该内容

16 篇文章 3 订阅

订阅专栏

本文节选自论文：Deep Learning for Image Super-resolution: A Survey

在图像超分中，用损失函数来衡量重建图像与真实图像之间的差异，并以此来引导模型的优化。

关于超分中常用的损失函数（Loss Functions）的总结：

1、Pixel Loss（基于像素级的损失）

包含 loss（平均绝对值误差）、 loss（均方误差）

其中， $\hat{I}$ 为重建图像，为真实图像，、、分别为图像的高、宽和通道数。

loss还有一个变种，称为 Charbonnier loss，其表达式如下：

其中， $\epsilon$ 是为了数值稳定而设的一个很小的常量。

在实际应用中， loss 比 loss 有着更好的效果和收敛性。因为图像重建的评价指标之一——峰值信噪比（PSNR）与像素级的差异有关，所以最小化 Pixel loss 能够直接最大化 PSNR ，因此 Pixel loss也就成为了一个广泛使用的损失函数。

缺点：Pixel loss 没有考虑到图像的感知质量（perceptual quality）和结构信息（textures），缺少高频信息，容易造成重建图像过度光滑。

2、Content Loss（基于内容的损失）

为了提高图像的感知质量（perceptual quality），Content loss 采用一个已经训练好的图像分类网络去衡量重建图像与真实图像之间的语义差异（semantic differences）。假设采用的、已经训练好的分类网络为 $\Phi$ ， $\Phi ^{(l)}(I)$ 表示图像在 $\Phi$ 的第层上的高阶表示（high-level representations）（其实，就是把输入到 $\Phi$ 中，然后在第层上的输出特征图（这个称为语义信息）），则 content loss 可以表示如下：

也就是衡量 $\hat{I}$ 与在网络 $\Phi$ 中第层上的特征图的差异（语义差异）。

content loss 产生的图像有更好的感知效果（人眼看的效果），常用的网络 $\Phi$ 有 VGG 和 ResNet。

3、Texture Loss（基于结构/纹理的损失）

重建图像应该与真实图像具有相同的样式（如颜色、纹理、对比度等）

与上面 Content Loss 有点类似，Texture loss也是针对特征图的计算。图像纹理被认为是不同特征通道之间的相关性，用Gram matrix $G_{ij}^{(l))}(I)$ 来表示在 $\Phi$ 网络上第层的特征图的第个通道与第个通道之间的这种相关性。

优点：能产生更加真实的纹理结构和符合视觉的效果。

缺点：确定去匹配纹理结构的图像块的大小仍以经验为主。

4、Adversarial Loss（对抗损失）

我们将 SR 模型看成是一个生成器，再加一个判别器去分辨重建图像和真实图像。

SRGAN采用的损失函数如下：

第一个为生成器（SR model）的对抗损失，第二个为判别器的损失。 $I_{s}$ 是从真实图像中随机采样的点。

后面又有如下的损失函数：

相比于SRGAN，它能使训练更稳定并且产生更好的效果。

与上面的工作不同的是，Park et al. 认为基于像素层面的判别器只会让生成器生成一些无意义的高频噪声，使用特征层面的判别器将会使重建图像更加真实（之前传给判别器的是图像，这里是让图像先通过一个预训练的网络，再将网络的某层的特征图（high-level representations）传给判别器）。

5、Cycle Consistency Loss（循环一致性损失）

这是受到CycleGAN的启发，对于一幅图像，先重建出它的图像 $\hat{I}$ ，再将 $\hat{I}$ 通过一个CNN网络下采样成图像 {I}' ,比较 {I}' 与之间的差异，计算公式如下：