Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LamRA: Large Multimodal Model as Your Advanced Retrieval Assistant

About

With the rapid advancement of multimodal information retrieval, increasingly complex retrieval tasks have emerged. Existing methods predominately rely on task-specific fine-tuning of vision-language models, often those trained with image-text contrastive learning. In this paper, we explore the possibility of re-purposing generative Large Multimodal Models (LMMs) for retrieval. This approach enables unifying all retrieval tasks under the same formulation and, more importantly, allows for extrapolation towards unseen retrieval tasks without additional training. Our contributions can be summarised in the following aspects: (i) We introduce LamRA, a versatile framework designed to empower LMMs with sophisticated retrieval and reranking capabilities. (ii) For retrieval, we adopt a two-stage training strategy comprising language-only pre-training and multimodal instruction tuning to progressively enhance LMM's retrieval performance. (iii) For reranking, we employ joint training for both pointwise and listwise reranking, offering two distinct ways to further boost the retrieval performance. (iv) Extensive experimental results underscore the efficacy of our method in handling more than ten retrieval tasks, demonstrating robust performance in both supervised and zero-shot settings, including scenarios involving previously unseen retrieval tasks.

Yikun Liu, Pingan Chen, Jiayin Cai, Xiaolong Jiang, Yao Hu, Jiangchao Yao, Yanfeng Wang, Weidi Xie• 2024

Related benchmarks

TaskDatasetResultRank
Text-to-Video RetrievalDiDeMo (test)
R@146.3
399
Text-to-Video RetrievalMSR-VTT
Recall@144.7
369
Text-to-Video RetrievalMSVD
R@152.4
264
Text-to-Video RetrievalMSR-VTT (test)
R@148.9
255
Text-to-Video RetrievalMSVD (test)
R@155.7
204
Video-to-Text retrievalMSR-VTT
Recall@150.9
185
Video-to-Text retrievalDiDeMo
R@147.1
130
Text-to-Video RetrievalActivityNet (test)
R@158.5
108
Video-to-Text retrievalMSVD
R@181.2
93
Visual Question AnsweringE-VQA (test)
Accuracy56.2
85
Showing 10 of 86 rows
...

Other info

Code

Follow for update