RL微调会放大预训练数据中的行为模式,影响模型推理表现。
Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretraining
- 从零训练模型,用开放数据集系统研究RL微调机制。
- 不同规模模型在相同数据下收敛到不同输出分布,存在规模依赖偏差。
- 简单任务微调可提升复杂任务表现,说明推理能力具有泛化性。
基于强化学习(RL)的微调已成为提升语言模型数学推理与编程能力的关键步骤。尽管前沿模型的成功表明RL微调能持续提升性能,甚至在小模型上也有效,但其背后机制尚不明确。理解该过程需分离预训练数据构成、超参数和模型规模的影响,但现有模型常缺乏透明训练数据。本文通过完全从头训练模型,在多种全开源数据混合上系统研究了数学推理的RL微调。我们考察了PPO、GRPO和专家迭代等算法在不同规模模型上的效果。结果发现,RL算法会稳定收敛至主导输出分布,放大预训练数据中的模式。不同规模模型在相同数据混合下趋向不同输出分布,表明模型泛化存在规模依赖偏差。此外,对简单问题进行RL微调可带来更难问题上的性能提升,表明某些推理能力具备跨任务泛化性。我们的研究证明,在可控环境下,小规模代理模型也能揭示RL对语言模型行为塑造的关键作用。
原文摘要 · Abstract (English)
Reinforcement learning (RL)-based fine-tuning has become a crucial step in post-training language models for advanced mathematical reasoning and coding. Following the success of frontier reasoning models, recent work has demonstrated that RL fine-tuning consistently improves performance, even in smaller-scale models; however, the underlying mechanisms driving these improvements are not well-understood. Understanding the effects of RL fine-tuning requires disentangling its interaction with pretraining data composition, hyperparameters, and model scale, but such problems are exacerbated by the lack of transparency regarding the training data used in many existing models. In this work, we present a systematic end-to-end study of RL fine-tuning for mathematical reasoning by training models entirely from scratch on different mixtures of fully open datasets. We investigate the effects of various RL fine-tuning algorithms (PPO, GRPO, and Expert Iteration) across models of different scales. Our study reveals that RL algorithms consistently converge towards a dominant output distribution, amplifying patterns in the pretraining data. We also find that models of different scales trained on the same data mixture will converge to distinct output distributions, suggesting that there are scale-dependent biases in model generalization. Moreover, we find that RL post-training on simpler questions can lead to performance gains on harder ones, indicating that certain reasoning capabilities generalize across tasks. Our findings show that small-scale proxies in controlled settings can elicit interesting insights regarding the role of RL in shaping language model behavior.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。