3 个月前

LDRE:基于LLM的发散推理与集成方法用于零样本组合图像检索

LDRE:基于LLM的发散推理与集成方法用于零样本组合图像检索

摘要

零样本组合图像检索(Zero-Shot Composed Image Retrieval, ZS-CIR)近年来受到越来越多关注,其目标是在无训练样本的情况下,基于由参考图像与修改文本组成的查询,检索出目标图像。具体而言,修改文本描述了参考图像与目标图像之间的语义差异。当前主流方法通常采用预训练的图文模型将查询中的图像与文本统一转换为单一文本表示,再通过CLIP模型将其投影至共享特征空间以实现目标图像的检索。然而,这些方法忽视了ZS-CIR本质上是一种模糊检索任务——目标图像的语义并未被查询图像与文本严格限定。为克服这一局限,本文提出一种无需训练的基于大语言模型(LLM)的发散推理与集成方法(Divergent Reasoning and Ensemble, LDRE),以捕捉组合结果的多种潜在语义。首先,我们利用预训练的图像描述生成模型,为参考图像生成多视角的密集描述(dense captions),以涵盖参考图像的不同语义维度。随后,通过提示(prompt)大语言模型(LLMs),基于这些密集描述与修改文本进行发散式组合推理,生成多条涵盖目标组合图像可能语义的编辑后描述(divergent edited captions)。最后,设计一种发散描述集成机制,根据语义相关性得分对各编辑描述进行加权,融合得到最终的集成描述特征,并将其用于CLIP特征空间中的目标图像检索。在三个公开数据集上的大量实验表明,所提出的LDRE方法取得了当前最优的性能,达到了新的技术水平。

基准测试

基准方法指标
zero-shot-composed-image-retrieval-zs-cir-onLDRE (CLIP L/14)
mAP@10: 24.03
zero-shot-composed-image-retrieval-zs-cir-onLDRE (CLIP B/32)
mAP@10: 18.32
zero-shot-composed-image-retrieval-zs-cir-onLDRE (CLIP G/14)
mAP@10: 32.24
zero-shot-composed-image-retrieval-zs-cir-on-1LDRE (CLIP L/14)
R@5: 55.57
zero-shot-composed-image-retrieval-zs-cir-on-1LDRE (CLIP G/14)
R@5: 66.39
zero-shot-composed-image-retrieval-zs-cir-on-1LDRE (CLIP B/32)
R@5: 55.13
zero-shot-composed-image-retrieval-zs-cir-on-2LDRE (CLIP G/14)
(Recall@10+Recall@50)/2: 43.98

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
LDRE:基于LLM的发散推理与集成方法用于零样本组合图像检索 | 论文 | HyperAI超神经