arXiv:2412.19531cs.CVcs.AI2024-12被引 3

研究发现图像生成模型对标题噪声敏感,提出用置信度降噪提升鲁棒性。

Is Your Text-to-Image Model Robust to Caption Noise?

  • 用VLM置信度识别并过滤标题中的幻觉噪声。
  • 微调时标题质量差异直接影响生成效果,细微偏差即显著影响表征学习。
  • 适合关注生成模型鲁棒性与数据质量的研究者。

在文本到图像(T2I)生成中,常使用视觉语言模型(VLM)进行图像重标注。尽管VLM存在幻觉问题——生成与视觉现实不符的描述内容,但此类标题幻觉对T2I生成性能的影响尚未深入探讨。我们通过实证研究构建了一个包含VLM生成标题的全面数据集,并系统分析了标题幻觉对生成结果的影响。研究发现:(1)标题质量差异在微调过程中持续影响模型输出;(2)VLM置信度得分可有效指示数据分布中的噪声模式;(3)标题保真度的细微变化会显著影响学习表征的质量。这些发现凸显了标题质量对模型性能的深远影响,强调需发展更先进的鲁棒训练算法。为此,我们提出一种基于VLM置信度的降噪方法,以增强T2I模型对标题幻觉的鲁棒性。

原文摘要 · Abstract (English)

In text-to-image (T2I) generation, a prevalent training technique involves utilizing Vision Language Models (VLMs) for image re-captioning. Even though VLMs are known to exhibit hallucination, generating descriptive content that deviates from the visual reality, the ramifications of such caption hallucinations on T2I generation performance remain under-explored. Through our empirical investigation, we first establish a comprehensive dataset comprising VLM-generated captions, and then systematically analyze how caption hallucination influences generation outcomes. Our findings reveal that (1) the disparities in caption quality persistently impact model outputs during fine-tuning. (2) VLMs confidence scores serve as reliable indicators for detecting and characterizing noise-related patterns in the data distribution. (3) even subtle variations in caption fidelity have significant effects on the quality of learned representations. These findings collectively emphasize the profound impact of caption quality on model performance and highlight the need for more sophisticated robust training algorithm in T2I. In response to these observations, we propose a approach leveraging VLM confidence score to mitigate caption noise, thereby enhancing the robustness of T2I models against hallucination in caption.

文本生成图像生成鲁棒性数据质量

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