HyperAIHyperAI

Command Palette

Search for a command to run...

ViLT:无需卷积或区域监督的视觉-语言 Transformer 模型

Wonjae Kim; Bokyung Son; Ildoo Kim

摘要

视觉-语言预训练(VLP)在各种联合视觉-语言下游任务中提高了性能。目前的VLP方法严重依赖于图像特征提取过程,其中大多数涉及区域监督(例如,目标检测)和卷积架构(例如,ResNet)。尽管文献中对此较少关注,但我们发现这种方法在效率/速度和表达能力方面存在问题:(1) 效率/速度方面,仅提取输入特征所需的计算量就远超多模态交互步骤;(2) 表达能力方面,其上限受制于视觉嵌入器及其预定义的视觉词汇表。在本文中,我们提出了一种最小化的VLP模型——视觉-语言 Transformer (ViLT),该模型在处理视觉输入时进行了大幅简化,采用了与处理文本输入相同的无卷积方式。实验结果表明,ViLT比之前的VLP模型快数十倍,同时在下游任务性能上具有竞争力或更优。我们的代码和预训练权重可在https://github.com/dandelin/vilt 获取。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供