arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2211.00575cs.CVcs.AIcs.LG
收藏

使用噪声注入CLIP的纯文本训练图像描述生成

Text-Only Training for Image Captioning using Noise-Injected CLIP

Tel Aviv University(特拉维夫大学)

David Nukrai, Ron Mokady, Amir Globerson

AI总结:

提出仅用文本数据训练图像描述生成模型的方法,利用CLIP嵌入空间并通过噪声注入弥合差距,实现四个基准上的最先进零样本性能。

摘要:

我们考虑仅使用CLIP模型和额外的文本数据进行训练,而不使用额外带标注的图像,来完成图像描述生成任务。我们的方法基于这样一个事实:CLIP被训练为使视觉和文本嵌入相似。因此,我们只需学习如何将CLIP的文本嵌入转换回文本,而这可以通过仅使用文本,为冻结的CLIP文本编码器学习一个解码器来实现。我们认为这一直觉“几乎正确”,因为嵌入空间之间存在差距,并提议在训练期间通过噪声注入来修正这一问题。我们通过在四个基准上展示最先进的零样本图像描述生成结果(包括风格迁移)来证明我们方法的有效性。代码、数据和模型已在GitHub上提供。

英文摘要:

We consider the task of image-captioning using only the CLIP model and additional text data at training time, and no additional captioned images. Our approach relies on the fact that CLIP is trained to make visual and textual embeddings similar. Therefore, we only need to learn how to translate CLIP textual embeddings back into text, and we can learn how to do this by learning a decoder for the frozen CLIP text encoder using only text. We argue that this intuition is "almost correct" because of a gap between the embedding spaces, and propose to rectify this via noise injection during training. We demonstrate the effectiveness of our approach by showing SOTA zero-shot image captioning across four benchmarks, including style transfer. Code, data, and models are available on GitHub.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑