arXiv:2409.01936cs.CVcs.LG2024-09被引 16

优化CLIP图像检索能力,保持图文对齐不降级。

Optimizing CLIP Models for Image Retrieval with Maintained Joint-Embedding Alignment

  • 分步微调图像编码器并重校文本编码器,提升图像相似度区分力。
  • 引入伪描述增强嵌入空间对齐,在多个基准上性能全面提升。
  • 适合需要高效大规模多模态搜索的工程应用,部署更简单。

对比语言与图像配对(CLIP)是多媒体检索领域的革新方法,通常通过并行训练两个神经网络生成文本与图像对的联合嵌入。然而,直接应用时,模型常难以区分视觉差异大但描述相似的图像,导致基于图像的相似性搜索表现不佳。本文针对多种图像相似性搜索场景,优化CLIP模型,同时保持其在文本搜索任务(如文到图检索、零样本分类)中的有效性。提出两种新方法:一是先优化图像编码器以提升图像检索精度,再重新对齐文本编码器;二是在检索优化阶段引入伪描述,促进嵌入空间内的直接对齐。大量实验表明,该方法在图像检索、k-NN分类和零样本文本分类等基准上均有提升,且维持文到图检索鲁棒性。优化后的模型仅需为每张图像保存一个嵌入,显著简化了大规模多模态相似性搜索系统的基础设施。

原文摘要 · Abstract (English)

Contrastive Language and Image Pairing (CLIP), a transformative method in multimedia retrieval, typically trains two neural networks concurrently to generate joint embeddings for text and image pairs. However, when applied directly, these models often struggle to differentiate between visually distinct images that have similar captions, resulting in suboptimal performance for image-based similarity searches. This paper addresses the challenge of optimizing CLIP models for various image-based similarity search scenarios, while maintaining their effectiveness in text-based search tasks such as text-to-image retrieval and zero-shot classification. We propose and evaluate two novel methods aimed at refining the retrieval capabilities of CLIP without compromising the alignment between text and image embeddings. The first method involves a sequential fine-tuning process: initially optimizing the image encoder for more precise image retrieval and subsequently realigning the text encoder to these optimized image embeddings. The second approach integrates pseudo-captions during the retrieval-optimization phase to foster direct alignment within the embedding space. Through comprehensive experiments, we demonstrate that these methods enhance CLIP's performance on various benchmarks, including image retrieval, k-NN classification, and zero-shot text-based classification, while maintaining robustness in text-to-image retrieval. Our optimized models permit maintaining a single embedding per image, significantly simplifying the infrastructure needed for large-scale multi-modal similarity search systems.

CLIP图像检索多模态嵌入优化

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