arXiv:2412.09169cs.CV2024-12被引 2

通过投影文本嵌入消除无关语义,提升图像生成与提示的契合度。

DECOR:Decomposition and Projection of Text Embeddings for Text-to-Image Customization

  • 分解并分析文本嵌入空间,定位过拟合根源
  • 将嵌入投影至无关词向量的正交空间,抑制噪声语义
  • 显著提升提示对齐度,适合需要精准控制生成的场景

文本到图像(T2I)模型能有效捕捉参考图像的内容或风格以实现高质量定制。代表性方法是使用低秩适配(LoRA)进行微调,可高效利用参考图像完成模型定制。然而,当参考图像数量有限时,微调易导致过拟合,引发提示错位或内容泄露等问题,使模型无法准确遵循输入提示或生成非期望对象。为此,本文研究了推理过程中引导扩散模型的文本嵌入。通过分解文本嵌入矩阵并进行成分分析,揭示嵌入空间的几何结构,识别过拟合成因。基于此,提出DECOR方法,将文本嵌入投影至与非期望词向量正交的向量空间,从而降低嵌入中不需要语义的影响。实验表明,DECOR在文本与视觉对齐评估指标上优于现有顶尖定制模型,达到帕累托前沿性能。同时生成图像更忠实于输入提示,有效缓解过拟合并提升文本到图像定制效果。

原文摘要 · Abstract (English)

Text-to-image (T2I) models can effectively capture the content or style of reference images to perform high-quality customization. A representative technique for this is fine-tuning using low-rank adaptations (LoRA), which enables efficient model customization with reference images. However, fine-tuning with a limited number of reference images often leads to overfitting, resulting in issues such as prompt misalignment or content leakage. These issues prevent the model from accurately following the input prompt or generating undesired objects during inference. To address this problem, we examine the text embeddings that guide the diffusion model during inference. This study decomposes the text embedding matrix and conducts a component analysis to understand the embedding space geometry and identify the cause of overfitting. Based on this, we propose DECOR, which projects text embeddings onto a vector space orthogonal to undesired token vectors, thereby reducing the influence of unwanted semantics in the text embeddings. Experimental results demonstrate that DECOR outperforms state-of-the-art customization models and achieves Pareto frontier performance across text and visual alignment evaluation metrics. Furthermore, it generates images more faithful to the input prompts, showcasing its effectiveness in addressing overfitting and enhancing text-to-image customization.

文本生成图像定制嵌入投影过拟合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。