论文标题
结构增强图像超分辨率的梯度差异损失
Gradient Variance Loss for Structure-Enhanced Image Super-Resolution
论文作者
论文摘要
通过在图像空间中优化L1或L2损耗的图像空间中,单图超分辨率(SISR)领域的最新成功是通过优化深卷卷神经网络(CNN)实现的。但是,当接受这些损失功能训练时,模型通常无法恢复高分辨率(HR)图像中存在的尖锐边缘,因为该模型倾向于给出潜在的HR解决方案的统计平均值。在我们的研究过程中,我们观察到,由L1或L2损失训练的模型产生的图像的梯度图的方差明显低于原始高分辨率图像的梯度图。在这项工作中,我们建议通过引入结构增强的损失函数,创造的梯度差异(GV)损失并生成具有知觉愉悦细节的纹理来减轻上述问题。具体而言,在模型的训练期间,我们从目标的梯度图中提取斑块并生成输出,计算每个贴片的方差和形成方差图的方差,并为这两个图像。此外,我们将计算方差图之间的距离最小化,以强制执行模型以产生高方差梯度图,从而导致产生具有更锐利边缘的高分辨率图像。实验结果表明,GV损失可以显着改善现有图像超分辨率(SR)深度学习模型的结构相似性(SSIM)和峰值信噪比(PSNR)的性能。
Recent success in the field of single image super-resolution (SISR) is achieved by optimizing deep convolutional neural networks (CNNs) in the image space with the L1 or L2 loss. However, when trained with these loss functions, models usually fail to recover sharp edges present in the high-resolution (HR) images for the reason that the model tends to give a statistical average of potential HR solutions. During our research, we observe that gradient maps of images generated by the models trained with the L1 or L2 loss have significantly lower variance than the gradient maps of the original high-resolution images. In this work, we propose to alleviate the above issue by introducing a structure-enhancing loss function, coined Gradient Variance (GV) loss, and generate textures with perceptual-pleasant details. Specifically, during the training of the model, we extract patches from the gradient maps of the target and generated output, calculate the variance of each patch and form variance maps for these two images. Further, we minimize the distance between the computed variance maps to enforce the model to produce high variance gradient maps that will lead to the generation of high-resolution images with sharper edges. Experimental results show that the GV loss can significantly improve both Structure Similarity (SSIM) and peak signal-to-noise ratio (PSNR) performance of existing image super-resolution (SR) deep learning models.