arXiv:2409.17531cs.CVcs.AI2024-09NeurIPS被引 64

SimVG通过解耦融合与任务,提升复杂句子的视觉定位效果。

SimVG: A Simple Framework for Visual Grounding with Decoupled Multi-modal Fusion

  • 分离多模态融合与下游任务,利用预训练模型增强表达
  • 轻量MLP分支结合动态权重蒸馏,加速推理且保持精度
  • 在6个数据集上达新SOTA,适合追求高效精准的视觉定位场景

视觉定位是将描述性语句定位到图像对应区域的常见视觉任务。现有方法多采用独立的图文编码,并使用复杂的手工模块或编码器-解码器架构进行模态交互与查询推理,但在处理复杂文本表达时性能显著下降。这是因为其仅依赖有限的下游数据来拟合多模态特征融合,仅在文本表达较简单时有效。鉴于文本表达的多样性及下游数据的独特性,现有融合模块对视觉语言上下文内容的挖掘尚未充分展开。本文提出一种简单而稳健的基于Transformer的框架SimVG。具体而言,通过利用现有的多模态预训练模型,并引入额外的对象标记,将视觉-语言特征融合与下游任务解耦,促进下游与预训练任务的深度整合。此外,在多分支同步学习过程中设计了动态权重平衡蒸馏方法,以增强仅包含轻量级MLP的分支的表征能力。该分支结构简化,提升了推理速度。在六个常用数据集(RefCOCO/+/g、ReferIt、Flickr30K、GRefCOCO)上的实验表明,SimVG具有优越性能。最终,该方法不仅在效率和收敛速度上取得提升,还在这些基准上达到新状态最优水平。

原文摘要 · Abstract (English)

Visual grounding is a common vision task that involves grounding descriptive sentences to the corresponding regions of an image. Most existing methods use independent image-text encoding and apply complex hand-crafted modules or encoder-decoder architectures for modal interaction and query reasoning. However, their performance significantly drops when dealing with complex textual expressions. This is because the former paradigm only utilizes limited downstream data to fit the multi-modal feature fusion. Therefore, it is only effective when the textual expressions are relatively simple. In contrast, given the wide diversity of textual expressions and the uniqueness of downstream training data, the existing fusion module, which extracts multimodal content from a visual-linguistic context, has not been fully investigated. In this paper, we present a simple yet robust transformer-based framework, SimVG, for visual grounding. Specifically, we decouple visual-linguistic feature fusion from downstream tasks by leveraging existing multimodal pre-trained models and incorporating additional object tokens to facilitate deep integration of downstream and pre-training tasks. Furthermore, we design a dynamic weight-balance distillation method in the multi-branch synchronous learning process to enhance the representation capability of the simpler branch. This branch only consists of a lightweight MLP, which simplifies the structure and improves reasoning speed. Experiments on six widely used VG datasets, i.e., RefCOCO/+/g, ReferIt, Flickr30K, and GRefCOCO, demonstrate the superiority of SimVG. Finally, the proposed method not only achieves improvements in efficiency and convergence speed but also attains new state-of-the-art performance on these benchmarks. Codes and models will be available at \url{https://github.com/Dmmm1997/SimVG}.

视觉定位多模态融合轻量模型Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。