用自生成多样解法数据,提升大模型强化学习效果
Mid-Training with Self-Generated Data Improves Reinforcement Learning in Language Models

- 用波利亚解题法生成多路径正确答案,丰富训练数据
- 在数学推理等任务上显著优于基线模型,提升稳定
- 适合想提升模型泛化能力的RL研究者
大型语言模型的强化学习效果取决于训练前和训练中数据的性质与多样性。特别是,推理问题常有多种解法,若训练数据仅覆盖有限解法,会限制强化学习效果。为此,我们提出在强化学习中间阶段引入自生成的多样化数据作为预训练步骤。具体地,采用基于乔治·波利亚问题解决方法的自举式数据生成框架,为每个问题生成多种正确答案变体,并进行微调。我们从理论上分析了该方法如何通过策略梯度更新激励模型融合多种解法。实验表明,使用该方法初始化的强化学习模型在多个数学推理基准及代码生成、叙事推理等分布外任务上均取得一致提升。
原文摘要 · Abstract (English)
The effectiveness of Reinforcement Learning (RL) in Large Language Models (LLMs) depends on the nature and diversity of the data used before and during RL. In particular, reasoning problems can often be approached in multiple ways that rely on different forms of reasoning, and exposure to only a limited range of such approaches in the training data may limit the effectiveness of RL. Motivated by this, we investigate using diverse self-generated data during mid-training as an intermediate step before RL training. Specifically, we adopt a bootstrapped data-generation framework guided by George Polya's problem-solving approaches for generating multiple variants of correct answers for each question in the training data, and then perform fine-tuning. We first provide a theoretical perspective on how mid-training on such data improves RL and explain how policy-gradient updates can incentivize combining multiple approaches. We then empirically demonstrate that RL-trained models initialized with our mid-training data achieve consistent improvements across various mathematical reasoning benchmarks and other OOD tasks like code generation and narrative reasoning. Overall, our investigative study shows that a language model learning multiple problem-solving approaches, through self-generated data helps subsequent RL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。