Co-learning Single-Step Diffusion Upsampler and Downsampler with Two Discriminators and Distillation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kim, Sohwi, Kim, Tae-Kyun
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917957517443072
author Kim, Sohwi
Kim, Tae-Kyun
author_facet Kim, Sohwi
Kim, Tae-Kyun
contents Super-resolution (SR) aims to reconstruct high-resolution (HR) images from their low-resolution (LR) counterparts, often relying on effective downsampling to generate diverse and realistic training pairs. In this work, we propose a co-learning framework that jointly optimizes a single-step diffusion-based upsampler and a learnable downsampler, enhanced by two discriminators and a cyclic distillation strategy. Our learnable downsampler is designed to better capture realistic degradation patterns while preserving structural details in the LR domain, which is crucial for enhancing SR performance. By leveraging a diffusion-based approach, our model generates diverse LR-HR pairs during training, enabling robust learning across varying degradations. We demonstrate the effectiveness of our method on both general real-world and domain-specific face SR tasks, achieving state-of-the-art performance in both fidelity and perceptual quality. Our approach not only improves efficiency with a single inference step but also ensures high-quality image reconstruction, bridging the gap between synthetic and real-world SR scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2410_07663
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Co-learning Single-Step Diffusion Upsampler and Downsampler with Two Discriminators and Distillation
Kim, Sohwi
Kim, Tae-Kyun
Image and Video Processing
Computer Vision and Pattern Recognition
Super-resolution (SR) aims to reconstruct high-resolution (HR) images from their low-resolution (LR) counterparts, often relying on effective downsampling to generate diverse and realistic training pairs. In this work, we propose a co-learning framework that jointly optimizes a single-step diffusion-based upsampler and a learnable downsampler, enhanced by two discriminators and a cyclic distillation strategy. Our learnable downsampler is designed to better capture realistic degradation patterns while preserving structural details in the LR domain, which is crucial for enhancing SR performance. By leveraging a diffusion-based approach, our model generates diverse LR-HR pairs during training, enabling robust learning across varying degradations. We demonstrate the effectiveness of our method on both general real-world and domain-specific face SR tasks, achieving state-of-the-art performance in both fidelity and perceptual quality. Our approach not only improves efficiency with a single inference step but also ensures high-quality image reconstruction, bridging the gap between synthetic and real-world SR scenarios.
title Co-learning Single-Step Diffusion Upsampler and Downsampler with Two Discriminators and Distillation
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.07663