自动提取用户偏好中的规则,提升强化学习的反馈效果。
AutoRule: Reasoning Chain-of-thought Extracted Rule-based Rewards Improve Preference Learning

- 用推理模型解析偏好,自动提取可执行规则
- 在AlpacaEval2.0上提升28.6%胜率,MT-Bench提升6.1%
- 减少奖励黑客,适配不同数据集特性
基于规则的奖励为改进人类反馈强化学习(RLHF)提供了有前景的策略,但现有方法多依赖人工规则设计。本文提出AutoRule,一种完全自动化的规则提取方法,从偏好反馈中生成规则并转化为规则奖励。该方法分三阶段:利用推理模型解读用户偏好,从推理链中识别候选规则,并合成统一规则集。基于最终规则集,使用语言模型验证器计算每条输出满足规则的比例,作为辅助奖励与学习到的奖励模型联合优化策略。在Llama-3-8B上训练时,相较于仅使用学习奖励模型的GRPO基线,AutoRule在AlpacaEval2.0长度控制胜率上实现28.6%相对提升,在保留的MT-Bench子集第二轮对话性能上提升6.1%。分析表明,提取规则与数据集偏好高度一致;在两轮测试中,AutoRule比学习奖励模型更少出现奖励黑客行为。案例研究显示,提取规则捕捉了不同数据集中特有的价值特征。规则详情见附录,代码已开源至https://github.com/cxcscmu/AutoRule。
原文摘要 · Abstract (English)
Rule-based rewards offer a promising strategy for improving reinforcement learning from human feedback (RLHF), but current approaches often rely on manual rule engineering. We present AutoRule, a fully automated method for extracting rules from preference feedback and formulating them into rule-based rewards. AutoRule extraction operates in three stages: it leverages a reasoning model to interpret user preferences, identifies candidate rules from the reasoning chain of these interpretations, and synthesizes them into a unified rule set. Leveraging the finalized rule set, we employ language-model verifiers to compute the fraction of rules satisfied by each output, using this metric as an auxiliary reward alongside the learned reward model during policy optimization. Training a Llama-3-8B model with AutoRule results in a 28.6\% relative improvement in length-controlled win rate on AlpacaEval2.0, and a 6.1\% relative gain in second-turn performance on a held-out MT-Bench subset, compared to a GRPO baseline trained with the same learned reward model but without the rule-based auxiliary reward. Our analysis confirms that the extracted rules exhibit good agreement with dataset preference. We find that AutoRule demonstrates reduced reward hacking compared to a learned reward model when run over two episodes. Finally, our case study suggests that the extracted rules capture unique qualities valued in different datasets. The extracted rules are provided in the appendix, and the code is open-sourced at https://github.com/cxcscmu/AutoRule.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。