arXiv:2411.00062cs.CLcs.AI2024-11被引 19

让AI自己设计训练提示,大幅提升模型对齐效果。

Scalable Reinforcement Post-Training Beyond Static Human Prompts: Evolving Alignment via Asymmetric Self-Play

  • 用不对称自博弈让模型自主生成训练提示。
  • Gemma-2-9b-it在Arena-Hard上胜率提升至62.4%。
  • 无需人工提示,适合追求高效对齐的AI研发者。

当前大语言模型强化学习后训练框架通常假设提示分布固定,限制了可扩展性。以往工作虽探索提示演化,但多局限于监督微调阶段,且提示均匀采样无反馈信号。本文提出一种范式转变:通过不对称自博弈实现对齐演化(eva),将后训练视为无限博弈,由创造者动态生成新提示,求解者学习生成偏好响应。eva是首个支持离线与在线强化学习中自适应生成训练提示的方法。设计简单有效:在挑战性基准上达到新SOTA,无需额外人工提示。例如,Gemma-2-9b-it在Arena-Hard上的胜率从51.6%(DPO)和52.6%(RLOO)分别提升至60.1%和62.4%,超越Claude-3-Opus,接近更大规模的Gemini-1.5-Pro。大量实验表明,eva能生成有效强化学习课程,且对消融鲁棒。我们认为自适应演化提示是下一代强化学习后训练的关键。

原文摘要 · Abstract (English)

Current reinforcement learning (RL) frameworks for large language models (LLM) post-training typically assume a fixed prompt distribution, which is sub-optimal and bottlenecks scalability. Prior works have explored prompt evolving, but are often limited to the supervised fine-tuning stage, and prompts are sampled and evolved uniformly without signals. This empirical work presents a paradigm shift: Evolving Alignment via Asymmetric Self-Play (eva), that casts post-training as an infinite game with regret-based signals for 2 players: (i) a creator, who strategically samples and creates new informative prompts and (ii) a solver, who learns to produce preferred responses. eva is the first method that allows language models to adaptively create training prompts in both offline and online RL post-training. The design is simple, easy-to-use yet remarkably effective: eva sets a new SOTA on challenging benchmarks, without any extra human prompts, e.g. it boosts the win-rate of gemma-2-9b-it on Arena-Hard by 51.6% -> 60.1% for DPO and 52.6% -> 62.4% for RLOO, surpassing claude-3-opus and catching up to gemini-1.5-pro, both of which are orders of magnitude larger. Extensive experiments show eva can create effective RL curricula and is robust across ablations. We believe adaptively evolving prompts are key to designing the next-generation RL post-training scheme.

强化学习提示演化对齐优化自博弈

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