Channel-Partitioned Windowed Attention And Frequency Learning for Single Image Super-Resolution

23 Jul 2024  ·  Dinh Phu Tran, Dao Duy Hung, Daeyoung Kim ·

Recently, window-based attention methods have shown great potential for computer vision tasks, particularly in Single Image Super-Resolution (SISR). However, it may fall short in capturing long-range dependencies and relationships between distant tokens. Additionally, we find that learning on spatial domain does not convey the frequency content of the image, which is a crucial aspect in SISR. To tackle these issues, we propose a new Channel-Partitioned Attention Transformer (CPAT) to better capture long-range dependencies by sequentially expanding windows along the height and width of feature maps. In addition, we propose a novel Spatial-Frequency Interaction Module (SFIM), which incorporates information from spatial and frequency domains to provide a more comprehensive information from feature maps. This includes information about the frequency content and enhances the receptive field across the entire image. Experimental findings show the effectiveness of our proposed modules and architecture. In particular, CPAT surpasses current state-of-the-art methods by up to 0.31dB at x2 SR on Urban100.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Image Super-Resolution BSD100 - 2x upscaling CPAT+ PSNR 32.66 # 9
SSIM 0.9058 # 8
Image Super-Resolution BSD100 - 2x upscaling CPAT PSNR 32.64 # 10
SSIM 0.9056 # 9
Image Super-Resolution BSD100 - 3x upscaling CPAT PSNR 29.56 # 6
SSIM 0.8174 # 5
Image Super-Resolution BSD100 - 3x upscaling CPAT+ PSNR 29.59 # 4
SSIM 0.8177 # 3
Image Super-Resolution BSD100 - 4x upscaling CPAT+ PSNR 28.06 # 5
SSIM 0.7532 # 10
Image Super-Resolution BSD100 - 4x upscaling CPAT PSNR 28.04 # 8
SSIM 0.7527 # 11
Image Super-Resolution Manga109 - 2x upscaling CPAT PSNR 40.48 # 8
SSIM 0.9814 # 7
Image Super-Resolution Manga109 - 2x upscaling CPAT+ PSNR 40.59 # 7
SSIM 0.9816 # 5
Image Super-Resolution Manga109 - 3x upscaling CPAT+ PSNR 35.77 # 5
SSIM 0.9563 # 4
Image Super-Resolution Manga109 - 3x upscaling CPAT PSNR 35.66 # 7
SSIM 0.9559 # 6
Image Super-Resolution Manga109 - 4x upscaling CPAT+ PSNR 32.85 # 7
SSIM 0.9318 # 6
Image Super-Resolution Set14 - 2x upscaling CPAT+ PSNR 34.97 # 6
SSIM 0.9280 # 6
Image Super-Resolution Set14 - 2x upscaling CPAT PSNR 34.91 # 9
SSIM 0.9277 # 7
Image Super-Resolution Set14 - 3x upscaling CPAT+ PSNR 31.19 # 6
SSIM 0.8559 # 5
Image Super-Resolution Set14 - 3x upscaling CPAT PSNR 31.15 # 7
SSIM 0.8557 # 6
Image Super-Resolution Set14 - 4x upscaling CPAT PSNR 29.34 # 9
SSIM 0.7991 # 12
Image Super-Resolution Set14 - 4x upscaling CPAT+ PSNR 29.36 # 7
SSIM 0.7996 # 10
Image Super-Resolution Set5 - 2x upscaling CPAT+ PSNR 38.72 # 6
SSIM 0.9635 # 6
Image Super-Resolution Set5 - 2x upscaling CPAT PSNR 38.68 # 8
SSIM 0.9633 # 7
Image Super-Resolution Set5 - 3x upscaling CPAT PSNR 35.16 # 7
SSIM 0.9334 # 7
Image Super-Resolution Set5 - 3x upscaling CPAT+ PSNR 35.19 # 5
SSIM 0.9335 # 5
Image Super-Resolution Set5 - 4x upscaling CPAT PSNR 33.19 # 4
SSIM 0.9069 # 4
Image Super-Resolution Set5 - 4x upscaling CPAT+ PSNR 33.24 # 3
SSIM 0.9071 # 3
Image Super-Resolution Urban100 - 2x upscaling CPAT+ PSNR 34.89 # 5
SSIM 0.9487 # 5
Image Super-Resolution Urban100 - 2x upscaling CPAT PSNR 34.76 # 7
SSIM 0.9481 # 6
Image Super-Resolution Urban100 - 3x upscaling CPAT PSNR 30.52 # 6
SSIM 0.8923 # 5
Image Super-Resolution Urban100 - 3x upscaling CPAT+ PSNR 30.63 # 5
SSIM 0.8934 # 4
Image Super-Resolution Urban100 - 4x upscaling CPAT PSNR 28.22 # 8
SSIM 0.8408 # 8
Image Super-Resolution Urban100 - 4x upscaling CPAT+ PSNR 28.33 # 7
SSIM 0.8425 # 7

Methods