SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2405.04940 · CVPR · 2024

Harnessing the Power of MLLMs for Transferable Text-to-Image Person ReID

Fei Wang, Yibing Zhan, Changxing Ding, Wentao Tan, Dapeng Tao, Jiayu Jiang

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 10 functions out of this paper's own repositories and ran 9 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
wentaotan/mllm4text-reid canonical 9 of 10
FunctionStatusWhere it lives
build_transforms Ran wentaotan/mllm4text-reid/datasets/build.py
pointer only (licence: NONE) · get_code("cb870e3cbf51d74a")
collate Ran wentaotan/mllm4text-reid/datasets/build.py
pointer only (licence: NONE) · get_code("e1cb4f3d2779e452")
compute_patch Ran wentaotan/mllm4text-reid/model/objectives.py
pointer only (licence: NONE) · get_code("d42185181d52b9d0")
find_mutual_nearest_neighbors Ran wentaotan/mllm4text-reid/model/objectives.py
pointer only (licence: NONE) · get_code("17b7054641464ad3")
mc Ran wentaotan/mllm4text-reid/model/memory.py
pointer only (licence: NONE) · get_code("8fbaad777bcbec96")
remove_punctuation_and_spaces Ran wentaotan/mllm4text-reid/datasets/cuhkpedes.py
pointer only (licence: NONE) · get_code("0770b49b19f8c704")
resize_pos_embed Ran wentaotan/mllm4text-reid/model/clip_model.py
pointer only (licence: NONE) · get_code("489d084ef038d428")
softmax Ran wentaotan/mllm4text-reid/datasets/bases.py
pointer only (licence: NONE) · get_code("c0cf1196424d4f71")
tokenize Ran wentaotan/mllm4text-reid/datasets/bases.py
pointer only (licence: NONE) · get_code("3d862712396854e3")
compute_part Not yet run wentaotan/mllm4text-reid/model/objectives.py
pointer only (licence: NONE) · get_code("ff8611f714fb9669")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Text-to-image person re-identification (ReID) retrieves pedestrian images according to textual descriptions. Manually annotating textual descriptions is time-consuming, restricting the scale of existing datasets and therefore the generalization ability of ReID models. As a result, we study the transferable text-to-image ReID problem, where we train a model on our proposed large-scale database and directly deploy it to various datasets for evaluation. We obtain substantial training data via Multi-modal Large Language Models (MLLMs). Moreover, we identify and address two key challenges in utilizing the obtained textual descriptions. First, an MLLM tends to generate descriptions with similar structures, causing the model to overfit specific sentence patterns. Thus, we propose a novel method that uses MLLMs to caption images according to various templates. These templates are obtained using a multi-turn dialogue with a Large Language Model (LLM). Therefore, we can build a large-scale dataset with diverse textual descriptions. Second, an MLLM may produce incorrect descriptions. Hence, we introduce a novel method that automatically identifies words in a description that do not correspond with the image. This method is based on the similarity between one text and all patch token embeddings in the image. Then, we mask these words with a larger probability in the subsequent training epoch, alleviating the impact of noisy textual descriptions. The experimental results demonstrate that our methods significantly boost the direct transfer text-to-image ReID performance. Benefiting from the pre-trained model weights, we also achieve state-of-the-art performance in the traditional evaluation settings.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2405.04940")
get_code_for_paper("2405.04940")
have("2405.04940")

Connect an agent — have() is free.