Learning from a Generative Oracle: Domain Adaptation for Restoration

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Hu, Yuyang, Sahraee-Ardakan, Mojtaba, Bansal, Arpit, Mei, Kangfu, Qi, Christian, Milanfar, Peyman, Delbracio, Mauricio
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909957556797440
author Hu, Yuyang
Sahraee-Ardakan, Mojtaba
Bansal, Arpit
Mei, Kangfu
Qi, Christian
Milanfar, Peyman
Delbracio, Mauricio
author_facet Hu, Yuyang
Sahraee-Ardakan, Mojtaba
Bansal, Arpit
Mei, Kangfu
Qi, Christian
Milanfar, Peyman
Delbracio, Mauricio
contents Pre-trained image restoration models often fail on real-world, out-of-distribution degradations due to significant domain gaps. Adapting to these unseen domains is challenging, as out-of-distribution data lacks ground truth, and traditional adaptation methods often require complex architectural changes. We propose LEGO (Learning from a Generative Oracle), a practical three-stage framework for post-training domain adaptation without paired data. LEGO converts this unsupervised challenge into a tractable pseudo-supervised one. First, we obtain initial restorations from the pre-trained model. Second, we leverage a frozen, large-scale generative oracle to refine these estimates into high-quality pseudo-ground-truths. Third, we fine-tune the original model using a mixed-supervision strategy combining in-distribution data with these new pseudo-pairs. This approach adapts the model to the new distribution without sacrificing its original robustness or requiring architectural modifications. Experiments demonstrate that LEGO effectively bridges the domain gap, significantly improving performance on diverse real-world benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2512_11121
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Learning from a Generative Oracle: Domain Adaptation for Restoration
Hu, Yuyang
Sahraee-Ardakan, Mojtaba
Bansal, Arpit
Mei, Kangfu
Qi, Christian
Milanfar, Peyman
Delbracio, Mauricio
Computer Vision and Pattern Recognition
Image and Video Processing
Pre-trained image restoration models often fail on real-world, out-of-distribution degradations due to significant domain gaps. Adapting to these unseen domains is challenging, as out-of-distribution data lacks ground truth, and traditional adaptation methods often require complex architectural changes. We propose LEGO (Learning from a Generative Oracle), a practical three-stage framework for post-training domain adaptation without paired data. LEGO converts this unsupervised challenge into a tractable pseudo-supervised one. First, we obtain initial restorations from the pre-trained model. Second, we leverage a frozen, large-scale generative oracle to refine these estimates into high-quality pseudo-ground-truths. Third, we fine-tune the original model using a mixed-supervision strategy combining in-distribution data with these new pseudo-pairs. This approach adapts the model to the new distribution without sacrificing its original robustness or requiring architectural modifications. Experiments demonstrate that LEGO effectively bridges the domain gap, significantly improving performance on diverse real-world benchmarks.
title Learning from a Generative Oracle: Domain Adaptation for Restoration
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2512.11121