论文标题

剩余的本地特征网络,用于有效的超分辨率

Residual Local Feature Network for Efficient Super-Resolution

论文作者

Kong, Fangyuan, Li, Mingxi, Liu, Songwei, Liu, Ding, He, Jingwen, Bai, Yang, Chen, Fangmin, Fu, Lean

论文摘要

基于深度学习的方法在单图像超分辨率(SISR)中取得了出色的性能。但是,有效的超分辨率的最新进展集中在减少参数和失败的数量上,并且它们通过通过复杂的层连接策略来改善功能利用来汇总更强大的功能。这些结构对于达到更高的运行速度可能不是必需的,这使得它们难以将其部署到资源约束的设备上。在这项工作中,我们提出了一个新型的残留本地特征网络(RLFN)。主要思想是使用三个卷积层进行残留的本地特征学习来简化特征聚合,这在模型性能和推理时间之间取决了良好的权衡。此外,我们重新审视流行的对比损失,并观察到其特征提取器的中间特征的选择对性能有很大影响。此外,我们提出了一种新型的多阶段温暖启动训练策略。在每个阶段,都利用了先前阶段的预训练权重来改善模型性能。结合改进的对比损失和训练策略,该提议的RLFN在运行时胜过所有最先进的图像SR模型,同​​时维持SR的PSNR和SSIM。此外,我们赢得了NTIRE 2022高效超分辨率挑战的运行时赛道的第一名。代码将在https://github.com/fyan111/rlfn上找到。

Deep learning based approaches has achieved great performance in single image super-resolution (SISR). However, recent advances in efficient super-resolution focus on reducing the number of parameters and FLOPs, and they aggregate more powerful features by improving feature utilization through complex layer connection strategies. These structures may not be necessary to achieve higher running speed, which makes them difficult to be deployed to resource-constrained devices. In this work, we propose a novel Residual Local Feature Network (RLFN). The main idea is using three convolutional layers for residual local feature learning to simplify feature aggregation, which achieves a good trade-off between model performance and inference time. Moreover, we revisit the popular contrastive loss and observe that the selection of intermediate features of its feature extractor has great influence on the performance. Besides, we propose a novel multi-stage warm-start training strategy. In each stage, the pre-trained weights from previous stages are utilized to improve the model performance. Combined with the improved contrastive loss and training strategy, the proposed RLFN outperforms all the state-of-the-art efficient image SR models in terms of runtime while maintaining both PSNR and SSIM for SR. In addition, we won the first place in the runtime track of the NTIRE 2022 efficient super-resolution challenge. Code will be available at https://github.com/fyan111/RLFN.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源