arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.03411cs.CV

基于大型语言模型的交互式文本到图像检索:一种即插即用方法

Interactive Text-to-Image Retrieval with Large Language Models: A Plug-and-Play Approach

Seoul National University(首尔大学);Interdisciplinary Program in Artificial Intelligence, Seoul National University(首尔大学人工智能跨学科项目)

Saehyung Lee, Sangwon Yu, Junsung Park, Jihun Yi, Sungroh Yoon

AI总结:

本文提出PlugIR方法,利用LLM重写对话上下文并生成非冗余问题,实现无需微调的即插即用交互式文本到图像检索,并提出BRI评估指标,在多个基准上优于基线。

摘要:

在本文中,我们主要解决交互式文本到图像检索任务中的对话形式上下文查询问题。我们的方法PlugIR以两种方式积极利用LLM的通用指令遵循能力。首先,通过重写对话形式的上下文,我们消除了在现有视觉对话数据上微调检索模型的必要性,从而允许使用任意黑盒模型。其次,我们构建了LLM提问器,基于当前上下文中检索候选图像的信息,生成关于目标图像属性的非冗余问题。这种方法缓解了生成问题中的噪声和冗余问题。除方法外,我们提出了一种新的评估指标——最佳对数秩积分(BRI),用于全面评估交互式检索系统。PlugIR在多个基准上相比零样本和微调基线均表现出更优性能。此外,构成PlugIR的两种方法可以灵活地组合或分别应用于各种场景。代码可在https://github.com/Saehyung-Lee/PlugIR获取。

英文摘要:

In this paper, we primarily address the issue of dialogue-form context query within the interactive text-to-image retrieval task. Our methodology, PlugIR, actively utilizes the general instruction-following capability of LLMs in two ways. First, by reformulating the dialogue-form context, we eliminate the necessity of fine-tuning a retrieval model on existing visual dialogue data, thereby enabling the use of any arbitrary black-box model. Second, we construct the LLM questioner to generate non-redundant questions about the attributes of the target image, based on the information of retrieval candidate images in the current context. This approach mitigates the issues of noisiness and redundancy in the generated questions. Beyond our methodology, we propose a novel evaluation metric, Best log Rank Integral (BRI), for a comprehensive assessment of the interactive retrieval system. PlugIR demonstrates superior performance compared to both zero-shot and fine-tuned baselines in various benchmarks. Additionally, the two methodologies comprising PlugIR can be flexibly applied together or separately in various situations. Our codes are available at https://github.com/Saehyung-Lee/PlugIR.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑