【图像超分辨】EnhanceNet
ICCV2017年的“EnhanceNet: Single Image Super-Resolution Through Automated Texture Synthesis”是与SRGAN同期的研究工作,有相似之处,使用perceptual loss提高生成图像的真实性,并使用GAN进行训练,但EnhanceNet更强调纹理统计特性的局部匹配可以产生更加真实纹理,减少不真实的伪影而没有额外使用正则化技术。
网络结构
网络结构采用全卷积的方式,使得输入图像可以是任意尺寸。受到VGG网络的启发,卷积核全部采用3*3的尺寸,在保持一定量参数的情况下构建更深的网络。网络的输入是低分辨率图像,在网络末端采用最近邻的方法上采样达到高分辨率图像的尺寸,这样有利于降低计算复杂度,参数的初始化采用Xavier,网络的学习目标是超分辨率图像与输入的低分辨率图像线性插值的差,整个网络架构如下图所示:
损失函数
pixel-wise loss:
是网络的输出结果,是图像的原始高分辨率图像。那么pixel-wise MSE定义为:
pixel-wise loss强调的是两幅图像之间每个对应像素的匹配,这与人眼的感知结果有所区别。通过pixel-wise loss训练的图片通常会较为平滑,缺少高频信息。即使输出图片具有较高的PSNR,视觉效果也并没有很突出。
perceptual loss:
我们避免了要求网络输出图像与原始高分辨率图像pixel-wise上的一致,而是鼓励两幅图具有相似的特征。
对于函数Φ,这篇论文采用了预训练好的VGG-19网络,通过VGG-19网络来提取出图像的有效特征。为了提取出图像的高层和低层特征,我们使用VGG-19的第二和第五pooling层来获取图像的有效特征,在这基础上计算它们的MSE。
texture matching loss:
texture matching loss 可以促进产生的图像具有更丰富的纹理信息。通过计算如下公式来计算texture matching loss:
其中,Φ是用于特征提取的函数(论文中采用了VGG-19网络)。
在具体的操作层面,基于patch来计算texture matching loss以此来保证纹理信息在局部上的一致。实验表明,patch大小选为16*16的像素时,效果最好。如果在整幅图上计算texture matching loss,由于纹理信息的多样性,必将导致平均化,效果会相应地变差。
Adversarial training:
生成对抗网络包含两个网络,其中一个是生成网络G,另一个是判别网络D。G用于接收噪声Z并通过产生数据分布,判别网络则是用来判断x来自于真实的数据分布还是。两个网络通过不断地对抗学习,最终达到G网络产生的数据分布,此时判别网络D已无法分辨x来自于哪一个数据集。
D和G两者相互对抗,优化如下的最小最大公式:
其中的max指寻找最优的判别器D,min则是在此基础上优化生成器G,使得G产生的数据分布更接近,以致于迷惑D。
在具体的训练过程中,通常是优化判别器K次,才会优化生成器D一次。这样可以保证判别器D始终处于或接近最优。
在训练初期,由于与相差太大,D可以很轻松地分辨出数据来自哪个数据集,导致不能够提供足够的梯度来训练G。通过将最小化改为最大化,便可以获得足够的梯度来训练G网络。
Adversarial loss:
利用GAN网络来实现超分辨率,首先把图一中的网络结构用于GAN网络中的生成网络G,G的目标是最小化:
对应的判别网络则是最小化:
这样的设计使得训练效果更好。
其中,判别网络结构如下图所示:
训练细节
具体的训练方面,对于判别网络,采用leaky ReLU作为**函数,strided convolutions来逐渐减少图像的空间维度。在整个判别网络的训练过程中,不采用dropout。当判别网络对以及的判别结果低于一定阈值的时候,才开始训练判别网络。
实验结果
将不同的loss计算方式结合在一起,训练结果也会相应地不同。图三中展示了不同的loss组合方式所展现的不同的超分辨率效果,其中E指MSE,P指perceptual loss,A指Adversarial loss,T指texture loss。
相比ENet-E(仅采用MSE作为loss),Perceptual loss的训练结果具有较为锐化的细节,但是它在纹理区域会产生较多的噪声。ENet-PA(采用perceptual loss和adversarialloss)所获得的图像具有很多的高频信息,也显得更为锐化。但是有时会在平滑区域误生成很多高频噪声。ENet-PAT(采用perceptual loss,adversarial loss和texture loss)的效果最好,能够产生合理有意义的纹理,极大减少artifact,许多时候甚至可以达到原图的效果。