arXiv:2506.05748cs.LGcs.AI2025-06被引 1

用小模型加微调实现高效在线奖励判断,性能超大模型。

Efficient Online RFT with Plug-and-Play LLM Judges: Unlocking State-of-the-Art Performance

  • 用70亿参数的冻结大模型加极小LoRA适配器做奖励判断。
  • 在GSM-8K上达92%准确率,超越700亿参数基线。
  • 可解释性强,生成理由与人类相似度达9/10,适合部署调试。

现代强化学习人类反馈(RLHF)中,奖励模型训练是主要成本瓶颈,通常需数十亿参数并进行离线偏好调优。本文提出一种高效在线奖励判断方法:将一个指令微调过的70亿参数大模型(LLM)仅通过一行JSON规则和一个秩为16的LoRA适配器(仅影响0.8%参数)即可替代原有重型评估模型。该即插即用的裁判在RewardBench上达到96.2%准确率,优于270亿至700亿参数的专用奖励网络。同时,使用该裁判的70亿参数智能体在在线PPO训练中,于GSM-8K任务上取得92%精确匹配准确率,超越700亿参数的DPO基线(61.8%)。消融实验表明,六条上下文示例贡献了大部分零样本到少样本性能提升(+2个百分点),而LoRA有效缓解了安全性和对抗性场景下的差距。研究引入HH-Rationales——来自Anthropic HH-RLHF的10,000对数据,附带人工生成的解释,经GPT-4评估,其生成理由与人类相似度约为9/10,远高于零样本模型的5/10。结果表明,提示工程结合微小LoRA可构建低成本、高透明、易调整的奖励函数,在移除离线阶段的同时,实现静态评估与在线RLHF的新纪录。

原文摘要 · Abstract (English)

Reward-model training is the cost bottleneck in modern Reinforcement Learning Human Feedback (RLHF) pipelines, often requiring tens of billions of parameters and an offline preference-tuning phase. In the proposed method, a frozen, instruction-tuned 7B LLM is augmented with only a one line JSON rubric and a rank-16 LoRA adapter (affecting just 0.8% of the model's parameters), enabling it to serve as a complete substitute for the previously used heavyweight evaluation models. The plug-and-play judge achieves 96.2% accuracy on RewardBench, outperforming specialized reward networks ranging from 27B to 70B parameters. Additionally, it allows a 7B actor to outperform the top 70B DPO baseline, which scores 61.8%, by achieving 92% exact match accuracy on GSM-8K utilizing online PPO. Thorough ablations indicate that (i) six in context demonstrations deliver the majority of the zero-to-few-shot improvements (+2pp), and (ii) the LoRA effectively addresses the remaining disparity, particularly in the safety and adversarial Chat-Hard segments. The proposed model introduces HH-Rationales, a subset of 10,000 pairs from Anthropic HH-RLHF, to examine interpretability, accompanied by human generated justifications. GPT-4 scoring indicates that our LoRA judge attains approximately = 9/10 in similarity to human explanations, while zero-shot judges score around =5/10. These results indicate that the combination of prompt engineering and tiny LoRA produces a cost effective, transparent, and easily adjustable reward function, removing the offline phase while achieving new state-of-the-art outcomes for both static evaluation and online RLHF.

RLHF大模型奖励模型轻量化

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