arXiv:2508.11272cs.CVcs.AI2025-08

用推理增强视觉表征,提升图像检索精度

Enhancing Supervised Composed Image Retrieval via Reasoning-Augmented Representation Engineering

  • 通过金字塔分块模块增强多粒度视觉理解
  • 无需额外训练,实现比现有方法更高的检索准确率
  • 适合需要高精度图像检索的应用场景

组成图像检索(CIR)要求同时理解参考图像和修改后的文本指令以找到相关目标图像。现有方法多采用两阶段流程,常需额外训练排序模型。尽管思维链(CoT)技术在降低语言模型训练成本方面表现优异,但在CIR任务中应用受限——主要因视觉信息压缩为文本或依赖复杂提示设计。此外,以往工作仅用于零样本CIR,难以在有监督的预训练模型上取得良好效果。本文提出一种无需训练的框架PMTFR,引入金字塔分块模块,增强金字塔匹配模型对多尺度视觉信息的理解能力。受表征工程启发,从CoT数据中提取表征并注入大型视觉语言模型(LVLMs),从而在不依赖显式文本推理的前提下获得优化的检索分数。在多个CIR基准上的大量实验表明,PMTFR在有监督CIR任务中超越现有最佳方法。代码将公开。

原文摘要 · Abstract (English)

Composed Image Retrieval (CIR) presents a significant challenge as it requires jointly understanding a reference image and a modified textual instruction to find relevant target images. Some existing methods attempt to use a two-stage approach to further refine retrieval results. However, this often requires additional training of a ranking model. Despite the success of Chain-of-Thought (CoT) techniques in reducing training costs for language models, their application in CIR tasks remains limited -- compressing visual information into text or relying on elaborate prompt designs. Besides, existing works only utilize it for zero-shot CIR, as it is challenging to achieve satisfactory results in supervised CIR with a well-trained model. In this work, we proposed a framework that includes the Pyramid Matching Model with Training-Free Refinement (PMTFR) to address these challenges. Through a simple but effective module called Pyramid Patcher, we enhanced the Pyramid Matching Model's understanding of visual information at different granularities. Inspired by representation engineering, we extracted representations from COT data and injected them into the LVLMs. This approach allowed us to obtain refined retrieval scores in the Training-Free Refinement paradigm without relying on explicit textual reasoning, further enhancing performance. Extensive experiments on CIR benchmarks demonstrate that PMTFR surpasses state-of-the-art methods in supervised CIR tasks. The code will be made public.

图像检索视觉语言模型推理增强

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