arXiv:2605.10640cs.CLcs.AI2026-05中稿 · ICML

揭示大模型持续学新知识的机制,提出高效防遗忘方法

Towards Understanding Continual Factual Knowledge Acquisition of Language Models: From Theory to Algorithm

论文配图:Towards Understanding Continual Factual Knowledge Acquisition of Language Models: From Theory to Algorithm
图 1 · 摘自论文原文
  • 用单层Transformer建模持续学知识过程,解析不同方法原理
  • 发现正则化无法改变遗忘趋势,数据重放可稳定旧知识
  • 提出STOC方法,智能选关键事实片段提升学习效果

持续预训练(CPT)对语言模型在不遗忘旧知识的前提下融入新知识至关重要。尽管数据重放等经典方法已成为主流,但语言模型如何随时间持续获取并保留事实知识(即持续事实知识获取,cFKA)的内在机制仍不清晰。本文基于单层Transformer构建理论框架,统一解释了代表性CPT方法的行为。分析表明,正则化方法仅调节参数收敛速度,未改变固有的遗忘倾向;而数据重放能改变收敛动态并稳定预训练知识。基于此,我们提出一种生成式数据重放方法——通过注意力贡献选择关键事实片段(STOC),指导重放数据生成。在合成与真实数据集上的大量实验验证了理论发现,并表明STOC能有效缓解灾难性遗忘,显著提升cFKA性能。

原文摘要 · Abstract (English)

Continual Pre-Training (CPT) is essential for enabling Language Models (LMs) to integrate new knowledge without erasing old. While classical CPT techniques like data replay have become the standard paradigm, the mechanisms underlying how LMs acquire and retain facts over time, termed as continual Factual Knowledge Acquisition (cFKA), remain unclear. In this work, we present a theoretical framework that characterizes the training dynamics of cFKA using a single-layer Transformer, offering a unified explanation for the behavior of representative CPT methods. Our analysis reveals that regularization-based methods merely adjust the convergence rate of parameters without altering the inherent forgetting tendency, whereas data replay methods succeed in shifting convergence dynamics and stabilizing pretrained knowledge. Building on these insights, we propose a novel generative data replay approach, called \textbf{S}electing \textbf{T}okens via attenti\textbf{O}n \textbf{C}ontribution~(STOC), which identifies influential factual snippets to guide replay data generation. Extensive experiments on both synthetic and real-world datasets validate our findings and demonstrate that STOC effectively enhances cFKA by mitigating catastrophic forgetting.

持续学习知识获取语言模型遗忘抑制

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