arXiv:2507.05158cs.CL2025-07NeurIPS被引 1

通过引导模型记忆使用,提升后训练阶段的知识利用率。

Steering Information Utility in Key-Value Memory for Language Model Post-Training

  • 将前馈层视为键值记忆,用正则化干预激活存储信息
  • 在15个任务上实现跨模型家族的性能提升
  • 让模型更专注有意义词汇,减少对简单词的资源消耗

近期语言模型的发展凸显了后训练的重要性。然而,监督微调等方法并不能确保充分利用预训练阶段获得的知识。为此,我们提出InfoSteer,一种轻量级方法,旨在后训练阶段促进参数化信息的有效利用。具体而言,InfoSteer将前馈网络(FFN)层视为关联的键值记忆,并通过前向传播干预或反向传播中的正则化来推动存储向量的使用。这一简单指导在多个模型家族(包括Qwen、Gemma和Llama)中均带来一致的性能提升,覆盖15个下游任务,涵盖分布内(ID)与分布外(OOD)评估。此外,经引导的模型能自适应分配信息,更关注生成语义丰富的词元,而减少对简单过渡词(如‘,’或‘and’)的资源投入。本工作表明,原始后训练无法完全释放预训练带来的潜力,而对隐空间的引导是提升性能与可解释性的有效路径。代码已开源:https://github.com/chili-lab/InfoSteer。

原文摘要 · Abstract (English)

Recent advancements in language models (LMs) have marked a shift toward the growing importance of post-training. Yet, post-training approaches such as supervised fine-tuning (SFT) do not guarantee the effective use of knowledge acquired during pretraining. We therefore introduce InfoSteer, a lightweight method that encourages parametric information utilization in LMs during post-training. Specifically, InfoSteer treats the feed-forward network (FFN) layer as associate key-value memory and promotes the use of stored memory vectors via forward-pass interventions or regularization during backpropagation. This simple guidance during post-training phase yields consistent performance improvements across diverse model families -- including Qwen, Gemma and Llama -- spanning 15 downstream tasks in both in-distribution (ID) and out-of-distribution (OOD) evaluations. Beyond performance gains, we also find that steered LMs can adaptively allocate information by placing more emphasis on generating semantically meaningful tokens, while using fewer resources on simple transition ones (e.g., `\texttt{,}' or `\texttt{and}'). Our work underscores that vanilla post-training does not fully exploit the potential gained during pre-training, and that steering LMs in latent representation space offers a promising approach to enhance both performance and interpretability. The code is available at: https://github.com/chili-lab/InfoSteer.

后训练知识利用模型优化可解释性

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