论文精选

生成式检索方法 GTR+ 用于无监督文本检索人物

Generative Retrieval for Unsupervised Text-Based Person Search

精选理由

朋友A对朋友B说:最近看到个挺有意思的论文,叫GTR+,是做无监督文本检索人物的,效果还不错,代码也开源了。

这篇论文提出了一种名为 GTR+ 的两阶段生成-检索框架,用于无监督文本检索人物。在生成阶段,它通过三级流程生成细粒度和风格多样的文本描述,从基础属性到高级风格化扩展。在检索阶段,它使用自适应置信度加权检索学习框架来处理伪文本噪声。此外,还贡献了一个名为 LargeFine-Person 的大规模 TBPS 数据集,用于无监督设置下的预训练基准。

原文 · arXiv cs.AI

Generative Retrieval for Unsupervised Text-Based Person Search

Text-based person search (TBPS) aims to retrieve images of a target person from a large image gallery based on a given natural language description. Most existing methods rely on supervised learning with manually annotated image-text pairs. In this paper, we explore unsupervised TBPS, with only unlabeled images. We propose GTR+, a two-stage generation-then-retrieval framework. In the generation stage, we introduce a tiered description generation framework designed to produce fine-grained and stylistically diverse textual descriptions through a three-tier sequential process. The base tier leverages an automated question-and-answer mechanism to generate basic visual attribute descriptions; the intermediate tier enhances fine-grained detail using an inter-sample contrastive mechanism; the advanced tier further enriches textual diversity via a stylized expansion mechanism. In the retrieval stage, to mitigate the impact of noisy pseudo texts, we develop an adaptive confidence-weighted retrieval learning framework. We model image-text pairs as clean or noisy using a Gaussian Mixture Model, calibrated by real-time image-text similarity and static text generation probability from the prior stage, yielding adaptive sample weights during training. Beyond that, we also contribute LargeFine-Person, a large-scale TBPS dataset with high-quality, fine-grained, and diverse textual annotations, enabling a practical and generalizable TBPS pre-training benchmark under unsupervised setting. Experiments on multiple TBPS benchmarks demonstrate the effectiveness and generalization of both GTR+ and LargeFine-Person. Code is available at: https://github.com/Flame-Chasers/GTR.