Referee Can Play: An Alternative Approach to Conditional Generation via Model Inversion

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liu, Xuantong, Hu, Tianyang, Wang, Wenjia, Kawaguchi, Kenji, Yao, Yuan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910343689666560
author Liu, Xuantong
Hu, Tianyang
Wang, Wenjia
Kawaguchi, Kenji
Yao, Yuan
author_facet Liu, Xuantong
Hu, Tianyang
Wang, Wenjia
Kawaguchi, Kenji
Yao, Yuan
contents As a dominant force in text-to-image generation tasks, Diffusion Probabilistic Models (DPMs) face a critical challenge in controllability, struggling to adhere strictly to complex, multi-faceted instructions. In this work, we aim to address this alignment challenge for conditional generation tasks. First, we provide an alternative view of state-of-the-art DPMs as a way of inverting advanced Vision-Language Models (VLMs). With this formulation, we naturally propose a training-free approach that bypasses the conventional sampling process associated with DPMs. By directly optimizing images with the supervision of discriminative VLMs, the proposed method can potentially achieve a better text-image alignment. As proof of concept, we demonstrate the pipeline with the pre-trained BLIP-2 model and identify several key designs for improved image generation. To further enhance the image fidelity, a Score Distillation Sampling module of Stable Diffusion is incorporated. By carefully balancing the two components during optimization, our method can produce high-quality images with near state-of-the-art performance on T2I-Compbench.
format Preprint
id arxiv_https___arxiv_org_abs_2402_16305
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Referee Can Play: An Alternative Approach to Conditional Generation via Model Inversion
Liu, Xuantong
Hu, Tianyang
Wang, Wenjia
Kawaguchi, Kenji
Yao, Yuan
Machine Learning
Artificial Intelligence
As a dominant force in text-to-image generation tasks, Diffusion Probabilistic Models (DPMs) face a critical challenge in controllability, struggling to adhere strictly to complex, multi-faceted instructions. In this work, we aim to address this alignment challenge for conditional generation tasks. First, we provide an alternative view of state-of-the-art DPMs as a way of inverting advanced Vision-Language Models (VLMs). With this formulation, we naturally propose a training-free approach that bypasses the conventional sampling process associated with DPMs. By directly optimizing images with the supervision of discriminative VLMs, the proposed method can potentially achieve a better text-image alignment. As proof of concept, we demonstrate the pipeline with the pre-trained BLIP-2 model and identify several key designs for improved image generation. To further enhance the image fidelity, a Score Distillation Sampling module of Stable Diffusion is incorporated. By carefully balancing the two components during optimization, our method can produce high-quality images with near state-of-the-art performance on T2I-Compbench.
title Referee Can Play: An Alternative Approach to Conditional Generation via Model Inversion
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2402.16305