arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.13593cs.CL
收藏

知识图谱增强的大语言模型编辑

Knowledge Graph Enhanced Large Language Model Editing

School of Computer Science and Technology, Shandong University(山东大学计算机科学与技术学院);School of Computer Science, Beijing University of Posts and Telecommunications(北京邮电大学计算机学院)

Mengqi Zhang, Xiaotian Ye, Qiang Liu, Pengjie Ren, Shu Wu, Zhumin Chen

AI总结:

本文提出GLAME方法,通过知识图谱增强模块和基于图的知识编辑模块,将结构化知识变化整合进大语言模型编辑中,从而提升编辑后模型运用已编辑知识的泛化能力。

摘要:

大语言模型(LLMs)在推进自然语言处理(NLP)任务中起着关键作用,但其效能受到不准确和过时知识的制约。模型编辑成为应对这些挑战的一种有前景的解决方案。然而,现有的编辑方法难以跟踪和整合与编辑相关的知识变化,这限制了编辑后LLMs在处理已编辑知识时的泛化能力。为了解决这些问题,我们提出了一种新颖的模型编辑方法,利用知识图谱来增强LLM编辑,即GLAME。具体而言,我们首先利用知识图谱增强模块来揭示因编辑而发生变化的相关知识,并获取其在LLMs中的内部表示。这种方法使得LLMs内部的知识变更能够通过外部图结构得以反映。随后,我们设计了一个基于图的知识编辑模块,将结构化知识整合到模型编辑中。这确保了更新后的参数不仅反映已编辑知识的修改,还反映编辑过程引起的其他相关知识的变化。在GPT-J和GPT-2 XL上进行的综合实验表明,GLAME显著提升了编辑后LLMs在使用已编辑知识时的泛化能力。

英文摘要:

Large language models (LLMs) are pivotal in advancing natural language processing (NLP) tasks, yet their efficacy is hampered by inaccuracies and outdated knowledge. Model editing emerges as a promising solution to address these challenges. However, existing editing methods struggle to track and incorporate changes in knowledge associated with edits, which limits the generalization ability of postedit LLMs in processing edited knowledge. To tackle these problems, we propose a novel model editing method that leverages knowledge graphs for enhancing LLM editing, namely GLAME. Specifically, we first utilize a knowledge graph augmentation module to uncover associated knowledge that has changed due to editing, obtaining its internal representations within LLMs. This approach allows knowledge alterations within LLMs to be reflected through an external graph structure. Subsequently, we design a graph-based knowledge edit module to integrate structured knowledge into the model editing. This ensures that the updated parameters reflect not only the modifications of the edited knowledge but also the changes in other associated knowledge resulting from the editing process. Comprehensive experiments conducted on GPT-J and GPT-2 XL demonstrate that GLAME significantly improves the generalization capabilities of post-edit LLMs in employing edited knowledge.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑