用自回归奖励模型实现零训练成本的文本生成对齐,支持实时偏好调整。
GenARM: Reward Guided Generation with Autoregressive Reward Model for Test-time Alignment
- 设计可预测下一个词奖励的自回归奖励模型,适配逐词生成过程。
- 在不重训练情况下,性能接近甚至超过传统训练对齐方法。
- 支持多目标偏好调节与大小模型间的高效对齐,适合个性化场景。
大型语言模型虽具备强大能力,但需与人类偏好对齐。传统训练时方法通过人类偏好数据微调模型,成本高且需重复训练以适应不同偏好。测试时对齐方法使用奖励模型(RM)指导冻结的LLM,避免重训练。然而现有方法依赖完整响应级奖励模型,无法用于需要从部分响应计算下一词奖励的自回归生成。为此,我们提出GenARM,一种利用自回归奖励模型(Autoregressive Reward Model)的测试时对齐方法,该模型能高效预测下一词奖励。理论上,该参数化可在KL正则化的强化学习框架下,证明性地引导冻结的LLM达到传统奖励模型所能实现的任意分布。实验表明,GenARM显著优于现有测试时对齐基线,并达到训练时方法的性能。此外,它支持弱到强的高效引导,使大模型可借助小奖励模型对齐,无需训练大模型;同时支持多目标对齐,实现实时偏好权衡,满足多样化用户需求。项目页面:https://genarm.github.io。
原文摘要 · Abstract (English)
Large Language Models (LLMs) exhibit impressive capabilities but require careful alignment with human preferences. Traditional training-time methods finetune LLMs using human preference datasets but incur significant training costs and require repeated training to handle diverse user preferences. Test-time alignment methods address this by using reward models (RMs) to guide frozen LLMs without retraining. However, existing test-time approaches rely on trajectory-level RMs which are designed to evaluate complete responses, making them unsuitable for autoregressive text generation that requires computing next-token rewards from partial responses. To address this, we introduce GenARM, a test-time alignment approach that leverages the Autoregressive Reward Model--a novel reward parametrization designed to predict next-token rewards for efficient and effective autoregressive generation. Theoretically, we demonstrate that this parametrization can provably guide frozen LLMs toward any distribution achievable by traditional RMs within the KL-regularized reinforcement learning framework. Experimental results show that GenARM significantly outperforms prior test-time alignment baselines and matches the performance of training-time methods. Additionally, GenARM enables efficient weak-to-strong guidance, aligning larger LLMs with smaller RMs without the high costs of training larger models. Furthermore, GenARM supports multi-objective alignment, allowing real-time trade-offs between preference dimensions and catering to diverse user preferences without retraining. Our project page is available at: https://genarm.github.io.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。