提出新损失函数,让模型能统一检索图文混合内容。
Generalized Contrastive Learning for Universal Multimodal Retrieval
- 用批次内多模态对比学习,无需新建数据集
- 在多个基准上提升现有模型检索性能
- 适合需要跨模态通用检索的研究者
尽管交叉模态检索模型(如CLIP)性能持续提升,但在检索包含融合图文的键(如含图片和文字的维基页面)时表现下降。为解决此问题,近年研究尝试构建统一的单模型以支持多种模态组合的检索。常见方法是构建新的图像-文本三元组数据集(如给定查询图像,检索对应图文对),但需精细标注且难以泛化到未见模态组合。为此,本文提出广义对比学习(GCL),一种新损失形式,可在不依赖新数据集的情况下提升多模态检索性能。GCL通过在每个小批次内对所有模态进行对比学习,利用现有的图像-标题配对数据集,学习统一表示空间。我们在M-BEIR、MMEB和CoVR基准上验证了GCL的有效性,显示其对VISTA、CLIP和TinyCLIP等现成模型均带来一致性能提升。
原文摘要 · Abstract (English)
Despite their consistent performance improvements, cross-modal retrieval models (e.g., CLIP) show degraded performances with retrieving keys composed of fused image-text modality (e.g., Wikipedia pages with both images and text). To address this critical challenge, multimodal retrieval has been recently explored to develop a unified single retrieval model capable of retrieving keys across diverse modality combinations. A common approach involves constructing new composed sets of image-text triplets (e.g., retrieving a pair of image and text given a query image). However, such an approach requires careful curation to ensure the dataset quality and fails to generalize to unseen modality combinations. To overcome these limitations, this paper proposes Generalized Contrastive Learning (GCL), a novel loss formulation that improves multimodal retrieval performance without the burdensome need for new dataset curation. Specifically, GCL operates by enforcing contrastive learning across all modalities within a mini-batch, utilizing existing image-caption paired datasets to learn a unified representation space. We demonstrate the effectiveness of GCL by showing consistent performance improvements on off-the-shelf multimodal retrieval models (e.g., VISTA, CLIP, and TinyCLIP) using the M-BEIR, MMEB, and CoVR benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。