ResDiff: Combining CNN and Diffusion Model for Image Super-Resolution

Abstract

Adapting the Diffusion Probabilistic Model (DPM) for direct image super-resolution is wasteful, given that a simple Convolutional Neural Network (CNN) can recover the main low-frequency content. Therefore, we present ResDiff, a novel Diffusion Probabilistic Model based on Residual structure for Single Image Super-Resolution (SISR). ResDiff utilizes a combination of a CNN, which restores primary low-frequency components, and a DPM, which predicts the residual between the ground-truth image and the CNN predicted image. In contrast to the common diffusion-based methods that directly use LR space to guide the noise towards HR space, ResDiff utilizes the CNN’s initial prediction to direct the noise towards the residual space between HR space and CNN-predicted space, which not only accelerates the generation process but also acquires superior sample quality. Additionally, a frequency-domain-based loss function for CNN is introduced to facilitate its restoration, and a frequency-domain guided diffusion is designed for DPM on behalf of predicting high-frequency details. The extensive experiments on multiple benchmark datasets demonstrate that ResDiff outperforms previous diffusion based methods in terms of shorter model convergence time, superior generation quality, and more diverse samples.

Cite

Text

Shang et al. "ResDiff: Combining CNN and Diffusion Model for Image Super-Resolution." AAAI Conference on Artificial Intelligence, 2024. doi:10.1609/AAAI.V38I8.28746

Markdown

[Shang et al. "ResDiff: Combining CNN and Diffusion Model for Image Super-Resolution." AAAI Conference on Artificial Intelligence, 2024.](https://mlanthology.org/aaai/2024/shang2024aaai-resdiff/) doi:10.1609/AAAI.V38I8.28746

BibTeX

@inproceedings{shang2024aaai-resdiff,
  title     = {{ResDiff: Combining CNN and Diffusion Model for Image Super-Resolution}},
  author    = {Shang, Shuyao and Shan, Zhengyang and Liu, Guangxing and Wang, Lunqian and Wang, Xinghua and Zhang, Zekai and Zhang, Jinglin},
  booktitle = {AAAI Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {8975-8983},
  doi       = {10.1609/AAAI.V38I8.28746},
  url       = {https://mlanthology.org/aaai/2024/shang2024aaai-resdiff/}
}