arXiv:2412.05756cs.CV2024-12被引 9

用指令调优的多模态模型提升图像修改查询的精准匹配能力。

Compositional Image Retrieval via Instruction-Aware Contrastive Learning

  • 基于指令微调的多模态大模型生成联合嵌入表示
  • 在四个数据集上超越现有最佳模型,显著提升零样本检索效果
  • 适合需要精准理解图文指令的图像检索场景

组合图像检索(CIR)是根据带有文本指令的图像对,检索出经过指定修改的目标图像。该任务本质是遵循指令的任务,要求模型理解并应用视觉修改。由于下游任务标注数据稀缺,零样本CIR(ZS-CIR)具有重要意义。尽管基于CLIP的现有模型已取得一定进展,但其对修改指令的理解和执行能力仍有限。部分研究尝试引入大语言模型(LLM),但难以有效融合多模态信息与指令理解。为此,我们提出一种新嵌入方法,利用指令调优的多模态大模型(MLLM)生成组合表示,显著增强指令跟随能力,实现图像与指令的深度融合。然而,直接使用MLLM面临挑战:其主要面向文本生成,而非嵌入提取。为此,我们设计两阶段训练策略,在类似CIR格式的三元组数据集上学习联合多模态嵌入空间,并进一步优化指令遵循能力。在FashionIQ、CIRR、GeneCIS和CIRCO四个公开数据集上的大量实验表明,我们的模型性能显著优于现有最先进方法。代码已开源。

原文摘要 · Abstract (English)

Composed Image Retrieval (CIR) involves retrieving a target image based on a composed query of an image paired with text that specifies modifications or changes to the visual reference. CIR is inherently an instruction-following task, as the model needs to interpret and apply modifications to the image. In practice, due to the scarcity of annotated data in downstream tasks, Zero-Shot CIR (ZS-CIR) is desirable. While existing ZS-CIR models based on CLIP have shown promising results, their capability in interpreting and following modification instructions remains limited. Some research attempts to address this by incorporating Large Language Models (LLMs). However, these approaches still face challenges in effectively integrating multimodal information and instruction understanding. To tackle above challenges, we propose a novel embedding method utilizing an instruction-tuned Multimodal LLM (MLLM) to generate composed representation, which significantly enhance the instruction following capability for a comprehensive integration between images and instructions. Nevertheless, directly applying MLLMs introduces a new challenge since MLLMs are primarily designed for text generation rather than embedding extraction as required in CIR. To address this, we introduce a two-stage training strategy to efficiently learn a joint multimodal embedding space and further refining the ability to follow modification instructions by tuning the model in a triplet dataset similar to the CIR format. Extensive experiments on four public datasets: FashionIQ, CIRR, GeneCIS, and CIRCO demonstrates the superior performance of our model, outperforming state-of-the-art baselines by a significant margin. Codes are available at the GitHub repository.

图像检索指令理解多模态零样本

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