In plain words: Web image captions are often messy, so an open LLaMA-3 vision model rewrote the text for 1.3 billion images with richer, accurate descriptions. Models trained on these captions matched images to text better and followed tricky drawing instructions more faithfully than the original captions.
Abstract
Web-crawled image-text pairs are inherently noisy. Prior studies demonstrate that semantically aligning and enriching textual descriptions of these pairs can significantly enhance model training across various vision-language tasks, particularly text-to-image generation. However, large-scale investigations in this area remain predominantly closed-source. Our paper aims to bridge this community effort, leveraging the powerful and \textit{open-sourced} LLaMA-3, a GPT-4 level LLM. Our recaptioning pipeline is simple: first, we fine-tune a LLaMA-3-8B powered LLaVA-1.5 and then employ it to recaption 1.3 billion images from the DataComp-1B dataset. Our empirical results confirm that this enhanced dataset, Recap-DataComp-1B, offers substantial benefits in training advanced vision-language models. For discriminative models like CLIP, we observe enhanced zero-shot performance in cross-modal retrieval tasks. For generative models like text-to-image Diffusion Transformers, the generated images exhibit a significant improvement in alignment with users' text instructions, especially in following complex queries. Our project page is https://www.haqtu.me/Recap-Datacomp-1B/
Xianhang Li, Haoqin Tu, Mude Hui, Zeyu Wang, Bingchen Zhao, Junfei Xiao, Sucheng Ren, Jieru Mei, Qing Liu, Huangjie Zheng, Yuyin Zhou, Cihang Xie
arXiv:2406.08478 · cs.CV, cs.CL · submitted Jun 12, 2024 · updated Jun 18, 2024
abstract · pdf · html · First five authors contributed equally
All the results that show improvement seem to be evaluations using LLMs, e.g. they are showing LLMs think the LLM-generated text is better, which is neither surprising nor expected to correlate with real downstream task performance - unless your final task is labelling for an LLM, e.g. retrieval I guess.