TSD-SR: One-Step Diffusion with Target Score Distillation for Real-World Image Super-Resolution

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Dong, Linwei, Fan, Qingnan, Guo, Yihong, Wang, Zhonghao, Zhang, Qi, Chen, Jinwei, Luo, Yawei, Zou, Changqing
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913868347867136
author Dong, Linwei
Fan, Qingnan
Guo, Yihong
Wang, Zhonghao
Zhang, Qi
Chen, Jinwei
Luo, Yawei
Zou, Changqing
author_facet Dong, Linwei
Fan, Qingnan
Guo, Yihong
Wang, Zhonghao
Zhang, Qi
Chen, Jinwei
Luo, Yawei
Zou, Changqing
contents Pre-trained text-to-image diffusion models are increasingly applied to real-world image super-resolution (Real-ISR) task. Given the iterative refinement nature of diffusion models, most existing approaches are computationally expensive. While methods such as SinSR and OSEDiff have emerged to condense inference steps via distillation, their performance in image restoration or details recovery is not satisfied. To address this, we propose TSD-SR, a novel distillation framework specifically designed for real-world image super-resolution, aiming to construct an efficient and effective one-step model. We first introduce the Target Score Distillation, which leverages the priors of diffusion models and real image references to achieve more realistic image restoration. Secondly, we propose a Distribution-Aware Sampling Module to make detail-oriented gradients more readily accessible, addressing the challenge of recovering fine details. Extensive experiments demonstrate that our TSD-SR has superior restoration results (most of the metrics perform the best) and the fastest inference speed (e.g. 40 times faster than SeeSR) compared to the past Real-ISR approaches based on pre-trained diffusion priors.
format Preprint
id arxiv_https___arxiv_org_abs_2411_18263
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TSD-SR: One-Step Diffusion with Target Score Distillation for Real-World Image Super-Resolution
Dong, Linwei
Fan, Qingnan
Guo, Yihong
Wang, Zhonghao
Zhang, Qi
Chen, Jinwei
Luo, Yawei
Zou, Changqing
Computer Vision and Pattern Recognition
Pre-trained text-to-image diffusion models are increasingly applied to real-world image super-resolution (Real-ISR) task. Given the iterative refinement nature of diffusion models, most existing approaches are computationally expensive. While methods such as SinSR and OSEDiff have emerged to condense inference steps via distillation, their performance in image restoration or details recovery is not satisfied. To address this, we propose TSD-SR, a novel distillation framework specifically designed for real-world image super-resolution, aiming to construct an efficient and effective one-step model. We first introduce the Target Score Distillation, which leverages the priors of diffusion models and real image references to achieve more realistic image restoration. Secondly, we propose a Distribution-Aware Sampling Module to make detail-oriented gradients more readily accessible, addressing the challenge of recovering fine details. Extensive experiments demonstrate that our TSD-SR has superior restoration results (most of the metrics perform the best) and the fastest inference speed (e.g. 40 times faster than SeeSR) compared to the past Real-ISR approaches based on pre-trained diffusion priors.
title TSD-SR: One-Step Diffusion with Target Score Distillation for Real-World Image Super-Resolution
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.18263