HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

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

Dinh Phu Tran Dao Duy Hung Daeyoung Kim

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

Abstract

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.

Benchmarks

BenchmarkMethodologyMetrics
image-super-resolution-on-bsd100-2x-upscalingCPAT
PSNR: 32.64
SSIM: 0.9056
image-super-resolution-on-bsd100-2x-upscalingCPAT+
PSNR: 32.66
SSIM: 0.9058
image-super-resolution-on-bsd100-3x-upscalingCPAT
PSNR: 29.56
SSIM: 0.8174
image-super-resolution-on-bsd100-3x-upscalingCPAT+
PSNR: 29.59
SSIM: 0.8177
image-super-resolution-on-bsd100-4x-upscalingCPAT+
PSNR: 28.06
SSIM: 0.7532
image-super-resolution-on-bsd100-4x-upscalingCPAT
PSNR: 28.04
SSIM: 0.7527
image-super-resolution-on-manga109-2xCPAT+
PSNR: 40.59
SSIM: 0.9816
image-super-resolution-on-manga109-2xCPAT
PSNR: 40.48
SSIM: 0.9814
image-super-resolution-on-manga109-3xCPAT
PSNR: 35.66
SSIM: 0.9559
image-super-resolution-on-manga109-3xCPAT+
PSNR: 35.77
SSIM: 0.9563
image-super-resolution-on-manga109-4xCPAT+
PSNR: 32.85
SSIM: 0.9318
image-super-resolution-on-set14-2x-upscalingCPAT
PSNR: 34.91
SSIM: 0.9277
image-super-resolution-on-set14-2x-upscalingCPAT+
PSNR: 34.97
SSIM: 0.9280
image-super-resolution-on-set14-3x-upscalingCPAT
PSNR: 31.15
SSIM: 0.8557
image-super-resolution-on-set14-3x-upscalingCPAT+
PSNR: 31.19
SSIM: 0.8559
image-super-resolution-on-set14-4x-upscalingCPAT
PSNR: 29.34
SSIM: 0.7991
image-super-resolution-on-set14-4x-upscalingCPAT+
PSNR: 29.36
SSIM: 0.7996
image-super-resolution-on-set5-2x-upscalingCPAT
PSNR: 38.68
SSIM: 0.9633
image-super-resolution-on-set5-2x-upscalingCPAT+
PSNR: 38.72
SSIM: 0.9635
image-super-resolution-on-set5-3x-upscalingCPAT
PSNR: 35.16
SSIM: 0.9334
image-super-resolution-on-set5-3x-upscalingCPAT+
PSNR: 35.19
SSIM: 0.9335
image-super-resolution-on-set5-4x-upscalingCPAT+
PSNR: 33.24
SSIM: 0.9071
image-super-resolution-on-set5-4x-upscalingCPAT
PSNR: 33.19
SSIM: 0.9069
image-super-resolution-on-urban100-2xCPAT+
PSNR: 34.89
SSIM: 0.9487
image-super-resolution-on-urban100-2xCPAT
PSNR: 34.76
SSIM: 0.9481
image-super-resolution-on-urban100-3xCPAT+
PSNR: 30.63
SSIM: 0.8934
image-super-resolution-on-urban100-3xCPAT
PSNR: 30.52
SSIM: 0.8923
image-super-resolution-on-urban100-4xCPAT
PSNR: 28.22
SSIM: 0.8408
image-super-resolution-on-urban100-4xCPAT+
PSNR: 28.33
SSIM: 0.8425

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp