arXiv:2506.15710cs.LGcs.AI2025-06NeurIPS被引 8

用小模型训练的推理能力,高效迁移给大模型

RAST: Reasoning Activation in LLMs via Small-model Transfer

  • 小模型经强化学习训练后,提取其概率调整模式
  • 迁移该模式到大模型,显著提升数学推理能力
  • 节省大量显存,效果甚至优于直接强化学习

强化学习(RL)已证明能有效提升大语言模型(LLM)的推理能力,如OpenAI的o1和Deepseek-R1。但大规模应用仍需高昂资源,包括多个模型副本和大量GPU算力。近期研究指出,RL并未赋予模型新知识,而是激活了基础模型中已存在的推理潜能,主要通过改变输出概率分布实现。基于此,我们提出假设:RL引起的概率变化在不同模型规模间具有高度一致性。通过逐标记分析解码轨迹,我们验证了该假设。据此,我们提出RAST方法——将小模型经RL训练后产生的概率调整注入大模型,以转移推理行为。在多个数学推理基准测试中,RAST显著且稳定地提升了基础模型的推理能力,所需显存远低于直接强化学习,部分场景下性能甚至超过原生强化学习模型。本研究揭示了强化学习驱动推理的本质,并提供了无需承担全部计算成本即可扩展其优势的实用路径。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become a powerful approach for improving the reasoning capabilities of large language models (LLMs), as evidenced by recent successes such as OpenAI's o1 and Deepseek-R1. However, applying RL at scale remains intimidatingly resource-intensive, requiring multiple model copies and extensive GPU workloads. On the other hand, while being powerful, recent studies suggest that RL does not fundamentally endow models with new knowledge; rather, it primarily reshapes the model's output distribution to activate reasoning capabilities latent in the base model. Building on this insight, we hypothesize that the changes in output probabilities induced by RL are largely model-size invariant, opening the door to a more efficient paradigm: training a small model with RL and transferring its induced probability shifts to larger base models. To verify our hypothesis, we conduct a token-level analysis of decoding trajectories and find high alignment in RL-induced output distributions across model scales, validating our hypothesis. Motivated by this, we propose RAST, a simple yet effective method that transfers reasoning behaviors by injecting RL-induced probability adjustments from a small RL-trained model into larger models. Experiments across multiple mathematical reasoning benchmarks show that RAST substantially and consistently enhances the reasoning capabilities of base models while requiring significantly lower GPU memory than direct RL training, sometimes even yielding better performance than the RL-trained counterparts. Our findings offer new insights into the nature of RL-driven reasoning and practical strategies for scaling its benefits without incurring its full computational cost. The project page of RAST is available at https://ozyyshr.github.io/RAST/.

强化学习模型迁移推理增强

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