Source-linked AI summary

DRCT: Saving Image Super-resolution away from Information Bottleneck

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

arXiv:2404.00722v5cs.CVcs.AI

TL;DR

Deep SISR models can show abrupt late-layer feature-map intensity drops that suggest spatial-information loss and an information bottleneck. DRCT adds dense connections within residual groups to stabilize information flow and enhance receptive fields. The authors report that it surpasses existing state-of-the-art methods without more parameters or sophisticated window-attention mechanisms.

  • Problem

    Deep SISR models can exhibit sharp late-layer feature-map intensity decreases, suggesting spatial-information loss and an information bottleneck; CNN receptive fields also limit capture of long-range dependencies.

    Source: §1 · p. 1
  • Method

    DRCT incorporates dense connections within residual groups of a Swin-Transformer-based SISR model to stabilize information flow and enhance receptive fields.

    Source: §1 · p. 2, §7 · p. 8
  • Results

    DRCT is reported to surpass existing state-of-the-art methods without sophisticated window-attention mechanisms or increased model parameters.

    Source: §7 · p. 8
  • Takeaways & Limitations

    The authors present dense-connected Swin-Transformer SISR as a way to achieve strong results with a simpler model design.

    Source: §1 · p. 2, §7 · p. 8

Abstract

from arXiv · show

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

1. Introduction

SISR reconstructs high-quality images from low-resolution inputs, but CNN receptive-field limits and late feature-map intensity drops suggest spatial-information loss in deep models. DRCT uses dense connections in residual groups to stabilize information flow and reports state-of-the-art performance while maintaining efficiency.

1. Introduction

  • CNN inductive bias limits long-range dependency capture because receptive fields scale with convolution parameters and kernel sizes.The authors say this may cause non-local spatial information to be neglected.Source: §1 · p. 1
  • Transformer-based SISR feature-map intensity often drops sharply near network ends, suggesting spatial-information loss and an information bottleneck.The authors describe this as a common phenomenon across datasets and relate it to constraints on model performance and potential.Source: §1 · p. 2
  • DRCT adds dense connections within residual groups to stabilize deep-feature information flow and address information bottlenecks.Its design also incorporates Swin Transformer layers and transition layers into each Residual Dense Group.Source: §1 · p. 2
  • DRCT integrates dense connections into a Swin-Transformer-based SISR model and is reported to achieve state-of-the-art performance while maintaining efficiency.The authors also describe its design as simpler than alternatives that expand receptive fields through more intricate architectures.Source: §1 · p. 2

2. Related works

Prior SISR work uses Transformer architectures to capture long-range dependencies, while auxiliary supervision and feature fusion offer ways to strengthen intermediate or multi-level representations. The section surveys these approaches and illustrates differing feature-map patterns in SwinIR, HAT, and DRCT.

2.1. Vision Transformer-based Super-Resolution

  • SwinIR, HAT, and related Transformer SISR models use varied attention and feature-extraction designs to improve receptive fields or restoration features.Examples include SwinIR’s local-window self-attention, UFormer’s local-enhancement block, and HAT’s channel attention with overlapping cross-attention.Source: §2.1 · p. 2

2.2. Auxiliary Supervision and Feature Fusion

  • Deep supervision adds intermediate prediction layers during training, while other auxiliary methods guide intermediate features with task-relevant metadata.The passage says appropriate auxiliary supervision can accelerate convergence and enhance efficiency and performance.Source: §2.2 · p. 2
  • Feature-fusion work combines multi-level or spatial and channel features, including through attention mechanisms in CNNs and Transformers.The cited examples include FPN, RTCS, SwinFusion, and channel attention in Transformers.Source: §2.2 · pp. 2–3
  • SwinIR and HAT feature-map intensities diminish toward deeper layers, whereas DRCT’s maps strengthen more steadily without obvious oscillations.The figure presents the three methods from top to bottom, with deeper network positions farther right.Source: Figure 2 · p. 3

3. Problem Statement

The paper frames information loss across consecutive layers as an information-bottleneck concern and relates it to deep SISR feature extraction. It argues that residual connections alone may not preserve spatial information within blocks, while dense connections can fuse information across residual groups.

3.1. Information Bottleneck Principle

  • The information-bottleneck principle describes possible information loss across consecutive layers and its potential association with vanishing gradients during parameter fitting and prediction.For SISR, the stated goal is to optimize a mapping that maximizes mutual information between HR and SR images.Source: §3.1 · p. 3

3.2. Spatial Information Vanish in Super-resolution

  • SISR networks typically comprise shallow feature extraction, deep feature extraction, and image reconstruction, with deep extraction differing across methods.The passage characterizes residual blocks as common components of deep feature extraction.Source: §3.2 · p. 3
  • Residual connections transmit spatial information between residual blocks, but information within a block may not preserve it, contributing to non-smooth feature-map intensity and bottlenecks in deep layers.The authors connect this condition with potential spatial-information loss and reduced data efficiency or more complex designs for higher performance.Source: §3.2 · pp. 3–4
  • Dense connections are described as fusing spatial information across residual groups, preserving high-frequency features and smoothing information transmission.The passage also attributes receptive-field enhancement to this integration of spatial information.Source: §3.2 · p. 4

4. Motivation

The motivation is to stabilize information flow within residual groups and expand receptive fields in Swin-Transformer-based SISR. Dense connections are proposed as an alternative to increasingly sophisticated window-attention designs.

4. Motivation

  • Dense-residual connections are proposed to aggregate multi-level spatial information and stabilize information flow within residual groups.The authors relate this motivation to prior work on stabilizing training with dense-residual connections.Source: §4 · p. 4
  • Adding dense connections within Swin-Transformer blocks is intended to enhance receptive fields while capturing long-range dependencies.The authors position this as a route to strong performance with simpler architectures or shallower SISR networks.Source: §4 · p. 4

5. Methodology

DRCT extracts shallow and deep features before reconstructing the super-resolved image by fusing those features. Its deep feature extractor uses Residual Dense Groups with Swin-Dense-Residual-Connected Blocks, and training follows a same-task progressive strategy.

5.1. Network Architecture

  • DRCT has three stages: shallow feature extraction, deep feature extraction, and image reconstruction that aggregates shallow and deep features.A 3 × 3 convolution extracts shallow features; the reconstruction function fuses high-frequency deep features with low-frequency shallow features.Source: §5.1 · p. 4

5.2. Deep Feature Extraction

  • Each Residual Dense Group uses Swin-Dense-Residual-Connected Blocks, combining Swin Transformer Layers with dense-residual connections for feature extraction.The deep feature extractor contains K Residual Dense Groups and a convolution layer for feature transition; Figure 3 depicts five consecutive blocks in each RDG.Source: §5.2 · pp. 4–5, §5.1 · p. 4, Figure 3 · p. 5

5.3. Same-task Progressive Training Strategy

  • Same-task progressive training pre-trains DRCT on ImageNet, fine-tunes it on specific datasets with L1 loss, then uses L2 loss to eliminate singular pixels and artifacts.The authors state that the final L2 stage further improves PSNR.Source: §5.3 · p. 5

6. Experiment Results

Across benchmark datasets, DRCT is reported to outperform peer methods, with DRCT-L bringing further improvement. The experiments also examine visual reconstruction, model complexity, and an NTIRE x4 challenge entry.

6.1. Dataset

  • Training data came from DF2K—800 DIV2K and 2,650 Flickr2K images—with bicubic LR generation at scales 2, 3, and 4; evaluation used five standard SISR benchmarks.The benchmarks are Set5, Set14, BSD100, Urban100, and Manga109.Source: §6.1 · p. 5

6.2. Implementation Details

  • Training used three phases: ImageNet pretraining, optimization on the given dataset, and an L2-loss phase for PSNR enhancement.DRCT matches HAT’s depth and width configuration, including six RDGs, six SDRCB units, and 180 intermediate channels.Source: §6.2 · pp. 5–7

6.3. Quantitative Results

  • DRCT outperforms peer methods on all benchmark datasets, and DRCT-L further improves results.The comparison evaluates PSNR and SSIM; its caption identifies the top three results and marks pretraining strategies.Source: Table 1 · p. 6, §6.3 · p. 7

6.4. Visual Comparison

  • On Urban100 x4, DRCT restores structures where competing methods look blurry; its fewer-parameter setting also has higher DI, indicating wider attention.The visual comparison marks image patches with red boxes; DI is an attribution-based measure of the range of involved pixels.Source: §6.4 · p. 7, Figure 4 · p. 7

6.5. Model Complexity

  • DRCT-L surpasses HAT-L across model-complexity metrics, while HAT and SwinIR performance curves approach a plateau as parameters scale.The analysis concerns x4 SISR on Urban100; the paper attributes DRCT’s reduced computational burden to dense connections that stabilize information flow.Source: §6.5 · p. 8, Table 2 · p. 8, Figure 6 · p. 8

6.6. NTIRE Image Super-Resolution (x4) Challenge

  • The NTIRE x4 entry used rotation-and-flip self-ensemble and HAT–DRCT output ensembling, with submissions in validation and testing.The challenge data comprise DIV2K, Flickr2K, and LSDIR; the supplied caption identifies PSNR and SSIM as validation and testing metrics.Source: §6.6 · p. 8, Table 3 · p. 8

7. Conclusion

DRCT addresses spatial-information loss and a possible information bottleneck as SISR networks deepen. It uses dense connections within residual blocks alongside shifted-window attention, and the paper reports that it surpasses existing state-of-the-art methods without more model parameters or sophisticated window attention mechanisms.

  • The paper identifies spatial-information loss during deeper SISR forward propagation as a possible information bottleneck that may limit performance.Source: §7 · p. 8
  • DRCT stabilizes information flow and enhances receptive fields by adding dense connections within residual blocks and combining them with shift-window attention.Source: §7 · p. 8
  • DRCT better focuses on global spatial information and surpasses existing state-of-the-art methods without sophisticated window attention mechanisms or increased model parameters.Source: §7 · p. 8
Loading 2404.00722v5…