Generative Multi-Modal Knowledge Retrieval with Large Language Models

Xinwei Long,Jiali Zeng,Fandong Meng,Zhiyuan Ma,Kaiyan Zhang,Bowen Zhou,Jie Zhou

2024-01-16

Abstract:Knowledge retrieval with multi-modal queries plays a crucial role in supporting knowledge-intensive multi-modal applications. However, existing methods face challenges in terms of their effectiveness and training efficiency, especially when it comes to training and integrating multiple retrievers to handle multi-modal queries. In this paper, we propose an innovative end-to-end generative framework for multi-modal knowledge retrieval. Our framework takes advantage of the fact that large language models (LLMs) can effectively serve as virtual knowledge bases, even when trained with limited data. We retrieve knowledge via a two-step process: 1) generating knowledge clues related to the queries, and 2) obtaining the relevant document by searching databases using the knowledge clue. In particular, we first introduce an object-aware prefix-tuning technique to guide multi-grained visual learning. Then, we align multi-grained visual features into the textual feature space of the LLM, employing the LLM to capture cross-modal interactions. Subsequently, we construct instruction data with a unified format for model training. Finally, we propose the knowledge-guided generation strategy to impose prior constraints in the decoding steps, thereby promoting the generation of distinctive knowledge clues. Through experiments conducted on three benchmarks, we demonstrate significant improvements ranging from 3.0% to 14.6% across all evaluation metrics when compared to strong baselines.

Information Retrieval,Computation and Language

What problem does this paper attempt to address?

This paper proposes a new approach to solve the problem of multimodal knowledge retrieval, especially for the case of using multimodal queries such as text and images. Current methods face challenges in terms of effectiveness and training efficiency, especially when it is necessary to train and integrate multiple retrievers to handle multimodal queries. The paper presents an innovative end-to-end generation framework called GeMKR, which utilizes large-scale language models (LLMs) as a virtual knowledge base to effectively serve even with limited data.

In this framework, knowledge retrieval is divided into two steps: first, generating knowledge clues related to the query, and then using these clues to search the database for relevant documents. By introducing object-aware prefix fine-tuning technique to guide multi-granularity visual learning and aligning the multi-granularity visual features to the text feature space of LLM, it captures cross-modal interactions. In addition, they construct a unified format instruction data to train the model and propose a knowledge-guided generation strategy to impose prior constraints in the decoding step, promoting the generation of unique knowledge clues.

Experiments on three benchmark datasets demonstrate significant improvements, ranging from 3.0% to 14.6% in all evaluation metrics, compared to strong baselines. This approach overcomes the issues of existing methods in representing multimodal knowledge and inefficient handling of large-scale corpora, proving the model's good generalization ability in dealing with large-scale knowledge sources.