Investigating Multi-Hop Factual Shortcuts in Knowledge Editing of Large Language Models

Tianjie Ju,Yijin Chen,Xinwei Yuan,Zhuosheng Zhang,Wei Du,Yubin Zheng,Gongshen Liu
2024-06-02
Abstract:Recent work has showcased the powerful capability of large language models (LLMs) in recalling knowledge and reasoning. However, the reliability of LLMs in combining these two capabilities into reasoning through multi-hop facts has not been widely explored. This paper systematically investigates the possibilities for LLMs to utilize shortcuts based on direct connections between the initial and terminal entities of multi-hop knowledge. We first explore the existence of factual shortcuts through Knowledge Neurons, revealing that: (i) the strength of factual shortcuts is highly correlated with the frequency of co-occurrence of initial and terminal entities in the pre-training corpora; (ii) few-shot prompting leverage more shortcuts in answering multi-hop questions compared to chain-of-thought prompting. Then, we analyze the risks posed by factual shortcuts from the perspective of multi-hop knowledge editing. Analysis shows that approximately 20% of the failures are attributed to shortcuts, and the initial and terminal entities in these failure instances usually have higher co-occurrences in the pre-training corpus. Finally, we propose erasing shortcut neurons to mitigate the associated risks and find that this approach significantly reduces failures in multiple-hop knowledge editing caused by shortcuts.
Computation and Language
What problem does this paper attempt to address?
This paper aims to explore the possibility of large - language models (LLMs) using factual shortcuts when dealing with multi - hop knowledge problems and their potential risks. Specifically, the paper focuses on the following points: 1. **Research Background**: - Large - language models (such as ChatGPT and LLaMA - 2) show strong capabilities in knowledge recall and reasoning. - However, the reliability of these models when combining multiple facts for reasoning has not been fully explored. 2. **Research Questions**: - **Existence of Multi - Hop Factual Shortcuts**: The paper first explores whether there are factual shortcuts in multi - hop knowledge, that is, whether the model quickly arrives at an answer by directly connecting the initial entity and the terminal entity instead of step - by - step reasoning. - **Impact of Shortcuts**: Analyze the impact of these shortcuts on multi - hop knowledge editing, especially the problem of answer inconsistency that may be caused by shortcuts after knowledge update. - **Risk Assessment**: Evaluate the potential risks of multi - hop factual shortcuts in knowledge editing and propose mitigation strategies. 3. **Research Methods**: - **Frequency Analysis**: Explore the formation mechanism of shortcuts by analyzing the co - occurrence frequency of the initial entity and the terminal entity in the pre - training corpus. - **Neuron Activation Analysis**: Use the Knowledge Neurons (KN) method to quantify the difference in reasoning patterns between multi - hop problems and single - hop problems. - **Experimental Verification**: Verify the use of shortcuts for multi - hop problems under different prompting methods (such as few - shot prompting and chain - of - thought prompting) through experiments. 4. **Main Findings**: - **Relationship between Shortcuts and Co - occurrence Frequency**: The strength of multi - hop factual shortcuts is highly correlated with the co - occurrence frequency of the initial entity and the terminal entity in the pre - training corpus. - **Comparison between Few - Shot Prompting and Chain - of - Thought Prompting**: Few - shot prompting is more likely to use shortcuts than chain - of - thought prompting. - **Risks of Shortcuts**: Approximately 20% of multi - hop knowledge editing failures are attributed to shortcuts, especially when the co - occurrence frequency of the initial and terminal entities is high. 5. **Solutions**: - **Eliminating Shortcuts**: By erasing neurons related to shortcuts, significantly reduce the failure rate in multi - hop knowledge editing and improve the success rate of editing. In conclusion, this paper systematically studies the phenomenon of large - language models using shortcuts in multi - hop knowledge reasoning and proposes corresponding risk assessment and mitigation measures. This provides an important reference for future research and helps to improve the ability and reliability of multi - hop reasoning.