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

面向大语言模型的神经元级序列编辑

Neuron-Level Sequential Editing for Large Language Models

University of Science and Technology of China(中国科学技术大学);National University of Singapore(新加坡国立大学);Bytedance(字节跳动)

Houcheng Jiang, Junfeng Fang, Tianyu Zhang, An Zhang, Ruipeng Wang, Tao Liang, Xiang Wang

AI总结:

针对大语言模型序列编辑中现有参数修改法易出现遗忘与失效的问题,提出神经元级序列编辑方法NSE,通过原始权重优化隐藏状态、基于激活值选神经元编辑,性能显著优于现有方法。

摘要:

本研究探讨大语言模型(LLM)的序列模型编辑任务,该任务至关重要,需通过多轮编辑持续修改LLM内部知识,每轮编辑融入更新或修正内容以调整模型输出,无需成本高昂的重新训练。现有模型编辑方法,尤其是修改模型参数的方法,通常聚焦单轮编辑,在序列模型编辑中往往面临显著挑战——最突出的是模型遗忘和失效问题。为应对这些挑战,我们提出一种全新的模型编辑方法,即神经元级序列编辑(Neuron-level Sequential Editing,NSE),专门用于支持序列模型编辑。具体而言,我们利用模型原始权重优化目标层的隐藏状态,以防止模型失效。此外,我们基于激活值迭代选择多层中的神经元进行编辑,以缓解模型遗忘问题。实证实验表明,NSE的性能显著优于当前的参数修改类模型编辑方法,标志着序列模型编辑领域取得了实质性进展。我们的代码已发布于https://github.com/jianghoucheng/NSE。

英文摘要:

This work explores sequential model editing in large language models (LLMs), a critical task that involves modifying internal knowledge within LLMs continuously through multi-round editing, each incorporating updates or corrections to adjust the model outputs without the need for costly retraining. Existing model editing methods, especially those that alter model parameters, typically focus on single-round editing and often face significant challenges in sequential model editing-most notably issues of model forgetting and failure. To address these challenges, we introduce a new model editing method, namely \textbf{N}euron-level \textbf{S}equential \textbf{E}diting (NSE), tailored for supporting sequential model editing. Specifically, we optimize the target layer's hidden states using the model's original weights to prevent model failure. Furthermore, we iteratively select neurons in multiple layers for editing based on their activation values to mitigate model forgetting. Our empirical experiments demonstrate that NSE significantly outperforms current modifying parameters model editing methods, marking a substantial advancement in the field of sequential model editing. Our code is released on \url{https://github.com/jianghoucheng/NSE}.

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

↑