用少量例子提升文本查图的准确率,解决复杂查询难题。
Few Shots Text to Image Retrieval: New Benchmarking Dataset and Optimization Methods
- 设计新任务与数据集,支持少样本文本查图
- 在复杂和分布外查询上提升平均精度
- 方法可适配现有大模型,适合视觉语言研究者
预训练视觉语言模型在多模态任务中表现优异,通常将图像编码为嵌入向量存储于数据库,并通过近似最近邻搜索(ANNS)进行检索。然而,这些模型在组合式查询和分布外(OOD)图像-文本对上表现不佳。受人类认知能从极少示例中学习的启发,我们提出针对图像检索的少样本学习方法,以弥补性能差距。本文引入少样本文本到图像检索(FSIR)任务及其配套基准数据集FSIR-BD——首个明确聚焦于带参考示例的文本查图任务,特别关注具有挑战性的组合式与分布外查询。组合部分分为城市场景和自然物种,涵盖特定情境或显著特征。FSIR-BD包含38,353张图像和303个查询,其中82%为测试集(每查询平均含37个正样本和大量难负样本),18%为少样本参考集(FSR),包含正样本与难负样本示例。此外,我们提出两种新颖的检索优化方法,利用单样本或少样本参考示例提升性能。两种方法兼容任意预训练图像编码器,适用于现有大规模环境。实验表明:(1)FSIR-BD为图像检索提供了具有挑战性的基准;(2)所提方法在平均精度(mAP)上优于现有基线。未来对FSIR优化方法的研究有助于缩小机器与人类在有限示例下组合推理能力的差距。
原文摘要 · Abstract (English)
Pre-trained vision-language models (VLMs) excel in multimodal tasks, commonly encoding images as embedding vectors for storage in databases and retrieval via approximate nearest neighbor search (ANNS). However, these models struggle with compositional queries and out-of-distribution (OOD) image-text pairs. Inspired by human cognition's ability to learn from minimal examples, we address this performance gap through few-shot learning approaches specifically designed for image retrieval. We introduce the Few-Shot Text-to-Image Retrieval (FSIR) task and its accompanying benchmark dataset, FSIR-BD - the first to explicitly target image retrieval by text accompanied by reference examples, focusing on the challenging compositional and OOD queries. The compositional part is divided to urban scenes and nature species, both in specific situations or with distinctive features. FSIR-BD contains 38,353 images and 303 queries, with 82% comprising the test corpus (averaging per query 37 positives, ground truth matches, and significant number of hard negatives) and 18% forming the few-shot reference corpus (FSR) of exemplar positive and hard negative images. Additionally, we propose two novel retrieval optimization methods leveraging single shot or few shot reference examples in the FSR to improve performance. Both methods are compatible with any pre-trained image encoder, making them applicable to existing large-scale environments. Our experiments demonstrate that: (1) FSIR-BD provides a challenging benchmark for image retrieval; and (2) our optimization methods outperform existing baselines as measured by mean Average Precision (mAP). Further research into FSIR optimization methods will help narrow the gap between machine and human-level understanding, particularly for compositional reasoning from limited examples.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。