CVApr 5, 2023

Taming Encoder for Zero Fine-tuning Image Customization with Text-to-Image Diffusion Models

arXiv:2304.02642v1132 citationsh-index: 52
Originality Incremental advance
AI Analysis

This addresses the need for efficient image customization in text-to-image diffusion models, offering a zero fine-tuning approach that is incremental over previous optimization-heavy methods.

The paper tackles the problem of generating images of user-specified customized objects without requiring lengthy per-object optimization, achieving compelling output quality, appearance diversity, and object fidelity with only a single feed-forward pass.

This paper proposes a method for generating images of customized objects specified by users. The method is based on a general framework that bypasses the lengthy optimization required by previous approaches, which often employ a per-object optimization paradigm. Our framework adopts an encoder to capture high-level identifiable semantics of objects, producing an object-specific embedding with only a single feed-forward pass. The acquired object embedding is then passed to a text-to-image synthesis model for subsequent generation. To effectively blend a object-aware embedding space into a well developed text-to-image model under the same generation context, we investigate different network designs and training strategies, and propose a simple yet effective regularized joint training scheme with an object identity preservation loss. Additionally, we propose a caption generation scheme that become a critical piece in fostering object specific embedding faithfully reflected into the generation process, while keeping control and editing abilities. Once trained, the network is able to produce diverse content and styles, conditioned on both texts and objects. We demonstrate through experiments that our proposed method is able to synthesize images with compelling output quality, appearance diversity, and object fidelity, without the need of test-time optimization. Systematic studies are also conducted to analyze our models, providing insights for future work.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes