Exploiting Diffusion Prior for Generalizable Dense Prediction

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Lee, Hsin-Ying, Tseng, Hung-Yu, Yang, Ming-Hsuan
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929301026242560
author Lee, Hsin-Ying
Tseng, Hung-Yu
Lee, Hsin-Ying
Yang, Ming-Hsuan
author_facet Lee, Hsin-Ying
Tseng, Hung-Yu
Lee, Hsin-Ying
Yang, Ming-Hsuan
contents Contents generated by recent advanced Text-to-Image (T2I) diffusion models are sometimes too imaginative for existing off-the-shelf dense predictors to estimate due to the immitigable domain gap. We introduce DMP, a pipeline utilizing pre-trained T2I models as a prior for dense prediction tasks. To address the misalignment between deterministic prediction tasks and stochastic T2I models, we reformulate the diffusion process through a sequence of interpolations, establishing a deterministic mapping between input RGB images and output prediction distributions. To preserve generalizability, we use low-rank adaptation to fine-tune pre-trained models. Extensive experiments across five tasks, including 3D property estimation, semantic segmentation, and intrinsic image decomposition, showcase the efficacy of the proposed method. Despite limited-domain training data, the approach yields faithful estimations for arbitrary images, surpassing existing state-of-the-art algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2311_18832
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Exploiting Diffusion Prior for Generalizable Dense Prediction
Lee, Hsin-Ying
Tseng, Hung-Yu
Lee, Hsin-Ying
Yang, Ming-Hsuan
Computer Vision and Pattern Recognition
Contents generated by recent advanced Text-to-Image (T2I) diffusion models are sometimes too imaginative for existing off-the-shelf dense predictors to estimate due to the immitigable domain gap. We introduce DMP, a pipeline utilizing pre-trained T2I models as a prior for dense prediction tasks. To address the misalignment between deterministic prediction tasks and stochastic T2I models, we reformulate the diffusion process through a sequence of interpolations, establishing a deterministic mapping between input RGB images and output prediction distributions. To preserve generalizability, we use low-rank adaptation to fine-tune pre-trained models. Extensive experiments across five tasks, including 3D property estimation, semantic segmentation, and intrinsic image decomposition, showcase the efficacy of the proposed method. Despite limited-domain training data, the approach yields faithful estimations for arbitrary images, surpassing existing state-of-the-art algorithms.
title Exploiting Diffusion Prior for Generalizable Dense Prediction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2311.18832