arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.01751cs.AIcs.CLcs.LG
收藏

SAKE: Steering Activations for Knowledge Editing(用于知识编辑的激活引导方法)

SAKE: Steering Activations for Knowledge Editing

EPFL(苏黎世联邦理工学院);AXA(安盛集团);TRAIL, LIP6, Sorbonne Université(TRAIL、LIP6、索邦大学);Polish Academy of Science, IBS PAN(波兰科学院、IBS PAN)

Marco Scialanga, Thibault Laugel, Vincent Grari, Marcin Detyniecki

AI总结:

针对现有知识编辑方法上下文鲁棒性不足、无法泛化逻辑推论的问题,提出将待编辑事实建模为分布、借助最优传输调整相关分布上LLM行为的SAKE方法,实验证明其编辑鲁棒性优于现有方法。

摘要:

由于大语言模型已被证明能够记忆现实世界的事实,人们需要以可控且高效的方式更新这些知识。知识编辑(Knowledge Editing,KE)方法正是基于这些约束设计的,旨在修改预训练模型中的特定事实。然而,现有研究表明这类方法存在若干局限性,包括缺乏上下文鲁棒性,以及无法泛化到与该事实相关的逻辑推论。为解决这些问题,我们提出SAKE——一种激活引导方法,它将待编辑的事实建模为一个分布而非单个提示。借助最优传输(Optimal Transport)技术,SAKE在整个事实相关分布(定义为释义和逻辑推论)上改变大语言模型的行为。多项数值实验证明了该方法的有效性:SAKE能够实现比现有同类方法更鲁棒的编辑效果。

英文摘要:

As Large Langue Models have been shown to memorize real-world facts, the need to update this knowledge in a controlled and efficient manner arises. Designed with these constraints in mind, Knowledge Editing (KE) approaches propose to alter specific facts in pretrained models. However, they have been shown to suffer from several limitations, including their lack of contextual robustness and their failure to generalize to logical implications related to the fact. To overcome these issues, we propose SAKE, a steering activation method that models a fact to be edited as a distribution rather than a single prompt. Leveraging Optimal Transport, SAKE alters the LLM behavior over a whole fact-related distribution, defined as paraphrases and logical implications. Several numerical experiments demonstrate the effectiveness of this method: SAKE is thus able to perform more robust edits than its existing counterparts.

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

↑