Back to Search Start Over

Self-Retrieval: Building an Information Retrieval System with One Large Language Model

Authors :
Tang, Qiaoyu
Chen, Jiawei
Yu, Bowen
Lu, Yaojie
Fu, Cheng
Yu, Haiyang
Lin, Hongyu
Huang, Fei
He, Ben
Han, Xianpei
Sun, Le
Li, Yongbin
Publication Year :
2024

Abstract

The rise of large language models (LLMs) has transformed the role of information retrieval (IR) systems in the way to humans accessing information. Due to the isolated architecture and the limited interaction, existing IR systems are unable to fully accommodate the shift from directly providing information to humans to indirectly serving large language models. In this paper, we propose Self-Retrieval, an end-to-end, LLM-driven information retrieval architecture that can fully internalize the required abilities of IR systems into a single LLM and deeply leverage the capabilities of LLMs during IR process. Specifically, Self-retrieval internalizes the corpus to retrieve into a LLM via a natural language indexing architecture. Then the entire retrieval process is redefined as a procedure of document generation and self-assessment, which can be end-to-end executed using a single large language model. Experimental results demonstrate that Self-Retrieval not only significantly outperforms previous retrieval approaches by a large margin, but also can significantly boost the performance of LLM-driven downstream applications like retrieval augumented generation.

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2403.00801
Document Type :
Working Paper