用新算法让大模型强化学习更稳更快,不依赖KL正则也能高效训练。
Tapered Off-Policy REINFORCE: Stable and efficient reinforcement learning for LLMs
- 采用非对称截断重要性采样加速学习,保持稳定训练动态。
- 在GSM8K和MATH上提升生成与验证性能,数据效率显著提高。
- 可离线使用,统一处理正负样本,适合大模型微调场景。
我们提出一种用于大语言模型强化学习微调的新算法——截断离策略REINFORCE(TOPR)。该方法通过不对称的截断重要性采样,在无需KL正则化的情况下实现快速且稳定的训练。TOPR支持完全离线训练,统一处理正负样本,兼具蒙特卡洛方法的实现简洁性。在GSM8K和MATH推理基准上的实验表明,该方法能同时提升解题生成与生成验证的测试准确率,并显著提高训练数据效率。通过合理利用正负样本,避免了传统方法中“无效推理”的浪费。该优势在多轮训练中持续存在,结合数据集优化技术后,可使80亿参数模型达到700亿参数模型的性能水平。此外,我们发现REINFORCE中的基线参数在含负样本时具有意外关键作用,直接影响离策略性能表现。
原文摘要 · Abstract (English)
We propose a new algorithm for fine-tuning large language models using reinforcement learning. Tapered Off-Policy REINFORCE (TOPR) uses an asymmetric, tapered variant of importance sampling to speed up learning while maintaining stable learning dynamics, even without the use of KL regularization. TOPR can be applied in a fully offline fashion, allows the handling of positive and negative examples in a unified framework, and benefits from the implementational simplicity that is typical of Monte Carlo algorithms. We demonstrate the effectiveness of our approach with a series of experiments on the GSM8K and MATH reasoning benchmarks, finding performance gains for training both a model for solution generation and as a generative verifier. We show that properly leveraging positive and negative examples alike in the off-policy regime simultaneously increases test-time accuracy and training data efficiency, all the while avoiding the ``wasted inference'' that comes with discarding negative examples. We find that this advantage persists over multiple iterations of training and can be amplified by dataset curation techniques, enabling us to match 70B-parameter model performance with 8B language models. As a corollary to this work, we find that REINFORCE's baseline parameter plays an important and unexpected role in defining dataset composition in the presence of negative examples, and is consequently critical in driving off-policy performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。