arXiv:2505.12717cs.CL2025-05被引 6

用解谜游戏训练大模型,让其学会并行思考,提升推理效率。

ToTRL: Unlock LLM Tree-of-Thoughts Reasoning Potential through Puzzles Solving

  • 基于规则奖励的强化学习框架,引导模型从串行思考转向并行树状推理。
  • 在复杂推理任务上,模型性能显著提升,推理耗时和token数均减少。
  • 适合想提升大模型逻辑推理能力的研究者与开发者参考。

大型语言模型(LLMs)展现出强大的推理能力,尤其通过长链式思维(CoT)过程实现,该能力可通过强化学习(RL)激发。然而,长时间的CoT推理存在输出冗长的问题,且过程多为试错而非系统性逻辑推导。相比之下,思维树(ToT)方法更具先进性,将推理建模为树结构中的探索,支持并行生成与评估多个推理分支,可主动识别、评估并剪枝无效路径,从而提高性能并降低token消耗。本文提出一种新的在线策略强化学习框架ToTRL,基于规则奖励,引导模型从序列式CoT过渡到并行式ToT。训练中,使用大模型作为解谜游戏的玩家,此类任务要求处理相互依赖的选择与多重约束,天然适合构建与探索思维树,是培养ToT推理能力的理想场景。实验表明,经ToTRL训练的ToTQwen3-8B模型,在复杂推理任务上表现显著提升,推理效率更高,生成更精炼。

原文摘要 · Abstract (English)

Large language models (LLMs) demonstrate significant reasoning capabilities, particularly through long chain-of-thought (CoT) processes, which can be elicited by reinforcement learning (RL). However, prolonged CoT reasoning presents limitations, primarily verbose outputs due to excessive introspection. The reasoning process in these LLMs often appears to follow a trial-and-error methodology rather than a systematic, logical deduction. In contrast, tree-of-thoughts (ToT) offers a conceptually more advanced approach by modeling reasoning as an exploration within a tree structure. This reasoning structure facilitates the parallel generation and evaluation of multiple reasoning branches, allowing for the active identification, assessment, and pruning of unproductive paths. This process can potentially lead to improved performance and reduced token costs. Building upon the long CoT capability of LLMs, we introduce tree-of-thoughts RL (ToTRL), a novel on-policy RL framework with a rule-based reward. ToTRL is designed to guide LLMs in developing the parallel ToT strategy based on the sequential CoT strategy. Furthermore, we employ LLMs as players in a puzzle game during the ToTRL training process. Solving puzzle games inherently necessitates exploring interdependent choices and managing multiple constraints, which requires the construction and exploration of a thought tree, providing challenging tasks for cultivating the ToT reasoning capability. Our empirical evaluations demonstrate that our ToTQwen3-8B model, trained with our ToTRL, achieves significant improvement in performance and reasoning efficiency on complex reasoning tasks.

思维树强化学习推理优化大模型

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