Training-Free Large Model Priors for Multiple-in-One Image Restoration

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: He, Xuanhua, Li, Lang, Wang, Yingying, Zheng, Hui, Cao, Ke, Yan, Keyu, Li, Rui, Xie, Chengjun, Zhang, Jie, Zhou, Man
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911960594907136
author He, Xuanhua
Li, Lang
Wang, Yingying
Zheng, Hui
Cao, Ke
Yan, Keyu
Li, Rui
Xie, Chengjun
Zhang, Jie
Zhou, Man
author_facet He, Xuanhua
Li, Lang
Wang, Yingying
Zheng, Hui
Cao, Ke
Yan, Keyu
Li, Rui
Xie, Chengjun
Zhang, Jie
Zhou, Man
contents Image restoration aims to reconstruct the latent clear images from their degraded versions. Despite the notable achievement, existing methods predominantly focus on handling specific degradation types and thus require specialized models, impeding real-world applications in dynamic degradation scenarios. To address this issue, we propose Large Model Driven Image Restoration framework (LMDIR), a novel multiple-in-one image restoration paradigm that leverages the generic priors from large multi-modal language models (MMLMs) and the pretrained diffusion models. In detail, LMDIR integrates three key prior knowledges: 1) global degradation knowledge from MMLMs, 2) scene-aware contextual descriptions generated by MMLMs, and 3) fine-grained high-quality reference images synthesized by diffusion models guided by MMLM descriptions. Standing on above priors, our architecture comprises a query-based prompt encoder, degradation-aware transformer block injecting global degradation knowledge, content-aware transformer block incorporating scene description, and reference-based transformer block incorporating fine-grained image priors. This design facilitates single-stage training paradigm to address various degradations while supporting both automatic and user-guided restoration. Extensive experiments demonstrate that our designed method outperforms state-of-the-art competitors on multiple evaluation benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2407_13181
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Training-Free Large Model Priors for Multiple-in-One Image Restoration
He, Xuanhua
Li, Lang
Wang, Yingying
Zheng, Hui
Cao, Ke
Yan, Keyu
Li, Rui
Xie, Chengjun
Zhang, Jie
Zhou, Man
Computer Vision and Pattern Recognition
Image restoration aims to reconstruct the latent clear images from their degraded versions. Despite the notable achievement, existing methods predominantly focus on handling specific degradation types and thus require specialized models, impeding real-world applications in dynamic degradation scenarios. To address this issue, we propose Large Model Driven Image Restoration framework (LMDIR), a novel multiple-in-one image restoration paradigm that leverages the generic priors from large multi-modal language models (MMLMs) and the pretrained diffusion models. In detail, LMDIR integrates three key prior knowledges: 1) global degradation knowledge from MMLMs, 2) scene-aware contextual descriptions generated by MMLMs, and 3) fine-grained high-quality reference images synthesized by diffusion models guided by MMLM descriptions. Standing on above priors, our architecture comprises a query-based prompt encoder, degradation-aware transformer block injecting global degradation knowledge, content-aware transformer block incorporating scene description, and reference-based transformer block incorporating fine-grained image priors. This design facilitates single-stage training paradigm to address various degradations while supporting both automatic and user-guided restoration. Extensive experiments demonstrate that our designed method outperforms state-of-the-art competitors on multiple evaluation benchmarks.
title Training-Free Large Model Priors for Multiple-in-One Image Restoration
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.13181