arXiv:2510.09885cs.CLcs.AI2025-10被引 2

用去遮蔽目标让自回归大模型也能高效注入知识,无需改写数据。

Diffusion-Inspired Masked Fine-Tuning for Knowledge Injection in Autoregressive LLMs

  • 用去遮蔽任务替代传统改写增强,让模型从掩码文本中恢复原内容。
  • 在120万样本数据上,该方法使知识问答准确率超越所有其他微调方式。
  • 不仅提升事实知识注入,还能改善数学任务表现,适用性广。

大型语言模型常用于事实不断更新的场景,但通过非结构化文本微调来更新知识时,往往依赖计算量大的改写增强,并面临反向诅咒问题。近期研究表明,扩散型语言模型(dLLMs)在预训练中所需样本更少、损失更低,且对反向诅咒更具抵抗力,暗示其可能比自回归模型(arLLMs)更容易学习新知识。我们在受控的知识微调实验中验证了这一假设:arLLMs依赖改写增强才能将知识文本泛化为问答能力,而dLLMs无需改写即可达到高问答准确率。为探究去遮蔽目标是否能独立带来这种优势,我们提出针对arLLMs的掩码微调方法,即在上下文中让模型根据掩码版本重建原始文本。该方法显著提升了知识注入效果,无需改写且抗反向诅咒,缩小了arLLMs与dLLMs的差距。在大规模知识密集型数据集(120万样本)上,掩码SFT在GPQA-diamond上的下游准确率优于所有微调变体。去遮蔽目标还提升了数学任务的SFT表现,表明其具有广泛适用性。

原文摘要 · Abstract (English)

Large language models (LLMs) are often used in environments where facts evolve, yet factual knowledge updates via fine-tuning on unstructured text often suffer from 1) reliance on compute-heavy paraphrasing augmentation and 2) the reversal curse. Recent studies show diffusion large language models (dLLMs) require fewer training samples to achieve lower loss in pre-training and are more resistant to the reversal curse, suggesting dLLMs may learn new knowledge more easily than autoregressive LLMs (arLLMs). We test this hypothesis in controlled knowledge fine-tuning experiments and find that while arLLMs rely on paraphrase augmentation to generalize knowledge text into question-answering (QA) capability, dLLMs do not require paraphrases to achieve high QA accuracy. To further investigate whether the demasking objective alone can induce such a knowledge injection advantage in dLLMs regardless of their diffusion denoising paradigm, we propose masked fine-tuning for arLLMs, which prompts an arLLM to reconstruct the original text given a masked version in context. The masked fine-tuning for arLLMs substantially improves the efficacy of knowledge injection, i.e. no paraphrase needed and resistant to the reversal curse, closing the gap between arLLMs and dLLMs. We also demonstrate broader applicability: on a large-scale knowledge-intensive dataset (1.2M samples), masked SFT achieves the best downstream accuracy on GPQA-diamond among all fine-tuning variants. The demasking objective also improves SFT on math tasks, suggesting broad utility beyond factual knowledge injection.

知识注入自回归模型扩散模型微调方法

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