Exploring Knowledge Boundaries in Large Language Models for Retrieval Judgment

Zhen Zhang,Xinyu Wang,Yong Jiang,Zhuo Chen,Feiteng Mu,Mengting Hu,Pengjun Xie,Fei Huang
2024-11-09
Abstract:Large Language Models (LLMs) are increasingly recognized for their practical applications. However, these models often encounter challenges in dynamically changing knowledge, as well as in managing unknown static knowledge. Retrieval-Augmented Generation (RAG) tackles this challenge and has shown a significant impact on LLMs. Actually, we find that the impact of RAG on the question answering capabilities of LLMs can be categorized into three groups: beneficial, neutral, and harmful. By minimizing retrieval requests that yield neutral or harmful results, we can effectively reduce both time and computational costs, while also improving the overall performance of LLMs. This insight motivates us to differentiate between types of questions using certain metrics as indicators, to decrease the retrieval ratio without compromising performance. In our work, we propose a method that is able to identify different types of questions from this view by training a Knowledge Boundary Model (KBM). Experiments conducted on 11 English and Chinese datasets illustrate that the KBM effectively delineates the knowledge boundary, significantly decreasing the proportion of retrievals required for optimal end-to-end performance. Specifically, we evaluate the effectiveness of KBM in three complex scenarios: dynamic knowledge, long-tail static knowledge, and multi-hop problems, as well as its functionality as an external LLM plug-in.
Computation and Language
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the challenges faced by large - language models (LLMs) when dealing with dynamically changing knowledge and unknown static knowledge. Although the Retrieval - Augmented Generation (RAG) technique can effectively address these challenges, RAG also brings additional time and computational costs. The authors of the paper observe that the impact of RAG on the question - answering ability of LLMs can be divided into three categories: beneficial, neutral, and harmful. By reducing retrieval requests that lead to neutral or harmful results, time and computational costs can be effectively reduced while improving the overall performance of LLMs. Based on this insight, the authors propose a method to identify different types of question - answering requests by training a Knowledge Boundary Model (KBM), thereby reducing the retrieval ratio without sacrificing performance. Experimental results show that KBM can effectively demarcate knowledge boundaries and significantly reduce the retrieval proportion required to achieve optimal end - to - end performance. In particular, in complex scenarios such as dynamic knowledge, long - tail static knowledge, and multi - hop questions, KBM has shown good performance. In addition, KBM can also be used as an external plug - in to enhance the capabilities of LLMs.