Recent advances in image captioning have focused on scaling the data and model size, substantially increasing the cost of pre-training and finetuning. As an alternative to large models, we present SmallCap, which generates a caption conditioned on an input image and related captions retrieved from a datastore. Our model is lightweight and fast to train, as the only learned parameters are in newly introduced cross-attention layers between a pre-trained CLIP encoder and GPT-2 decoder. SmallCap can transfer to new domains without additional finetuning and can exploit large-scale data in a training-free fashion since the contents of the datastore can be readily replaced. Our experiments show that SmallCap, trained only on COCO, has competitive performance on this benchmark, and also transfers to other domains without retraining, solely through retrieval from target-domain data. Further improvement is achieved through the training-free exploitation of diverse human-labeled and web data, which proves to be effective for a range of domains, including the nocaps benchmark, designed to test generalization to unseen visual concepts.


翻译:近期图像字幕生成的进展聚焦于数据和模型规模的扩展,使得预训练和微调的成本大幅增加。为解决这一问题,我们提出了 SmallCap,该模型会根据在数据存储库中检索的相关字幕生成一种有人类识别准确率的字幕。我们的模型轻巧,训练速度快,因为唯一的学习参数是在预训练的 CLIP 编码器和 GPT-2 解码器之间的新引入的跨注意力层中设定。SmallCap 可以在不进行额外微调的情况下转移到新的领域,并且可以利用训练无关的大规模数据,因为数据存储库的内容可以很容易地替换。我们的实验表明,SmallCap 只需在 COCO 数据上训练就能在该基准测试中获得具有竞争力的性能,并且仅通过从目标领域的数据中检索字幕就能转移到其他领域。通过训练无关的利用多样化的人类和网络数据,我们实现了进一步的改进,这些数据对于各种领域都非常有效,包括用于测试对未见过的视觉概念的泛化能力的 nocaps 基准测试。

0
下载
关闭预览

相关内容

图像字幕(Image Captioning),是指从图像生成文本描述的过程,主要根据图像中物体和物体的动作。
ATMOL:利用对比学习预训练模型预测分子性质
专知会员服务
12+阅读 · 2022年8月14日
基于预训练语言模型的文本生成
专知会员服务
29+阅读 · 2022年1月28日
专知会员服务
21+阅读 · 2021年8月17日
专知会员服务
90+阅读 · 2021年6月29日
DeepMind开源最牛无监督学习BigBiGAN预训练模型
新智元
10+阅读 · 2019年10月10日
Transferring Knowledge across Learning Processes
CreateAMind
29+阅读 · 2019年5月18日
基于PyTorch/TorchText的自然语言处理库
专知
28+阅读 · 2019年4月22日
Unsupervised Learning via Meta-Learning
CreateAMind
43+阅读 · 2019年1月3日
国家自然科学基金
1+阅读 · 2013年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2009年12月31日
Arxiv
11+阅读 · 2018年5月13日
VIP会员
相关VIP内容
ATMOL:利用对比学习预训练模型预测分子性质
专知会员服务
12+阅读 · 2022年8月14日
基于预训练语言模型的文本生成
专知会员服务
29+阅读 · 2022年1月28日
专知会员服务
21+阅读 · 2021年8月17日
专知会员服务
90+阅读 · 2021年6月29日
相关基金
国家自然科学基金
1+阅读 · 2013年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2009年12月31日
Top
微信扫码咨询专知VIP会员