Retrieval-enhanced Knowledge Editing in Language Models for Multi-Hop Question Answering
This addresses the challenge of integrating real-time knowledge for multi-hop question answering in language models, representing an incremental improvement with a novel retrieval method.
The paper tackles the problem of large language models struggling with outdated or inaccurate knowledge in multi-hop question answering by proposing the Retrieval-Augmented model Editing (RAE) framework, which retrieves edited facts and refines the model through in-context learning, validated across various LLMs to provide accurate answers with updated knowledge.
Large Language Models (LLMs) have shown proficiency in question-answering tasks but often struggle to integrate real-time knowledge, leading to potentially outdated or inaccurate responses. This problem becomes even more challenging when dealing with multi-hop questions, since they require LLMs to update and integrate multiple knowledge pieces relevant to the questions. To tackle the problem, we propose the Retrieval-Augmented model Editing (RAE) framework for multi-hop question answering. RAE first retrieves edited facts and then refines the language model through in-context learning. Specifically, our retrieval approach, based on mutual information maximization, leverages the reasoning abilities of LLMs to identify chain facts that traditional similarity-based searches might miss. In addition, our framework includes a pruning strategy to eliminate redundant information from the retrieved facts, which enhances the editing accuracy and mitigates the hallucination problem. Our framework is supported by theoretical justification for its fact retrieval efficacy. Finally, comprehensive evaluation across various LLMs validates RAE's ability in providing accurate answers with updated knowledge. Our code is available at: https://github.com/sycny/RAE.