arXiv:2409.17819cs.CLcs.AI2024-09EMNLP被引 25

无需微调,推理时用价值函数引导模型生成更符合人类偏好内容。

Inference-Time Language Model Alignment via Integrated Value Guidance

  • 在词元和片段级别分别使用隐式与显式价值函数引导解码。
  • 在AlpacaEval 2.0上,模型胜率从19.51%提升至26.51%(Mistral-7B)。
  • 适用于指令跟随、摘要等任务,支持即插即用的价值函数。

大型语言模型通常通过微调来对齐人类偏好,但微调计算成本高且复杂。本文提出集成价值引导(IVG),通过隐式和显式价值函数分别在词元和片段级别引导模型解码,仅在推理阶段实现高效对齐,避免了直接微调的复杂性,并优于传统方法。实验表明,IVG在多种任务中具有通用性:在控制情感生成和摘要任务中,利用基于GPT2的价值函数显著提升了大模型的对齐效果;在更具挑战性的指令遵循基准AlpacaEval 2.0上,无论是专门训练还是现成的价值函数,均大幅提高大模型在长度控制下的胜率,例如,Mistral-7B-Instruct-v0.2从19.51%提升至26.51%,Mixtral-8x7B-Instruct-v0.1从25.58%提升至33.75%(采用Tulu引导)。

原文摘要 · Abstract (English)

Large language models are typically fine-tuned to align with human preferences, but tuning large models is computationally intensive and complex. In this work, we introduce $\textit{Integrated Value Guidance}$ (IVG), a method that uses implicit and explicit value functions to guide language model decoding at token and chunk-level respectively, efficiently aligning large language models purely at inference time. This approach circumvents the complexities of direct fine-tuning and outperforms traditional methods. Empirically, we demonstrate the versatility of IVG across various tasks. In controlled sentiment generation and summarization tasks, our method significantly improves the alignment of large models using inference-time guidance from $\texttt{gpt2}$-based value functions. Moreover, in a more challenging instruction-following benchmark AlpacaEval 2.0, we show that both specifically tuned and off-the-shelf value functions greatly improve the length-controlled win rates of large models against $\texttt{gpt-4-turbo}$ (e.g., $19.51\% \rightarrow 26.51\%$ for $\texttt{Mistral-7B-Instruct-v0.2}$ and $25.58\% \rightarrow 33.75\%$ for $\texttt{Mixtral-8x7B-Instruct-v0.1}$ with Tulu guidance).

语言模型推理对齐价值引导生成控制

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