AddSR: Accelerating Diffusion-based Blind Super-Resolution with Adversarial Diffusion Distillation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Xie, Rui, Zhao, Chen, Zhang, Kai, Zhang, Zhenyu, Zhou, Jun, Yang, Jian, Tai, Ying
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916542842667008
author Xie, Rui
Zhao, Chen
Zhang, Kai
Zhang, Zhenyu
Zhou, Jun
Yang, Jian
Tai, Ying
author_facet Xie, Rui
Zhao, Chen
Zhang, Kai
Zhang, Zhenyu
Zhou, Jun
Yang, Jian
Tai, Ying
contents Blind super-resolution methods based on stable diffusion showcase formidable generative capabilities in reconstructing clear high-resolution images with intricate details from low-resolution inputs. However, their practical applicability is often hampered by poor efficiency, stemming from the requirement of thousands or hundreds of sampling steps. Inspired by the efficient adversarial diffusion distillation (ADD), we design~\name~to address this issue by incorporating the ideas of both distillation and ControlNet. Specifically, we first propose a prediction-based self-refinement strategy to provide high-frequency information in the student model output with marginal additional time cost. Furthermore, we refine the training process by employing HR images, rather than LR images, to regulate the teacher model, providing a more robust constraint for distillation. Second, we introduce a timestep-adaptive ADD to address the perception-distortion imbalance problem introduced by original ADD. Extensive experiments demonstrate our~\name~generates better restoration results, while achieving faster speed than previous SD-based state-of-the-art models (e.g., $7$$\times$ faster than SeeSR).
format Preprint
id arxiv_https___arxiv_org_abs_2404_01717
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle AddSR: Accelerating Diffusion-based Blind Super-Resolution with Adversarial Diffusion Distillation
Xie, Rui
Zhao, Chen
Zhang, Kai
Zhang, Zhenyu
Zhou, Jun
Yang, Jian
Tai, Ying
Computer Vision and Pattern Recognition
Image and Video Processing
Blind super-resolution methods based on stable diffusion showcase formidable generative capabilities in reconstructing clear high-resolution images with intricate details from low-resolution inputs. However, their practical applicability is often hampered by poor efficiency, stemming from the requirement of thousands or hundreds of sampling steps. Inspired by the efficient adversarial diffusion distillation (ADD), we design~\name~to address this issue by incorporating the ideas of both distillation and ControlNet. Specifically, we first propose a prediction-based self-refinement strategy to provide high-frequency information in the student model output with marginal additional time cost. Furthermore, we refine the training process by employing HR images, rather than LR images, to regulate the teacher model, providing a more robust constraint for distillation. Second, we introduce a timestep-adaptive ADD to address the perception-distortion imbalance problem introduced by original ADD. Extensive experiments demonstrate our~\name~generates better restoration results, while achieving faster speed than previous SD-based state-of-the-art models (e.g., $7$$\times$ faster than SeeSR).
title AddSR: Accelerating Diffusion-based Blind Super-Resolution with Adversarial Diffusion Distillation
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2404.01717