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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yu, Weichen, Yang, Ziyan, Lin, Shanchuan, Zhao, Qi, Wang, Jianyi, Gui, Liangke, Fredrikson, Matt, Jiang, Lu
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917879942742016
author Yu, Weichen
Yang, Ziyan
Lin, Shanchuan
Zhao, Qi
Wang, Jianyi
Gui, Liangke
Fredrikson, Matt
Jiang, Lu
author_facet Yu, Weichen
Yang, Ziyan
Lin, Shanchuan
Zhao, Qi
Wang, Jianyi
Gui, Liangke
Fredrikson, Matt
Jiang, Lu
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2412_19531
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Is Your Text-to-Image Model Robust to Caption Noise?
Yu, Weichen
Yang, Ziyan
Lin, Shanchuan
Zhao, Qi
Wang, Jianyi
Gui, Liangke
Fredrikson, Matt
Jiang, Lu
Computer Vision and Pattern Recognition
Artificial Intelligence
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.
title Is Your Text-to-Image Model Robust to Caption Noise?
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2412.19531