arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2305.12964cs.CV
收藏

无需并行图文数据的基于文本的行人搜索

Text-based Person Search without Parallel Image-Text Data

Soochow University(苏州大学);Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所);Harbin Institute of Technology(哈尔滨工业大学)

Yang Bai, Jingyao Wang, Min Cao, Chen Chen, Ziqiang Cao, Liqiang Nie, Min Zhang

AI总结:

本文首次研究无需并行图文数据的行人搜索,提出先为图像生成伪文本再进行检索的GTR两阶段框架,并通过细粒度描述生成和置信度训练缓解噪声,在多个基准上取得有前景的性能。

摘要:

基于文本的行人搜索(TBPS)旨在根据给定的自然语言描述,从大规模图像库中检索目标人物的图像。现有方法主要依赖并行图文对训练模型,而这类数据的收集成本非常高。本文首次尝试探索无需并行图文数据的TBPS(μ-TBPS),在该设定下只能使用非并行的图像和文本,甚至仅使用图像数据。为此,我们提出一个两阶段框架,即先生成后检索(GTR),首先为每张图像生成对应的伪文本,然后以有监督方式进行检索。在生成阶段,我们提出一种细粒度图像描述策略,以获得人物图像的丰富描述:首先利用一组指令提示激活现成的预训练视觉-语言模型,以捕获并生成细粒度的人物属性,然后通过微调后的大语言模型或手工模板将提取的属性转换为文本描述。在检索阶段,考虑到生成文本对训练模型的噪声干扰,我们开发了一种基于置信度分数的训练方案,使更可靠的文本在训练过程中贡献更大。在多个TBPS基准(即CUHK-PEDES、ICFG-PEDES和RSTPReid)上的实验结果表明,所提出的GTR无需依赖并行图文数据即可取得有前景的性能。

英文摘要:

Text-based person search (TBPS) aims to retrieve the images of the target person from a large image gallery based on a given natural language description. Existing methods are dominated by training models with parallel image-text pairs, which are very costly to collect. In this paper, we make the first attempt to explore TBPS without parallel image-text data ($μ$-TBPS), in which only non-parallel images and texts, or even image-only data, can be adopted. Towards this end, we propose a two-stage framework, generation-then-retrieval (GTR), to first generate the corresponding pseudo text for each image and then perform the retrieval in a supervised manner. In the generation stage, we propose a fine-grained image captioning strategy to obtain an enriched description of the person image, which firstly utilizes a set of instruction prompts to activate the off-the-shelf pretrained vision-language model to capture and generate fine-grained person attributes, and then converts the extracted attributes into a textual description via the finetuned large language model or the hand-crafted template. In the retrieval stage, considering the noise interference of the generated texts for training model, we develop a confidence score-based training scheme by enabling more reliable texts to contribute more during the training. Experimental results on multiple TBPS benchmarks (i.e., CUHK-PEDES, ICFG-PEDES and RSTPReid) show that the proposed GTR can achieve a promising performance without relying on parallel image-text data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑