On the Role of Long-tail Knowledge in Retrieval Augmented Large Language Models

Dongyang Li,Junbing Yan,Taolin Zhang,Chengyu Wang,Xiaofeng He,Longtao Huang,Hui Xue,Jun Huang
2024-06-24
Abstract:Retrieval augmented generation (RAG) exhibits outstanding performance in promoting the knowledge capabilities of large language models (LLMs) with retrieved documents related to user queries. However, RAG only focuses on improving the response quality of LLMs via enhancing queries indiscriminately with retrieved information, paying little attention to what type of knowledge LLMs really need to answer original queries more accurately. In this paper, we suggest that long-tail knowledge is crucial for RAG as LLMs have already remembered common world knowledge during large-scale pre-training. Based on our observation, we propose a simple but effective long-tail knowledge detection method for LLMs. Specifically, the novel Generative Expected Calibration Error (GECE) metric is derived to measure the ``long-tailness'' of knowledge based on both statistics and semantics. Hence, we retrieve relevant documents and infuse them into the model for patching knowledge loopholes only when the input query relates to long-tail knowledge. Experiments show that, compared to existing RAG pipelines, our method achieves over 4x speedup in average inference time and consistent performance improvement in downstream tasks.
Information Retrieval
What problem does this paper attempt to address?
The paper aims to address the efficiency and accuracy issues of Retrieval-Augmented Generation (RAG) in large-scale language models (LLMs). Specifically, the paper points out that current RAG methods mainly focus on enhancing response quality through information retrieval but lack detailed handling of the types of knowledge required. The paper argues that long-tail knowledge is crucial for RAG because LLMs have already memorized common world knowledge during large-scale pre-training. Based on this observation, the authors propose a simple and effective method for detecting long-tail knowledge and introduce a new metric called Generation Expected Calibration Error (GECE) to measure the "long-tailness" of knowledge. Experimental results show that compared to existing RAG pipelines, this method not only improves the performance of downstream tasks but also significantly reduces inference time, with an average speedup of over 4 times.