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

DRCT: Saving Image Super-resolution away from Information Bottleneck

About

In recent years, Vision Transformer-based approaches for low-level vision tasks have achieved widespread success. Unlike CNN-based models, Transformers are more adept at capturing long-range dependencies, enabling the reconstruction of images utilizing non-local information. In the domain of super-resolution, Swin-transformer-based models have become mainstream due to their capability of global spatial information modeling and their shifting-window attention mechanism that facilitates the interchange of information between different windows. Many researchers have enhanced model performance by expanding the receptive fields or designing meticulous networks, yielding commendable results. However, we observed that it is a general phenomenon for the feature map intensity to be abruptly suppressed to small values towards the network's end. This implies an information bottleneck and a diminishment of spatial information, implicitly limiting the model's potential. To address this, we propose the Dense-residual-connected Transformer (DRCT), aimed at mitigating the loss of spatial information and stabilizing the information flow through dense-residual connections between layers, thereby unleashing the model's potential and saving the model away from information bottleneck. Experiment results indicate that our approach surpasses state-of-the-art methods on benchmark datasets and performs commendably at the NTIRE-2024 Image Super-Resolution (x4) Challenge. Our source code is available at https://github.com/ming053l/DRCT

Chih-Chung Hsu, Chia-Ming Lee, Yi-Shiuan Chou• 2024

Related benchmarks

TaskDatasetResultRank
Image Super-resolutionSet5 (test)
PSNR38.82
544
Single Image Super-ResolutionUrban100 (test)
PSNR35.11
289
Image Super-resolutionBSD100 (test)
PSNR32.76
216
Image Super-resolutionManga109 standard (test)
PSNR40.94
12
Image Super-resolutionSet14 standard (test)
PSNR35.24
6
Super-ResolutionReal Root Dataset (test)
BRISQUE45.33
6
Super-ResolutionSynthetic Root (test)
MSE18.14
4
Single Image Super-ResolutionNTIRE Image Super-Resolution x4 2024 (val)
PSNR31.182
1
Single Image Super-ResolutionNTIRE Image Super-Resolution x4 2024 (test)
PSNR31.1776
1
Showing 9 of 9 rows

Other info

Code

Follow for update