arXiv:2409.08206cs.CVcs.MM2024-09被引 1

让视觉语言模型更懂图像与文字的组合关系,提升细粒度对齐能力。

ComAlign: Compositional Alignment in Vision-Language Models

  • 通过轻量网络在现有编码器上实现图文组件的细粒度对齐
  • 在多个基准上显著提升检索与组合理解任务性能
  • 仅需弱监督的图文对即可训练,适合集成到现有模型

视觉语言模型(如CLIP)在提取可迁移特征方面表现优异,但其训练通常基于图像与文本全局嵌入间的粗粒度对比损失,可能丢失模态的组合结构。现有研究发现,这些模型缺乏属性绑定和对象关系识别等组合理解能力。尽管部分方法尝试实现更细粒度对齐,但或未能提取合适粒度的语义成分,或未充分利用图文对中的对应关系(尤其在包含多个成分的样本中)。为此,我们提出细粒度组合对齐方法ComAlign,仅使用图像-文本对的弱监督信号,发现图文组件间更精确的对应关系。方法强调从文本中提取的组合结构(包括实体与关系)必须在图像中保留。通过在现有视觉与语言编码器之上训练一个轻量级网络,利用小规模数据集学习跨模态节点与边的对齐。实验结果表明,在多种VLM与数据集上,该方法在检索与组合基准任务中均有显著提升,验证了其有效性。

原文摘要 · Abstract (English)

Vision-language models (VLMs) like CLIP have showcased a remarkable ability to extract transferable features for downstream tasks. Nonetheless, the training process of these models is usually based on a coarse-grained contrastive loss between the global embedding of images and texts which may lose the compositional structure of these modalities. Many recent studies have shown VLMs lack compositional understandings like attribute binding and identifying object relationships. Although some recent methods have tried to achieve finer-level alignments, they either are not based on extracting meaningful components of proper granularity or don't properly utilize the modalities' correspondence (especially in image-text pairs with more ingredients). Addressing these limitations, we introduce Compositional Alignment (ComAlign), a fine-grained approach to discover more exact correspondence of text and image components using only the weak supervision in the form of image-text pairs. Our methodology emphasizes that the compositional structure (including entities and relations) extracted from the text modality must also be retained in the image modality. To enforce correspondence of fine-grained concepts in image and text modalities, we train a lightweight network lying on top of existing visual and language encoders using a small dataset. The network is trained to align nodes and edges of the structure across the modalities. Experimental results on various VLMs and datasets demonstrate significant improvements in retrieval and compositional benchmarks, affirming the effectiveness of our plugin model.

视觉语言模型组合对齐细粒度对齐弱监督

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