arXiv:2411.01798cs.LG2024-11被引 4

用模型合集提升对齐效果,让大模型更懂人类偏好。

SALSA: Soup-based Alignment Learning for Stronger Adaptation in RLHF

  • 通过平均两个不同微调模型的权重生成新参考模型
  • 在多个基准上比PPO方法获得更高奖励和更强泛化能力
  • 适合需要更好对齐与探索能力的RLHF研究者

在大语言模型开发中,基于人类反馈的强化学习(RLHF)对对齐人类价值观至关重要。传统方法使用当前策略与冻结初始策略之间的KL散度作为约束,加入到PPO等优化算法中,虽能防止过度偏离初始检查点,但限制了对奖励空间的探索,使模型陷入参数空间的狭窄区域,导致对齐与性能不佳。本文提出SALSA(Soup-based Alignment Learning for Stronger Adaptation),通过平均两个独立监督微调(SFT)模型的权重生成模型汤(model soup),构建更灵活且位置更优的参考模型。该方法允许更大的KL偏差,促进对潜在优质解空间的探索,同时保持稳定性。实验在Llama2-7B、Mistral-7B和Gemma-2B等主流开源模型上进行,覆盖MT-Bench、Arena-Hard、UltraFeedback等多个基准,结果表明SALSA持续优于PPO,实现更深探索、更优对齐与更强鲁棒性及分布外泛化能力。

原文摘要 · Abstract (English)

In Large Language Model (LLM) development, Reinforcement Learning from Human Feedback (RLHF) is crucial for aligning models with human values and preferences. RLHF traditionally relies on the Kullback-Leibler (KL) divergence between the current policy and a frozen initial policy as a reference, which is added as a penalty in policy optimization algorithms like Proximal Policy Optimization (PPO). While this constraint prevents models from deviating too far from the initial checkpoint, it limits exploration of the reward landscape, reducing the model's ability to discover higher-quality solutions. As a result, policy optimization is often trapped in a narrow region of the parameter space, leading to suboptimal alignment and performance. This paper presents SALSA (Soup-based Alignment Learning for Stronger Adaptation), a novel approach designed to overcome these limitations by creating a more flexible and better located reference model through weight-space averaging of two independent supervised fine-tuned (SFT) models. This model soup allows for larger deviation in KL divergence and exploring a promising region of the solution space without sacrificing stability. By leveraging this more robust reference model, SALSA fosters better exploration, achieving higher rewards and improving model robustness, out-of-distribution generalization, and performance. We validate the effectiveness of SALSA through extensive experiments on popular open models (Llama2-7B, Mistral-7B, and Gemma-2B) across various benchmarks (MT-Bench, Arena-Hard, UltraFeedback), where it consistently surpasses PPO by fostering deeper exploration and achieving superior alignment in LLMs.

RLHF模型对齐强化学习大模型

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