发现大模型编辑中过度拟合问题,提出新方法提升知识泛化能力。
Uncovering Overfitting in Large Language Model Editing
- 通过多阶段推理约束机制引导模型类比上下文学习回忆新知识。
- 在多个任务上验证新方法有效缓解编辑过度拟合现象。
- 适用于需复杂推理的模型知识更新场景,如多跳问答。
知识编辑被广泛用于更新和修正大语言模型(LLMs)内部知识,但现有方法在复杂任务(如多跳推理)中表现不佳。本文揭示了‘编辑过度拟合’现象:编辑后的模型对目标知识赋予过高的概率,阻碍新知识在复杂场景下的泛化。我们归因于当前编辑范式过度强调输入提示与编辑目标之间的直接对应关系。为此,我们构建新基准EVOKE(EValuation of Editing Overfit in Knowledge Editing)及细粒度评估指标。实验表明,编辑过度拟合在现有方法中普遍存在,且常规正则化策略无效。受大模型知识回忆机制启发,我们提出即插即用的新策略Learn the Inference(LTI),引入多阶段推理约束模块,使编辑模型能像未编辑模型那样通过上下文学习方式回忆新知识。跨多种任务的实验证明,LTI可有效缓解编辑过度拟合问题。
原文摘要 · Abstract (English)
Knowledge editing has been proposed as an effective method for updating and correcting the internal knowledge of Large Language Models (LLMs). However, existing editing methods often struggle with complex tasks, such as multi-hop reasoning. In this paper, we identify and investigate the phenomenon of Editing Overfit, where edited models assign disproportionately high probabilities to the edit target, hindering the generalization of new knowledge in complex scenarios. We attribute this issue to the current editing paradigm, which places excessive emphasis on the direct correspondence between the input prompt and the edit target for each edit sample. To further explore this issue, we introduce a new benchmark, EVOKE (EValuation of Editing Overfit in Knowledge Editing), along with fine-grained evaluation metrics. Through comprehensive experiments and analysis, we demonstrate that Editing Overfit is prevalent in current editing methods and that common overfitting mitigation strategies are ineffective in knowledge editing. To overcome this, inspired by LLMs' knowledge recall mechanisms, we propose a new plug-and-play strategy called Learn the Inference (LTI), which introduce a Multi-stage Inference Constraint module to guide the edited models in recalling new knowledge similarly to how unedited LLMs leverage knowledge through in-context learning. Extensive experimental results across a wide range of tasks validate the effectiveness of LTI in mitigating Editing Overfit.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。