EDSR:增强的深度超分辨率网络.
- paper:Enhanced Deep Residual Networks for Single Image Super-Resolution
- arXiv:link
1. 研究背景
作者通过改进残差网络结构训练了一个增强版本的超分辨率卷积网络。分析对比现有的残差结构:
作者认为ResNet等残差网络的提出是针对图像分类等high-level任务(更关注图像的语义特性),而不适合直接用于超分辨率等low-level任务(更关注图像的像素特性)。
相较于最初的ResNet,作者进行了如下改进:
- 去除了残差块之间的ReLU激活函数,仅在残差块内部使用;
- 去除了BatchNorm,作者认为其破坏了网络的range flexibility;这一步相比于SRResNet降低了训练时$40\%$的内存使用率。
2. 模型结构
作者提出了两种网络结构,分别是单尺度模型enhanced deep super-resolution network (EDSR)和多尺度模型multi-scale deep super-resolution system (MDSR)。
EDSR
通常网络层数为$B$、特征通道数为$F$的卷积神经网络需要$O(BF)$内存和$O(BF^2)$参数量。
作者设置\(B = 32, F = 256\)训练网络的残差连接部分。对于不同分辨率的要求,只需要对网络最后的上采样部分进行修改,可以共用网络残差部分的参数;这样可以加快模型的训练速度。
为了使训练更稳定,作者使用了residual scaling,对每个残差块的残差路径最后乘以(Mult)因子$0.1$。
MDSR
作者还提出一个多尺度模型,可同时训练不同放大倍率的图像。
3. 实验分析
实验设置:
- 训练集:DIV2K
- 测试集:Set5, Set14, Urban100, B100
- 评估指标:PSNR, SSIM
作者使用了self-ensemble策略,测试时通过翻转和旋转测试图像,总共得到$8$张测试图像,将其喂入网络后再通过反向翻转和旋转得到$8$张高分辨率图像,最终结果取其平均值。
模型的实验结果如下: