arXiv:2509.03405cs.CL2025-09被引 2

构建可追踪知识学习过程的分析工具,揭示语言模型如何从预训练数据中获取知识。

LMEnt: A Suite for Analyzing Knowledge in Language Models from Pretraining Data to Representations

  • 基于维基百科构建带实体标注的预训练语料库
  • 实体检索方法性能比之前提升最高达80.4%
  • 提供12个大模型及4000个中间检查点,支持知识研究

语言模型在现实应用中日益依赖世界知识,但其将数据转化为知识表征的内部机制仍不清晰。为推动相关研究,我们提出LMEnt:一个用于分析预训练过程中知识获取的工具套件。该套件包含:(1) 一个富含知识的预训练语料库,基于维基百科并全量标注实体提及;(2) 一种基于实体的预训练数据检索方法,性能相比先前方法最高提升80.4%;(3) 12个预训练模型(最大参数量达10亿),以及4000个中间检查点,在知识基准测试上表现与主流开源模型相当。这些资源共同构成可控环境,用于分析预训练中的实体提及与下游性能之间的关联,以及因果干预的影响。我们通过分析不同检查点的知识获取过程发现,事实频率是关键因素,但无法完全解释学习趋势。我们已开源LMEnt,以支持知识表征、可塑性、编辑、归因与学习动态等研究。

原文摘要 · Abstract (English)

Language models (LMs) increasingly drive real-world applications that require world knowledge. However, the internal processes through which models turn data into representations of knowledge and beliefs about the world, are poorly understood. Insights into these processes could pave the way for developing LMs with knowledge representations that are more consistent, robust, and complete. To facilitate studying these questions, we present LMEnt, a suite for analyzing knowledge acquisition in LMs during pretraining. LMEnt introduces: (1) a knowledge-rich pretraining corpus, fully annotated with entity mentions, based on Wikipedia, (2) an entity-based retrieval method over pretraining data that outperforms previous approaches by as much as 80.4%, and (3) 12 pretrained models with up to 1B parameters and 4K intermediate checkpoints, with comparable performance to popular open-sourced models on knowledge benchmarks. Together, these resources provide a controlled environment for analyzing connections between entity mentions in pretraining and downstream performance, and the effects of causal interventions in pretraining data. We show the utility of LMEnt by studying knowledge acquisition across checkpoints, finding that fact frequency is key, but does not fully explain learning trends. We release LMEnt to support studies of knowledge in LMs, including knowledge representations, plasticity, editing, attribution, and learning dynamics.

知识表征预训练分析实体识别模型可解释性

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