4 个月前

使用噪声注入的CLIP进行图像标题生成的纯文本训练

使用噪声注入的CLIP进行图像标题生成的纯文本训练

摘要

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

代码仓库

zelaki/wsac
pytorch
GitHub 中提及
uriberger/re_cap
pytorch
GitHub 中提及
davidhuji/capdec
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-captioning-on-coco-captionsCapDec
BLEU-4: 26.4
CIDER: 91.8
METEOR: 25.1
image-captioning-on-flickrstyle10kCapDec
BLEU-1 (Romantic): 29.4
image-captioning-on-mscoco-1CapDec
BLEU-4: 26.4
semi-supervised-learning-for-image-captioning-2CapDec
CIDEr: 39.1
semi-supervised-learning-for-image-captioning-3CapDec
CIDEr: 30.0

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用噪声注入的CLIP进行图像标题生成的纯文本训练 | 论文 | HyperAI超神经