让大模型持续改错不重训,还能避免遗忘知识。
Neuron-Level Sequential Editing for Large Language Models
- 通过优化隐藏层状态实现逐轮编辑,不修改原始参数。
- 迭代选择高激活神经元编辑,显著降低知识遗忘率。
- 适合需要长期维护知识的大模型应用者使用。
本文研究大语言模型中的连续模型编辑任务,即通过多轮更新调整模型内部知识,无需重新训练即可修正输出错误。现有方法多聚焦单轮编辑,面临模型遗忘和失效等挑战。为此,我们提出神经元级连续编辑(NSE)方法:利用模型原始权重优化目标层隐藏状态,防止模型崩溃;并基于各层神经元激活值,迭代选择关键神经元进行编辑,有效缓解遗忘问题。实验证明,NSE显著优于当前参数修改类编辑方法,为连续编辑领域带来重要进展。代码已开源。
原文摘要 · Abstract (English)
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}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。