SwinFIR: Revisiting the SwinIR with Fast Fourier Convolution and Improved Training for Image Super-Resolution

Dafeng Zhang,Feiyu Huang,Shizhuo Liu,Xiaobing Wang,Zhezhu Jin

2023-09-24

Abstract:Transformer-based methods have achieved impressive image restoration performance due to their capacities to model long-range dependency compared to CNN-based methods. However, advances like SwinIR adopts the window-based and local attention strategy to balance the performance and computational overhead, which restricts employing large receptive fields to capture global information and establish long dependencies in the early layers. To further improve the efficiency of capturing global information, in this work, we propose SwinFIR to extend SwinIR by replacing Fast Fourier Convolution (FFC) components, which have the image-wide receptive field. We also revisit other advanced techniques, i.e, data augmentation, pre-training, and feature ensemble to improve the effect of image reconstruction. And our feature ensemble method enables the performance of the model to be considerably enhanced without increasing the training and testing time. We applied our algorithm on multiple popular large-scale benchmarks and achieved state-of-the-art performance comparing to the existing methods. For example, our SwinFIR achieves the PSNR of 32.83 dB on Manga109 dataset, which is 0.8 dB higher than the state-of-the-art SwinIR method.

Computer Vision and Pattern Recognition

What problem does this paper attempt to address?

The main goal of this paper is to improve image super-resolution (SR) technology, particularly addressing the limitations of Transformer-based methods in capturing global information. The authors propose a new model named SwinFIR, which is based on SwinIR and introduces Fast Fourier Convolution (FFC) to enhance the utilization of global information.

Specifically, SwinFIR addresses the problem through the following points:

1. **Introduction of the SFB Module**: SwinFIR introduces the Spatial Frequency Block (SFB), utilizing Fast Fourier Convolution (FFC) to capture global information, thereby overcoming the limitations of window-based and local attention strategies. The SFB module consists of two branches: the spatial branch and the frequency branch, which use traditional convolution operations and FFC respectively to extract features.

2. **Data Augmentation Techniques**: In addition to improvements in the model structure, the paper also explores various data augmentation techniques (such as channel shuffle, Mixup, etc.) to enhance the model's generalization ability.

3. **Feature Integration Method**: A new post-processing technique called feature integration is proposed. This method can integrate the results of multiple trained models without increasing training or testing time, thereby improving model performance without additional cost.

Experimental results show that SwinFIR achieves state-of-the-art performance on multiple benchmark datasets, particularly on the Manga109 and Urban100 datasets, with significant improvements in the PSNR metric compared to SwinIR and other existing methods. Furthermore, even without a pre-training phase, SwinFIR can achieve results comparable to or even better than pre-trained methods. This indicates that the proposed improvements effectively enhance the model's ability to capture global information, thereby improving the performance of the image super-resolution task.