arXiv:2505.18343cs.CL2025-05ACL被引 4

用双曲几何与图网络实现更稳定准确的模型知识编辑

Model Editing with Graph-Based External Memory

  • 用双曲空间建图表示知识三元组,保持层级结构
  • 通过双曲变换更新参数,避免关系距离扭曲
  • 结合梯度掩码与周期重置,防止遗忘关键知识

大型语言模型虽推动自然语言处理发展,但常因幻觉和过时参数知识限制实用性。现有模型编辑方法易出现过拟合与灾难性遗忘。为此,我们提出基于双曲几何与图神经网络的新型编辑框架HYPE,包含三部分:(i) 双曲图构建,利用庞加莱嵌入在双曲空间中表示知识三元组,保留层级关系,确保父概念修改不影响子概念;(ii) 莫比乌斯变换更新,通过双曲加法传播编辑,维持流形内结构一致性,优于传统欧氏更新导致的关系失真;(iii) 双重稳定机制,结合梯度掩码与周期性GNN参数重置,聚焦关键参数更新并保护长期知识。在CounterFact、CounterFact+和MQuAKE数据集上,使用GPT-J和GPT2-XL的实验表明,HYPE显著提升编辑稳定性、事实准确性及多跳推理能力。

原文摘要 · Abstract (English)

Large language models (LLMs) have revolutionized natural language processing, yet their practical utility is often limited by persistent issues of hallucinations and outdated parametric knowledge. Although post-training model editing offers a pathway for dynamic updates, existing methods frequently suffer from overfitting and catastrophic forgetting. To tackle these challenges, we propose a novel framework that leverages hyperbolic geometry and graph neural networks for precise and stable model edits. We introduce HYPE (HYperbolic Parameter Editing), which comprises three key components: (i) Hyperbolic Graph Construction, which uses Poincaré embeddings to represent knowledge triples in hyperbolic space, preserving hierarchical relationships and preventing unintended side effects by ensuring that edits to parent concepts do not inadvertently affect child concepts; (ii) Möbius-Transformed Updates, which apply hyperbolic addition to propagate edits while maintaining structural consistency within the hyperbolic manifold, unlike conventional Euclidean updates that distort relational distances; and (iii) Dual Stabilization, which combines gradient masking and periodic GNN parameter resetting to prevent catastrophic forgetting by focusing updates on critical parameters and preserving long-term knowledge. Experiments on CounterFact, CounterFact+, and MQuAKE with GPT-J and GPT2-XL demonstrate that HYPE significantly enhances edit stability, factual accuracy, and multi-hop reasoning.

模型编辑双曲几何知识更新

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。