arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.11523cs.LG

统一稳定优化与参考正则化在RLHF中

Unifying Stable Optimization and Reference Regularization in RLHF

Sydney AI Centre, The University of Sydney(悉尼人工智能中心、悉尼大学);CSIRO, Data61(澳大利亚联邦科学与工业研究组织、Data61);University of New South Wales(新南威尔士大学)

Li He, Qiang Qu, He Zhao, Stephen Wan, Dadong Wang, Lina Yao, Tongliang Liu

AI总结:

本文提出了一种统一正则化方法,通过平衡防止奖励黑客和稳定策略更新,提升RLHF的对齐性能和稳定性。

摘要:

强化学习从人类反馈(RLHF)在对齐能力上取得了显著进展,但仍然受到两个核心挑战的阻碍:奖励黑客和稳定优化。当前的解决方案通过单独的正则化策略分别解决这些问题,具体而言,通过KL散度惩罚对抗监督微调模型(π₀)以减轻奖励黑客,以及通过策略比率裁剪朝向当前策略(πₜ)以促进稳定对齐。然而,同时正则化向π₀和πₜ所隐含的权衡仍然未被充分探索。在本文中,我们引入了一种统一的正则化方法,明确平衡防止奖励黑客和维持稳定策略更新的目标。我们的简单而原则性的对齐目标产生了一种加权监督微调损失,具有更优的权衡,这明显提高了对齐结果和实现复杂性。在多样化的基准测试中进行的广泛实验验证了我们的方法在RLHF和在线偏好学习方法上的一致性优势,实现了增强的对齐性能和稳定性。

英文摘要:

Reinforcement Learning from Human Feedback (RLHF) has advanced alignment capabilities significantly but remains hindered by two core challenges: \textbf{reward hacking} and \textbf{stable optimization}. Current solutions independently address these issues through separate regularization strategies, specifically a KL-divergence penalty against a supervised fine-tuned model ($π_0$) to mitigate reward hacking, and policy ratio clipping towards the current policy ($π_t$) to promote stable alignment. However, the implicit trade-off arising from simultaneously regularizing towards both $π_0$ and $π_t$ remains under-explored. In this paper, we introduce a unified regularization approach that explicitly balances the objectives of preventing reward hacking and maintaining stable policy updates. Our simple yet principled alignment objective yields a weighted supervised fine-tuning loss with a superior trade-off, which demonstrably improves both alignment results and implementation complexity. Extensive experiments across diverse benchmarks validate that our method consistently outperforms RLHF and online preference learning methods, achieving enhanced alignment performance and stability.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑