arXiv:2411.17401cs.CL2024-11被引 2

发现跨语言知识神经元,精准定位大模型中的通用事实记忆

One Mind, Many Tongues: A Deep Dive into Language-Agnostic Knowledge Neurons in Large Language Models

  • 构建多语言重述数据集,提升知识定位一致性
  • 提出不确定性量化方法,准确识别跨语言知识神经元
  • 验证其在跨语言编辑与知识注入中的关键作用

大规模语言模型通过自监督预训练获得了海量事实知识,并具备多语言表达能力。然而,模型中知识的存储机制仍不清晰。已有研究从知识神经元视角出发,发现可跨语言存储事实知识的神经元,但存在两大局限:一是定位结果不确定性高,因仅依赖提示式探测,而大模型对语义等价查询无法保持一致回答;二是分析语言范围有限,仅涵盖英语和中文,缺乏对更多语言家族的探索,限制了结论泛化性。为此,本文构建新基准Rephrased Multilingual LAMA(RML-LAMA),包含每个事实的高质量多语言同义填空式查询。提出多语言集成梯度带不确定性估计方法(MATRICE),在知识定位中量化查询与语言间的不确定性。实验表明,该方法能更准确地定位跨语言知识神经元。进一步研究显示,这些神经元在跨语言知识编辑、知识增强和新知识注入中发挥核心作用。

原文摘要 · Abstract (English)

Large language models (LLMs) have learned vast amounts of factual knowledge through self-supervised pre-training on large-scale corpora. Meanwhile, LLMs have also demonstrated excellent multilingual capabilities, which can express the learned knowledge in multiple languages. However, the knowledge storage mechanism in LLMs still remains mysterious. Some researchers attempt to demystify the factual knowledge in LLMs from the perspective of knowledge neurons, and subsequently discover language-agnostic knowledge neurons that store factual knowledge in a form that transcends language barriers. However, the preliminary finding suffers from two limitations: 1) High Uncertainty in Localization Results. Existing study only uses a prompt-based probe to localize knowledge neurons for each fact, while LLMs cannot provide consistent answers for semantically equivalent queries. Thus, it leads to inaccurate localization results with high uncertainty. 2) Lack of Analysis in More Languages. The study only analyzes language-agnostic knowledge neurons on English and Chinese data, without exploring more language families and languages. Naturally, it limits the generalizability of the findings. To address aforementioned problems, we first construct a new benchmark called Rephrased Multilingual LAMA (RML-LAMA), which contains high-quality cloze-style multilingual parallel queries for each fact. Then, we propose a novel method named Multilingual Integrated Gradients with Uncertainty Estimation (MATRICE), which quantifies the uncertainty across queries and languages during knowledge localization. Extensive experiments show that our method can accurately localize language-agnostic knowledge neurons. We also further investigate the role of language-agnostic knowledge neurons in cross-lingual knowledge editing, knowledge enhancement and new knowledge injection.

知识神经元多语言大模型

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