用预训练模型的隐藏状态设计探索奖励,提升语言模型推理多样性与效率。
Representation-Based Exploration for Language Models: From Test-Time to Post-Training
- 基于预训练模型隐藏状态设计可解释的探索奖励,引导发现新行为。
- 推理时使用该方法,使Qwen-2.5-14b在多数任务上验证效率提升超50%。
- 后训练阶段实现3倍测试样本效率提升,显著优于标准强化学习方法。
强化学习(RL)有望拓展语言模型能力,但现有技术是否真正促进新行为发现仍不明确。本文研究主动探索的价值——显式激励模型发现新颖且多样化的行为,并探究预训练模型知识如何引导这一搜索过程。核心发现:基于预训练语言模型隐藏状态构建的、原理清晰的表示奖励,显著提升推理多样性与pass@k指标。该方法既适用于后训练,也适用于我们提出的新推理时扩展设置。在推理阶段,使用表示多样性探索可显著提高效率,在多个模型和推理任务中稳定提升pass@k率;例如,Qwen-2.5-14b-Instruct在多数任务上验证效率提升超过50%。在后训练阶段,将该探索策略融入强化学习流程,使推理性能超越初始模型及标准强化学习后训练方法。例如,在AIME 2024上,经过后训练的Qwen-2.5-7b-Instruct的pass@80达到同模型下GRPO的pass@256水平,测试样本效率提升3倍。整体表明,恰当的主动探索是突破原有能力边界、发现新行为的可行路径。
原文摘要 · Abstract (English)
Reinforcement learning (RL) promises to expand the capabilities of language models, but it is unclear if current RL techniques promote the discovery of novel behaviors, or simply sharpen those already present in the base model. In this paper, we investigate the value of deliberate exploration -- explicitly incentivizing the model to discover novel and diverse behaviors -- and aim to understand how the knowledge in pre-trained models can guide this search. Our main finding is that exploration with a simple, principled, representation-based bonus derived from the pre-trained language model's hidden states significantly improves diversity and pass@k rates -- both for post-training, and in a novel inference-time scaling setting we introduce. For inference-time, exploration with representation-based diversity improves efficiency, consistently improving pass@k rates across a variety of models and reasoning tasks. For example, for Qwen-2.5-14b-Instruct we obtain over 50% improvement in verifier efficiency on almost all tasks. For post-training, we show that integrating this exploration strategy into an RL pipeline improves reasoning performance over that of the initial model and over standard RL post-training. For example, on AIME 2024, our post-trained Qwen-2.5-7b-Instruct's pass@80 matches the pass@256 of GRPO on the same model, demonstrating a 3x improvement in test-time sample efficiency. Overall, our findings suggest that deliberate exploration -- with the right notion of diversity -- is a practical path toward discovery of new behaviors beyond sharpening.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。