arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2406.07657cs.LGcs.CL
收藏

OPTune:高效的在线偏好调优

OPTune: Efficient Online Preference Tuning

University of Maryland, College Park(马里兰大学帕克分校);Meta

Lichang Chen, Jiuhai Chen, Chenxi Liu, John Kirchenbauer, Davit Soselia, Chen Zhu, Tom Goldstein, Tianyi Zhou, Heng Huang

AI总结:

针对在线 RLHF 数据生成昂贵且质量不均的问题,OPTune 通过选择性生成高信息量响应并按对齐效用加权训练,在保持指令遵循能力的同时将训练速度提升 1.27–1.56 倍。

摘要:

基于人类反馈的强化学习(RLHF)对于使大型语言模型(LLM)与人类偏好保持一致至关重要。与被广泛研究的离线版 RLHF(例如直接偏好优化,DPO)相比,近期研究表明在线变体能够实现更好的对齐效果。然而,在线对齐需要即时生成新的训练数据,这不仅成本高昂、难以并行化,还面临质量和效用参差不齐的问题。本文提出了一种用于在线偏好调优的更高效数据探索策略 OPTune;该方法不依赖人工整理或预先收集的教师响应,而是动态采样信息丰富的响应,以进行同策略偏好对齐。在数据生成阶段,OPTune 只选择那些其(重新)生成的响应相比现有响应有可能提供更具信息量、更高质量训练信号的提示。在训练目标中,OPTune 根据每个生成响应(对)对改善对齐的效用进行加权,使学习能够集中于最有帮助的样本。在全部评估中,采用 OPTune 的 LLM 保持了标准偏好调优所提供的指令遵循优势,同时得益于高效的数据探索策略,训练速度提升了 1.27–1.56 倍。

英文摘要:

Reinforcement learning with human feedback~(RLHF) is critical for aligning Large Language Models (LLMs) with human preference. Compared to the widely studied offline version of RLHF, \emph{e.g.} direct preference optimization (DPO), recent works have shown that the online variants achieve even better alignment. However, online alignment requires on-the-fly generation of new training data, which is costly, hard to parallelize, and suffers from varying quality and utility. In this paper, we propose a more efficient data exploration strategy for online preference tuning (OPTune), which does not rely on human-curated or pre-collected teacher responses but dynamically samples informative responses for on-policy preference alignment. During data generation, OPTune only selects prompts whose (re)generated responses can potentially provide more informative and higher-quality training signals than the existing responses. In the training objective, OPTune reweights each generated response (pair) by its utility in improving the alignment so that learning can be focused on the most helpful samples. Throughout our evaluations, OPTune'd LLMs maintain the instruction-following benefits provided by standard preference tuning whilst enjoying 1.27-1.56x faster training speed due to the efficient data exploration strategy.

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

↑