arXiv:2505.14140cs.AI2025-05被引 19

用强化学习动态生成推理路径,让小模型也能有大模型的思考能力。

RL of Thoughts: Navigating LLM Reasoning with Inference-time Reinforcement Learning

  • 训练轻量导航器,按问题特点自动组合逻辑模块。
  • 在多个评测中提升效果,小模型性能接近百倍大的模型。
  • 仅3000参数,跨模型跨任务通用性强,适合部署优化。

尽管大型语言模型(LLMs)进展迅速,其逐标记自回归特性仍限制了复杂推理能力。为增强推理,推理时技术如思维链/树/图(Chain/Tree/Graph-of-Thoughts)通过构建复杂逻辑结构有效提升性能,且无需修改模型参数,成本较低。但这些手动预设、与任务无关的框架对所有任务统一使用,缺乏适应性。为此,我们提出强化学习思维(RLoT),训练一个轻量级导航器模型,利用强化学习在推理时自适应优化LLM推理。具体地,从人类认知视角设计五种基础逻辑块;推理过程中,训练好的RL导航器根据问题特征动态选择并组合逻辑块,形成特定任务的逻辑结构。在多个推理基准(AIME、MATH、GPQA等)上,使用多种模型(GPT、Llama、Qwen、DeepSeek)的实验表明,RLoT相比现有推理技术最高提升13.4%。尤为突出的是,仅需不到3000个参数,该导航器使小于100亿参数的模型达到1000亿规模模型的水平。此外,导航器展现出强泛化能力:在某一模型-任务对上训练后,可有效迁移至未见的模型和任务。代码已开源,便于复现。

原文摘要 · Abstract (English)

Despite rapid advancements in large language models (LLMs), the token-level autoregressive nature constrains their complex reasoning capabilities. To enhance LLM reasoning, inference-time techniques, including Chain/Tree/Graph-of-Thought(s), successfully improve the performance, as they are fairly cost-effective by guiding reasoning through sophisticated logical structures without modifying LLMs' parameters. However, these manually predefined, task-agnostic frameworks are applied uniformly across diverse tasks, lacking adaptability. To improve this, we propose RL-of-Thoughts (RLoT), where we train a lightweight navigator model with reinforcement learning (RL) to adaptively enhance LLM reasoning at inference time. Specifically, we design five basic logic blocks from the perspective of human cognition. During the reasoning process, the trained RL navigator dynamically selects the suitable logic blocks and combines them into task-specific logical structures according to problem characteristics. Experiments across multiple reasoning benchmarks (AIME, MATH, GPQA, etc.) with multiple LLMs (GPT, Llama, Qwen, and DeepSeek) illustrate that RLoT outperforms established inference-time techniques by up to 13.4%. Remarkably, with less than 3K parameters, our RL navigator is able to make sub-10B LLMs comparable to 100B-scale counterparts. Moreover, the RL navigator demonstrates strong transferability: a model trained on one specific LLM-task pair can effectively generalize to unseen LLMs and tasks. Our code is open-source at https://github.com/tsinghua-fib-lab/RL-LLM-Reasoning for reproducibility.

强化学习推理优化轻量化思维链

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