arXiv:2412.07619cs.CL2024-12被引 9

让大模型更懂如何选示例,提升视觉语言任务表现

DRUM: Learning Demonstration Retriever for Large MUlti-modal Models

  • 用图文嵌入拼接增强示例检索能力
  • 根据大模型反馈重排序示例,提升相关性
  • 迭代挖掘优质示例,训练更精准的检索器

大型视觉语言模型(LVLM)在上下文学习(ICL)中依赖示例,但现有方法如固定示例或直接通过视觉-语言嵌入模型选取,无法保证示例适配模型需求。为此,本文提出DRUM框架,通过微调视觉-语言嵌入模型以更好满足LVLM需求。首先,将图像与文本嵌入拼接以提升检索性能;其次,利用LVLM反馈对检索到的示例进行重排序,并使用列表级排序损失训练嵌入模型;第三,设计迭代示例挖掘策略优化嵌入模型训练。在3类视觉-语言任务、7个基准数据集上的实验证明,该框架能有效提升LVLM的上下文学习性能。

原文摘要 · Abstract (English)

Recently, large language models (LLMs) have demonstrated impressive capabilities in dealing with new tasks with the help of in-context learning (ICL). In the study of Large Vision-Language Models (LVLMs), when implementing ICL, researchers usually adopts the naive strategies like fixed demonstrations across different samples, or selecting demonstrations directly via a visual-language embedding model. These methods does not guarantee the configured demonstrations fit the need of the LVLMs. To address this issue, we now propose a novel framework, \underline{d}emonstration \underline{r}etriever for large m\underline{u}lti-modal \underline{m}odel (DRUM), which fine-tunes the visual-language embedding model to better meet the LVLM's needs. First, we discuss the retrieval strategies for a visual-language task, assuming an embedding model is given. And we propose to concate the image and text embeddings to enhance the retrieval performance. Second, we propose to re-rank the demonstrations retrieved by the embedding model via the LVLM's feedbacks, and calculate a list-wise ranking loss for training the embedding model. Third, we propose an iterative demonstration mining strategy to improve the training of the embedding model. Through extensive experiments on 3 types of visual-language tasks, 7 benchmark datasets, our DRUM framework is proven to be effective in boosting the LVLM's in-context learning performance via retrieving more proper demonstrations.

视觉语言上下文学习示例检索多模态

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