中文

Infusion:防止定制文本到图像扩散中的过拟合

计算机视觉与模式识别 2024-04-23 v1 人工智能

摘要

文本到图像(T2I)定制旨在创建体现文本描述中特定视觉概念的图像。然而,现有工作仍面临一个主要挑战:概念过拟合。为了解决这一挑战,我们首先分析过拟合,将其分为概念无关过拟合(破坏非定制概念知识)和概念特定过拟合(局限于在有限模态上定制,即背景、布局、风格)。为了评估过拟合程度,我们进一步引入两个指标:潜在 Fisher 散度和 Wasserstein 度量,分别测量非定制和定制概念分布的变化。基于分析,我们提出 Infusion,一种 T2I 定制方法,使得学习目标概念能够避免受限于有限训练模态,同时保留非定制知识。值得注意的是,Infusion 以极高的效率实现这一目标,仅需 11KB 的训练参数。大量实验也表明,我们的方法在单概念和多概念定制生成中均优于最先进的方法。

关键词

引用

@article{arxiv.2404.14007,
  title  = {Infusion: Preventing Customized Text-to-Image Diffusion from Overfitting},
  author = {Weili Zeng and Yichao Yan and Qi Zhu and Zhuo Chen and Pengzhi Chu and Weiming Zhao and Xiaokang Yang},
  journal= {arXiv preprint arXiv:2404.14007},
  year   = {2024}
}

备注

10 pages