Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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$).

Xiang Zhang, Yulun Zhang, Fisher Yu• 2024

Related benchmarks

TaskDatasetResultRank
Super-ResolutionSet14 4x (test)
PSNR28.87
117
Super-ResolutionSet5 x2 (test)
PSNR38.26
95
Image Super-resolutionUrban100 x4 (test)
PSNR26.8
90
Image Super-resolutionUrban100 x2 (test)
PSNR33.13
72
Image Super-resolutionUrban100 x3 (test)
PSNR28.99
58
Image Super-resolutionManga109 x2 (test)
PSNR39.47
52
Super-ResolutionManga109 x3 (test)
PSNR34.53
49
Super-ResolutionSet5 x4 (test)
PSNR32.55
46
Image Super-resolutionB100 x4 (test)
PSNR27.75
45
Image Super-resolutionManga109 x4 (test)
PSNR31.26
44
Showing 10 of 26 rows

Other info

Code

Follow for update