4 个月前

无需训练的零样本组合图像检索通过加权模态融合和相似性实现

无需训练的零样本组合图像检索通过加权模态融合和相似性实现

摘要

组合图像检索(CIR)通过将查询表示为参考图像和修改文本的组合,因其增强的捕捉用户意图的能力而成为一种新的图像搜索形式。然而,以监督方式训练CIR模型通常需要大量的人工收集(参考图像、文本修饰符、目标图像)三元组。尽管现有的零样本CIR(ZS-CIR)方法消除了对特定下游数据集进行训练的需求,但它们仍然需要在大规模图像数据集上进行额外的预训练。在本文中,我们介绍了一种无需训练的ZS-CIR方法。我们的方法——基于加权模态融合和相似性的CIR(WeiMoCIR)——假设图像和文本模态可以通过简单的加权平均有效结合。这使得查询表示可以直接从参考图像和文本修饰符构建。为了进一步提高检索性能,我们利用多模态大语言模型(MLLMs)生成数据库图像的标题,并通过加权平均将这些文本标题与图像信息结合,纳入相似性计算中。我们的方法简单易实现,并且其有效性已在FashionIQ和CIRR数据集上的实验中得到验证。代码可在https://github.com/whats2000/WeiMoCIR 获取。

代码仓库

whats2000/WeiMoCIR
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
zero-shot-composed-image-retrieval-zs-cir-on-1WeiMoCIR (CLIP G/14)
R@1: 31.04
R@5: 60.41
zero-shot-composed-image-retrieval-zs-cir-on-1WeiMoCIR (CLIP B/32)
R@1: 26.31
R@5: 57.69
zero-shot-composed-image-retrieval-zs-cir-on-1WeiMoCIR (CLIP H/14)
R@1: 29.11
R@5: 59.76
zero-shot-composed-image-retrieval-zs-cir-on-1WeiMoCIR (CLIP L/14)
R@1: 30.94
R@5: 60.87
zero-shot-composed-image-retrieval-zs-cir-on-2WeiMoCIR (CLIP G/14)
(Recall@10+Recall@50)/2: 47.16
zero-shot-composed-image-retrieval-zs-cir-on-2WeiMoCIR (CLIP H/14)
(Recall@10+Recall@50)/2: 44.58
zero-shot-composed-image-retrieval-zs-cir-on-2WeiMoCIR (CLIP L/14)
(Recall@10+Recall@50)/2: 41.27
zero-shot-composed-image-retrieval-zs-cir-on-2WeiMoCIR (CLIP B/32)
(Recall@10+Recall@50)/2: 39.84

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
无需训练的零样本组合图像检索通过加权模态融合和相似性实现 | 论文 | HyperAI超神经