arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.06331cs.CLcs.AIcs.LG
收藏

Locate-then-edit用于知识编辑下的多跳事实召回

Locate-then-edit for Multi-hop Factual Recall under Knowledge Editing

Peking University(北京大学);Provable Responsible AI and Data Analytics (PRADA) Lab(可验证负责任人工智能与数据分析(PRADA)实验室);King Abdullah University of Science and Technology(阿卜杜拉国王科技大学);South China University of Technology(华南理工大学)

Zhuoran Zhang, Yongxiang Li, Zijian Kan, Keyuan Cheng, Lijie Hu, Di Wang

AI总结:

针对LLMs多跳事实召回任务中现有知识编辑方法仅修改浅层MLP的局限,提出IFMET方法,通过结合单跳与多跳编辑提示同时编辑浅层和深层MLP,显著提升多跳事实召回性能。

摘要:

Locate-then-edit范式在大语言模型(LLMs)的知识编辑(KE)中展现出显著潜力。尽管先前的方法在单跳事实召回任务上表现良好,但它们在涉及新编辑知识的多跳事实召回任务上始终难以应对。本文利用机制可解释性工具,首先发现在多跳任务中,LLMs倾向于从较深的MLP层中检索具有隐式主语信息的知识,而单跳任务则依赖浅层。这一区别解释了当前方法在多跳查询中表现不佳的原因,因为它们主要关注使用单跳编辑提示编辑浅层,而保持深层不变。为解决此问题,我们提出IFMET,一种新颖的locate-then-edit KE方法,旨在同时编辑浅层和深层MLP层。除单跳编辑提示外,IFMET进一步结合多跳编辑提示,以定位和修改不同推理阶段的知识。实验结果表明,IFMET显著提升了多跳事实召回任务的性能,克服了先前locate-then-edit方法的局限性。

英文摘要:

The locate-then-edit paradigm has shown significant promise for knowledge editing (KE) in Large Language Models (LLMs). While previous methods perform well on single-hop fact recall tasks, they consistently struggle with multi-hop factual recall tasks involving newly edited knowledge. In this paper, leveraging tools in mechanistic interpretability, we first identify that in multi-hop tasks, LLMs tend to retrieve knowledge with implicit subject information from deeper MLP layers, unlike single-hop tasks, which rely on shallow layers. This distinction explains the poor performance of current methods in multi-hop queries, as they primarily focus on editing shallow layers with single-hop edit prompts, leaving deeper layers unchanged. To address this, we propose IFMET, a novel locate-then-edit KE approach designed to edit both shallow and deep MLP layers. Beyond single-hop editing prompts, IFMET further incorporates multi-hop editing prompts to locate and modify knowledge across different stages of reasoning. Experimental results demonstrate that IFMET significantly improves performance on multi-hop factual recall tasks, overcoming the limitations of previous locate-then-edit methods

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

↑