HiT-SR: Hierarchical Transformer for Efficient Image Super-Resolution
About
Transformers have exhibited promising performance in computer vision tasks including image super-resolution (SR). However, popular transformer-based SR methods often employ window self-attention with quadratic computational complexity to window sizes, resulting in fixed small windows with limited receptive fields. In this paper, we present a general strategy to convert transformer-based SR networks to hierarchical transformers (HiT-SR), boosting SR performance with multi-scale features while maintaining an efficient design. Specifically, we first replace the commonly used fixed small windows with expanding hierarchical windows to aggregate features at different scales and establish long-range dependencies. Considering the intensive computation required for large windows, we further design a spatial-channel correlation method with linear complexity to window sizes, efficiently gathering spatial and channel information from hierarchical windows. Extensive experiments verify the effectiveness and efficiency of our HiT-SR, and our improved versions of SwinIR-Light, SwinIR-NG, and SRFormer-Light yield state-of-the-art SR results with fewer parameters, FLOPs, and faster speeds ($\sim7\times$).
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Super-Resolution | Set14 4x (test) | PSNR28.87 | 117 | |
| Super-Resolution | Set5 x2 (test) | PSNR38.26 | 95 | |
| Image Super-resolution | Urban100 x4 (test) | PSNR26.8 | 90 | |
| Image Super-resolution | Urban100 x2 (test) | PSNR33.13 | 72 | |
| Image Super-resolution | Urban100 x3 (test) | PSNR28.99 | 58 | |
| Image Super-resolution | Manga109 x2 (test) | PSNR39.47 | 52 | |
| Super-Resolution | Manga109 x3 (test) | PSNR34.53 | 49 | |
| Super-Resolution | Set5 x4 (test) | PSNR32.55 | 46 | |
| Image Super-resolution | B100 x4 (test) | PSNR27.75 | 45 | |
| Image Super-resolution | Manga109 x4 (test) | PSNR31.26 | 44 |