Recent advancements in text-to-image diffusion models have shown remarkable creative capabilities with textual prompts, but generating personalized instances based on specific subjects, known as subject-driven generation, remains challenging. To tackle this issue, we present a new hybrid framework called \(\texttt{HybridBooth}\) , which merges the benefits of optimization-based and direct-regression methods. \(\texttt{HybridBooth}\)  operates in two stages: the Word Embedding Probe, which generates a robust initial word embedding using a fine-tuned encoder, and the Word Embedding Refinement, which further adapts the encoder to specific subject images by optimizing key parameters. This approach allows for effective and fast inversion of visual concepts into textual embedding, even from a single image, while maintaining the model’s generalization capabilities.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

HybridBooth: Hybrid Prompt Inversion for Efficient Subject-Driven Generation

  • Shanyan Guan,
  • Yanhao Ge,
  • Ying Tai,
  • Jian Yang,
  • Wei Li,
  • Mingyu You

摘要

Recent advancements in text-to-image diffusion models have shown remarkable creative capabilities with textual prompts, but generating personalized instances based on specific subjects, known as subject-driven generation, remains challenging. To tackle this issue, we present a new hybrid framework called \(\texttt{HybridBooth}\) , which merges the benefits of optimization-based and direct-regression methods. \(\texttt{HybridBooth}\)  operates in two stages: the Word Embedding Probe, which generates a robust initial word embedding using a fine-tuned encoder, and the Word Embedding Refinement, which further adapts the encoder to specific subject images by optimizing key parameters. This approach allows for effective and fast inversion of visual concepts into textual embedding, even from a single image, while maintaining the model’s generalization capabilities.