Our new X account is live! Follow @wizwand_team for updates
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
376
Text-to-Video RetrievalMSR-VTT
Recall@144.7
313
Text-to-Video RetrievalMSR-VTT (test)
R@148.9
234
Text-to-Video RetrievalMSVD
R@152.4
218
Text-to-Video RetrievalMSVD (test)
R@155.7
204
Video-to-Text retrievalMSR-VTT
Recall@150.9
157
Text-to-Video RetrievalActivityNet (test)
R@158.5
108
Video-to-Text retrievalDiDeMo
R@147.1
108
Video-to-Text retrievalMSVD
R@181.2
93
Video-to-Text retrievalVATEX
Recall@182.5
68
Showing 10 of 69 rows

Other info

Code

Follow for update