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

Transcending the Limit of Local Window: Advanced Super-Resolution Transformer with Adaptive Token Dictionary

About

Single Image Super-Resolution is a classic computer vision problem that involves estimating high-resolution (HR) images from low-resolution (LR) ones. Although deep neural networks (DNNs), especially Transformers for super-resolution, have seen significant advancements in recent years, challenges still remain, particularly in limited receptive field caused by window-based self-attention. To address these issues, we introduce a group of auxiliary Adaptive Token Dictionary to SR Transformer and establish an ATD-SR method. The introduced token dictionary could learn prior information from training data and adapt the learned prior to specific testing image through an adaptive refinement step. The refinement strategy could not only provide global information to all input tokens but also group image tokens into categories. Based on category partitions, we further propose a category-based self-attention mechanism designed to leverage distant but similar tokens for enhancing input features. The experimental results show that our method achieves the best performance on various single image super-resolution benchmarks.

Leheng Zhang, Yawei Li, Xingyu Zhou, Xiaorui Zhao, Shuhang Gu• 2024

Related benchmarks

TaskDatasetResultRank
Super-ResolutionSet5
PSNR38.61
751
Image Super-resolutionManga109
PSNR40.37
656
Super-ResolutionUrban100
PSNR34.7
603
Super-ResolutionSet14
PSNR34.95
586
Image Super-resolutionSet5 (test)
PSNR38.61
544
Super-ResolutionB100 (test)
PSNR32.65
363
Super-ResolutionBSD100
PSNR32.65
313
Super-ResolutionManga109
PSNR39.51
298
Single Image Super-ResolutionUrban100 (test)
PSNR28.17
289
Super-ResolutionSet14 (test)
PSNR34.95
246
Showing 10 of 27 rows

Other info

Code

Follow for update