用简单匹配指标BLEU做奖励,也能有效对齐大模型指令遵循能力
BLEUBERI: BLEU is a surprisingly effective reward for instruction following
- 用BLEU直接作为强化学习奖励,替代昂贵的偏好标注
- 在四个基准上表现媲美传统奖励模型,且更注重事实准确性
- 适合资源有限但需高效对齐的模型训练场景
奖励模型在对齐大语言模型与人类偏好中至关重要,但训练成本高昂,依赖大规模人工标注偏好数据和强大的预训练模型。随着高质量合成指令遵循数据集日益普及,我们提出:能否用更简单的参考文本匹配指标替代奖励模型?本文发现,基础的字符串匹配指标BLEU在通用指令遵循数据集上,与人类偏好的一致性可媲美强效奖励模型。基于此,我们提出BLEUBERI方法:先识别难例指令,再使用组相对策略优化(GRPO)以BLEU为奖励函数进行训练。实验表明,BLEUBERI训练的模型在四个挑战性指令遵循基准和三种不同基础模型上,性能可与奖励模型引导的强化学习模型相媲美。人工评估进一步证实,其输出质量与奖励模型对齐模型相当,且生成内容更具事实依据。总体而言,只要拥有高质量参考输出(可通过现有指令遵循数据集或合成数据生成获得),基于字符串匹配的指标即可作为廉价而有效的奖励模型替代方案。代码与数据已开源。
原文摘要 · Abstract (English)
Reward models are central to aligning LLMs with human preferences, but they are costly to train, requiring large-scale human-labeled preference data and powerful pretrained LLM backbones. Meanwhile, the increasing availability of high-quality synthetic instruction-following datasets raises the question: can simpler, reference-based metrics serve as viable alternatives to reward models during RL-based alignment? In this paper, we show first that BLEU, a basic string-matching metric, surprisingly matches strong reward models in agreement with human preferences on general instruction-following datasets. Based on this insight, we develop BLEUBERI, a method that first identifies challenging instructions and then applies Group Relative Policy Optimization (GRPO) using BLEU directly as the reward function. We demonstrate that BLEUBERI-trained models are competitive with models trained via reward model-guided RL across four challenging instruction-following benchmarks and three different base language models. A human evaluation further supports that the quality of BLEUBERI model outputs is on par with those from reward model-aligned models. Moreover, BLEUBERI models generate outputs that are more factually grounded than competing methods. Overall, we show that given access to high-quality reference outputs (easily obtained via existing instruction-following datasets or synthetic data generation), string matching-based metrics are cheap yet effective proxies for reward models during alignment. We release our code and data at https://github.com/lilakk/BLEUBERI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。