arXiv:2503.21819cs.CL2025-03被引 28

用多目标强化学习让大模型更安全、更对齐,训练更快更稳。

Optimizing Safe and Aligned Language Generation: A Multi-Objective GRPO Approach

  • 通过分组对比生成结果优化策略,无需价值函数
  • 在0.5B~14B模型上同时提升安全与质量指标
  • 适合追求高效对齐的AI研发人员使用

大语言模型对齐人类价值观和安全约束面临挑战,尤其当帮助性、真实性与避免伤害等目标冲突时。尽管基于人类反馈的强化学习(RLHF)表现良好,但过程复杂且不稳定。近期的直接偏好优化(DPO)虽简化了流程,却可能引入偏差或牺牲部分目标。本文提出一种分组相对策略优化(GRPO)框架,结合多标签奖励回归模型,实现安全且对齐的语言生成。GRPO通过比较采样响应组来优化策略,无需独立价值评判器,提升了训练效率。我们训练奖励模型以预测多个对齐得分(如安全性、帮助性等),并融合为单一奖励信号。文中提供了该学习到的多维度奖励在GRPO中使用的理论推导,并讨论其优势与局限。实验表明,该方法在0.5B、7B和14B参数规模的模型上均显著提升各项安全与质量指标,实现了多目标间的稳健平衡。与基于PPO的RLHF和DPO相比,GRPO在达成对齐的同时计算成本更低,且显式处理多目标。所有训练模型将开源至https://huggingface.co/hydroxai。

原文摘要 · Abstract (English)

Aligning large language models (LLMs) with human values and safety constraints is challenging, especially when objectives like helpfulness, truthfulness, and avoidance of harm conflict. Reinforcement Learning from Human Feedback (RLHF) has achieved notable success in steering models, but is complex and can be unstable. Recent approaches such as Direct Preference Optimization (DPO) simplify preference-based fine-tuning but may introduce bias or trade-off certain objectives~\cite{dpo}. In this work, we propose a Group Relative Policy Optimization (GRPO) framework with a multi-label reward regression model to achieve safe and aligned language generation. The GRPO algorithm optimizes a policy by comparing groups of sampled responses, eliminating the need for a separate value critic and improving training efficiency~\cite{grpo}. We train a reward model to predict multiple alignment scores (e.g., safety, helpfulness, etc.), which are combined into a single reward signal. We provide a theoretical derivation for using this learned multi-aspect reward within GRPO and discuss its advantages and limitations. Empirically, our approach improves all the safety and quality metrics evaluated in language generation tasks on model scales (0.5B, 7B, and 14B parameters), demonstrating a robust balance of objectives. We compare GRPO to PPO-based RLHF and DPO, highlighting that GRPO achieves alignment with significantly lower computational cost and explicit multi-objective handling. \textbf{We will open-source all trained models at https://huggingface.co/hydroxai.

大模型对齐强化学习多目标优化安全生成

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