Hybrid-Policy Self-Editing for Composable Unstructured Knowledge Editing

TL;DR

提出混合策略自我编辑(HPSE)提升无结构知识编辑的可组合性,实验证明在四大模型和两类编辑器上均有显著改善。

cs.CL 🔴 高级 2026-08-12 36 次浏览
Tianci Liu Zihan Dong Tianchun Li Yi-Chung Chen Qiming Cao Xingchen Wang Shiyang Wang Zichen Miao Linjun Zhang Haoyu Wang Jing Gao
知识编辑 大规模语言模型 自我蒸馏 可组合性 深度学习

核心发现

方法论

本文提出HPSE,通过构建混合滚动策略,将模型自身的privileged状态引入自我蒸馏过程中,解决纯on-policy蒸馏中知识覆盖不足的问题。HPSE结合了偏好模型的逐步引导与模型自身的轨迹,利用阈值机制触发step-in操作,增强对新知识的覆盖。该方法无需外部监督,具有良好的插件性,适用于多种梯度更新方法。理论分析表明,HPSE在覆盖新知识方面优于纯on-policy方法,特别适合多事实、多段落的无结构知识编辑。

关键结果

  • 在Qwen2.5-7B、Qwen3-8B、Llama-3.1-8B、Gemma-2-9B模型上,HPSE在单次和连续编辑场景中平均提升知识解码准确率6.8和5.0点,最高提升幅度达67.9%。在两大知识编辑基准UnKEBench和MQuAKE-uns中,HPSE显著优于MEMIT、AlphaEdit、AnyEdit等传统方法,尤其在知识的解构和多跳推理能力上表现出色,验证了其在实际应用中的有效性。
  • 结果显示,HPSE通过hybrid rollout显著改善模型对新知识的覆盖和组合能力,减少了纯on-policy蒸馏中的coverage failure,增强了知识的可用性和泛化能力。实验证明其在不同模型和编辑器上的适用性和优越性,展现出强大的插件潜力。
  • 此外,ABlation分析确认hybrid策略的关键作用,模型在多轮连续编辑中表现出更稳定的知识保持和推理能力,验证了理论分析的正确性。整体而言,HPSE为大规模语言模型的知识更新提供了新思路,推动了知识编辑技术的实用化和智能化发展。

研究意义

该研究突破了无结构知识编辑中知识的解构与组合瓶颈,解决了模型对新知识的覆盖不足问题。通过引入主动自我蒸馏策略,显著提升了模型在多事实、多跳推理中的表现,为动态知识更新提供了理论基础和工程方案。此方法不仅增强了模型的知识可用性,也推动了大模型在实际应用中的持续学习能力,具有重要的学术价值和产业前景。未来,HPSE有望在知识图谱、问答系统等领域实现更广泛的应用,推动智能系统的知识动态管理。

技术贡献

本文提出的HPSE方法引入混合滚动策略,结合偏好模型与模型自身轨迹,创新性地解决了纯on-policy蒸馏中的coverage问题。理论分析证明其在新知识覆盖方面具有线性增长优势,且无需外部监督,极大增强了方法的插件性。该技术突破了现有知识编辑的局限,为多事实、多段落的无结构知识编辑提供了有效解决方案。通过在多模型、多场景中的实证验证,展示了其优越的适应性和扩展性,为未来知识动态更新提供了新思路。

新颖性

本研究首次系统性提出混合策略自我蒸馏(HPSE),通过引入hybrid rollout机制,有效缓解了无结构知识编辑中的coverage不足问题。与传统纯on-policy方法相比,HPSE在知识解构与多跳推理方面表现出明显优势,特别是在处理长文本、多事实场景中,展现出创新的理论和工程价值。这一方法在模型自主知识利用和插件式应用方面具有开创性意义,填补了当前研究的空白。

局限性

  • HPSE在极端多事实场景中仍可能面临step-in触发频繁的问题,增加训练成本,且对阈值参数敏感。模型在连续多轮编辑后可能出现知识遗忘或冲突,需进一步优化策略以增强长期稳定性。
  • 该方法依赖于偏好模型的可靠性,若偏好模型对新知识的判断不准确,可能引入偏差。此外,当前实验主要集中在特定模型和数据集,泛化到更复杂的知识体系仍需验证。
  • 训练过程中,hybrid rollout的构建增加了计算负担,尤其在大模型上,可能影响实际部署效率。未来需探索更高效的采样和优化机制。

未来方向

未来将结合强化学习和多任务训练,进一步提升HPSE在更大规模、多模态知识体系中的表现。探索自适应阈值调节和多轮连续编辑的鲁棒性,增强模型的长期记忆和推理能力。此外,推动HPSE在知识图谱、问答系统、智能助手等实际场景中的应用,促进知识的动态管理与持续学习。

AI 总览摘要

在快速变化的知识环境中,保持大规模语言模型(LLMs)知识的实时更新成为核心挑战。传统的静态训练方式难以应对新知识的快速涌现,知识编辑(KE)技术因此应运而生。现有方法多依赖被动注入长文本 passage,忽视了模型对新知识的解构与多跳推理能力,导致知识的可用性不足。本文提出了混合策略自我编辑(HPSE),通过构建混合滚动机制,结合偏好模型与模型自身轨迹,有效提升无结构知识的解构与组合能力。HPSE无需外部监督,具有良好的插件性,理论分析显示其在新知识覆盖方面优于纯on-policy蒸馏。实验证明,在Qwen、Llama、Gemma等模型上,HPSE在多场景、多轮编辑中均实现了显著性能提升,平均提升达6.8点,最大提升67.9%。该方法突破了现有知识编辑的瓶颈,为模型持续学习和动态知识管理提供了新思路。未来,HPSE有望在知识图谱、问答系统等领域实现广泛应用,推动智能系统的知识更新技术迈向新高度。

深度分析

研究背景

近年来,随着大规模语言模型(如GPT、BERT、LLaMA)的快速发展,模型在自然语言理解和生成任务中展现出强大能力。传统训练方式依赖静态大规模语料,难以应对知识的快速变化。知识编辑(KE)技术应运而生,旨在在不影响模型整体能力的前提下,局部更新特定知识。早期方法多采用结构化知识(如三元组)编辑,效果有限。近年来,研究逐渐转向无结构知识(UKE),利用长文本passage进行知识注入,但面临知识解构和多跳推理的挑战。现有方法多依赖被动学习,难以实现知识的灵活利用。

核心问题

无结构知识编辑中,模型注入新知识后,往往只能记忆 passage,而无法进行解构和多跳推理。这限制了知识的可用性,影响模型的推理能力和泛化能力。纯on-policy蒸馏在覆盖新知识方面效果有限,尤其在长文本、多事实场景中表现出coverage不足的问题。如何让模型主动利用新知识,增强其解构和组合能力,成为当前的核心难题。解决这一问题对于实现动态知识更新、提升模型推理能力具有重要意义。

核心创新

本研究提出HPSE,核心创新在于引入hybrid rollout机制,结合偏好模型与模型自身轨迹,主动引导模型覆盖新知识。具体包括:

  • �� 构建hybrid策略,动态触发step-in操作,修正模型偏离passage的轨迹;
  • �� 设计无监督的自我蒸馏目标,增强模型对新知识的理解和利用能力;
  • �� 理论分析证明该机制在新知识覆盖上具有线性增长优势,显著优于纯on-policy方法。这一创新突破了传统被动学习的限制,为无结构知识编辑提供了新思路。

方法详解

  • �� 以passage c和编辑指令x为输入,初始化学生模型πθ和偏好模型π⋆。
  • �� 构建hybrid rollout:在每个时间步t,根据策略πρ在偏好模型和学生模型之间切换,触发step-in条件时由偏好模型引入新知识。
  • �� 触发条件包括:偏好模型与学生模型输出差异超过阈值τ,且偏好模型对该预测有足够信心。
  • �� 训练目标结合hybrid分布和passage级似然,最小化KL散度,确保新知识在轨迹中得到强化。
  • �� 通过多轮迭代,逐步优化模型参数,实现知识的动态融合与利用。

实验设计

在Qwen2.5-7B、Qwen3-8B、Llama-3.1-8B和Gemma-2-9B模型上,采用UnKEBench和MQuAKE-uns两个基准,评估知识解构和多跳推理能力。对比MEMIT、AlphaEdit等方法,验证HPSE在单次和连续编辑中的性能提升。超参数包括阈值τ、κ,训练轮数和采样策略。通过ablation分析,确认hybrid机制的关键作用。结果显示,HPSE在平均指标上提升6.8点,最大达67.9%,显著优于传统方法。

结果分析

实验证明HPSE在多模型、多场景中均实现了优越性能,尤其在知识的解构和多跳推理方面表现突出。具体而言,在UnKEBench中,HPSE提升了passage级准确率和单个事实的解码准确率,平均提升幅度达6.8点,最大67.9%。在连续多轮编辑中,模型表现出更好的知识保持和推理能力,验证了理论分析的有效性。整体结果表明,HPSE有效缓解了coverage不足问题,增强了模型对新知识的主动利用能力。

应用场景

该方法适用于需要动态知识更新的应用场景,如智能问答、知识图谱维护、自动文摘等。通过插件式设计,能在现有大模型基础上快速集成,提升模型的知识时效性和推理能力。未来,结合强化学习和多模态信息,有望实现更智能的知识管理系统,推动行业智能化升级。

局限与展望

当前HPSE在极端多事实场景中仍存在step-in频繁的问题,增加训练成本。对偏好模型的依赖可能引入偏差,模型在连续多轮编辑后可能出现知识遗忘。未来需优化阈值调节机制,提升长时序稳定性,并探索多模态知识融合的方法。

通俗解读 非专业人士也能看懂

想象你在厨房做饭,平时只用食谱一步步跟着做,结果新食谱出现了很多新菜,但你只知道原来的做法,不能灵活搭配。现在,你的厨师朋友告诉你一些新技巧,但你自己不主动去试,结果只能记住那几道菜,不能自己创新或组合新菜。本文就像给你装了个智能厨师助手,它不仅记住新技巧,还会主动帮你搭配菜肴,确保你能用新食谱做出多样的菜。这种方法让厨师变得更聪明,不仅记住新菜,还能灵活组合,做出更多新菜肴。

简单解释 像给14岁少年讲一样

想象你在玩一款游戏,游戏里你可以学到很多新技能,但有时候你只记住了技能的名字,不能用它们组合出新玩法。这个研究就像给你开发了一个超级助手,它不仅记住新技能,还会主动帮你把技能拼在一起,创造出新玩法。这样,你就能在游戏中变得更厉害,不只是学会技能,还能用它们组合出新花样。这个方法让你的游戏角色变得更聪明、更有创造力,不仅能记住新东西,还能自己动脑筋用起来。

术语表

Knowledge Editing (知识编辑)

在预训练模型基础上,局部更新模型知识的技术,旨在保持模型整体能力同时修正特定信息。

论文中用于描述模型局部知识更新的方法。

Unstructured Knowledge (无结构知识)

以自由文本形式表达的知识,没有明确的三元组结构,难以直接操作。

指论文中所研究的知识注入场景。

Self-Distillation (自我蒸馏)

模型利用自身的输出作为学习目标,提升自身的性能和知识利用能力。

HPSE中的核心机制之一。

Hybrid Rollout (混合滚动)

结合偏好模型和学生模型输出的生成策略,用于强化新知识的覆盖。

HPSE的关键技术。

Coverage (覆盖)

模型对新知识的理解和利用程度,衡量知识解构和多跳推理能力。

评估知识编辑效果的重要指标。

开放问题 这项研究留下的未解疑问

  • 1 如何在极端多事实场景中保持模型稳定性和知识一致性仍未解决,未来需探索更鲁棒的机制。
  • 2 当前HPSE主要验证在文本场景,跨模态知识编辑的适应性和效果仍待研究。

应用场景

近期应用

动态知识问答系统

利用HPSE实现问答系统中知识的实时更新和多跳推理,提升回答准确性和时效性。

知识图谱维护

在知识图谱中快速插入和组合新事实,保持图谱的动态更新和推理能力。

远期愿景

智能助手的持续学习

让智能助手不断吸收新知识,主动利用新信息进行推理和决策,推动AI的自主学习能力。

原文摘要

Large language models (LLMs) achieve remarkable performance across natural language tasks, yet they are trained on static corpora and their knowledge quickly becomes outdated in a fast-changing world. This motivates knowledge editing (KE), which updates specific knowledge in an LLM without changing unrelated others. Recent works move from structured knowledge triples toward unstructured KE (UKE), where the edit is a free-form passage that may state multiple facts at once. Nonetheless, existing editors inject such a passage yet fail to use it: the edited model can recall the passage, but can neither answer atomic questions about its facts nor compose them into multi-hop reasoning. We attribute this missing property, which we term composability, to editors' passive reliance on the fixed passage as the sole learning source. In response, we cast editing as a proactive self-distillation from a privileged in-context state of the same model, which requires no external supervision. We further reveal that due to the novelty of the injected knowledge, the pre-edited model's own rollouts rarely cover it, which limits the effectiveness of pure on-policy distillation. To close this gap, we propose HPSE, which builds a hybrid rollout that steps in to place missing facts onto the student's own trajectory precisely where its coverage fails, while staying on-policy elsewhere. We theoretically analyze HPSE's advantage over pure on-policy distillation, and empirically establish its plug-and-play improvements across four LLM backbones and two KE editors under various scenarios.

cs.CL cs.AI cs.LG