无需修改模型参数,用强化学习让大模型更符合人类偏好
Aligning Frozen LLMs by Reinforcement Learning: An Iterative Reweight-then-Optimize Approach
- 通过迭代重采样与轻量价值函数优化,实现冻结模型的偏好对齐
- 测试时用搜索优化生成结果,避免高推理开销,输出质量优于单次引导
- 适合无权重访问场景,用户可自定义数据集做对齐训练
将大语言模型(LLM)对齐人类偏好通常需要微调方法如RLHF和DPO。这些方法直接优化模型参数,无法用于测试阶段提升性能,也不适用于权重不可访问的情况。相比之下,测试时方法通过奖励函数引导生成,但推理成本高,且单次引导依赖不完善的奖励或价值函数,导致输出次优。本文提出一种名为迭代重采样-优化(IRO)的强化学习框架,可在不修改基础模型参数的情况下实现偏好对齐。训练阶段每轮:(i) 从基础模型采样候选输出,(ii) 使用当前价值函数重采样,(iii) 训练新的轻量级价值函数以指导下一轮解码。测试时,利用价值函数通过基于搜索的优化过程引导基础模型生成。用户可使用IRO在自有数据集上对模型进行对齐,类似OpenAI的强化微调(RFT),但无需模型权重访问权限。
原文摘要 · Abstract (English)
Aligning large language models (LLMs) with human preferences usually requires fine-tuning methods such as RLHF and DPO. These methods directly optimize the model parameters, so they cannot be used in test-time to improve model performance, nor are they applicable when the model weights are not accessible. In contrast, test-time methods sidestep weight updates by leveraging reward functions to guide and improve output quality. However, they incur high inference costs, and their one-shot guidance is often based on imperfect reward or value functions, leading to suboptimal outputs. In this work, we present a method named Iterative Reweight-then-Optimize (IRO), a reinforcement learning (RL) framework that performs RL-style alignment of the (frozen) base model without touching its parameters. During training, each iteration (i) samples candidates from the base model, (ii) resamples using current value functions, and (iii) trains a new lightweight value function that guides the next decoding pass. At test time, the value functions are used to guide the base model generation via a search-based optimization process. Notably, users can apply IRO to align a model on their own dataset, similar to OpenAI's reinforcement fine-tuning (RFT), but without requiring access to the model weights.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。