重磅干货,第一时间送达 摘要 多模态推理系统依靠预先训练的目标检测器从图像中提取感兴趣的区域。然而,这个关键模块通常用作一个黑盒,它独立于下游任务进行训练,并且使用固定的对象和属性词汇表。这使得这类系统很难捕捉以自由形式文本表达的视觉概念的长尾。在本文中,作者提出了MDETR,一种端到端调制检测器,用于检测基于原始文本查询(如标题或问题)的图像中的对象。在模型的早期阶段,作者使用一个基于变压器的架构,通过融合两种模式来共同推理文本和图像。作者在1.3M文本-图像对上预先训练该网络,从已有的多模态数据集中挖掘,文本中的短语和图像中的对象之间有明确的对齐。然后,作者对几个下游任务进行微调,如短语基础、引用表达理解和分割,在流行的基准上获得最先进的结果。作者还研究了作者的模型作为一个对象检测器在给定的标签集上的效用,当微调在几个镜头设置。作者展示了作者的训练前方法提供了一种处理对象类别长尾的方法,这些类别只有很少的标记实例。作者的方法可以很容易地扩展到视觉问答,在GQA和cleverr上获得具有竞争力的性能。 代码链接:https://github.com/ashkamath/mdetr 作者的贡献如下:
框架结构 MDETR使用卷积主干提取视觉特征,使用RoBERTa等语言模型提取文本特征。这两种模式的特征被投射到一个共享的嵌入空间,连接和馈送到一个变压器编码器-解码器,它预测对象的边界框和它们在文本中的基础。 在MDETR预处理训练过程中,训练模型检测问题中提到的所有对象。为了将其扩展为问答,作者除了提供对象查询外,还提供特定于QA的查询作为转换器解码器的输入。作者用专门的头来处理不同类型的问题。 实验结果 作者对注释的组合产生了如下示例:手腕上戴着手表、穿着灰色衬衫的人。另一个穿蓝色毛衣的人。第三个人穿着灰色外套,围着围巾。作者为这个标题显示了模型的预测。它能够关注图像中所有的物体,然后根据文本对它们进行消歧。该模型被训练为预测短语的根作为正标记跨度,正如作者在图中可以看到的,正确地指的是三个不同的人。 结论 作者提出了MDETR,一种完全可微调制检测器。作者建立了它在多种数据集上的多模态理解任务上的强大性能,并展示了它在其他下游应用如少镜头检测和视觉问答中的潜力。作者希望这项工作为开发完全集成的多模态架构开辟新的机会,而不依赖黑匣子目标探测器。 论文链接:https:///pdf/2104.12763.pdf 每日坚持论文分享不易,如果喜欢我们的内容,希望可以推荐或者转发给周围的同学。 |
|