arXiv:2601.05053cs.AIcs.CL2026-01ACL被引 7

通过语义多样性探索提升大模型推理效率与质量

Reinforced Efficient Reasoning via Semantically Diverse Exploration

  • 引入语义熵分支策略与ε探索机制,增强推理路径多样性
  • 长度感知的优势估计算法,奖励简洁准确的推理过程
  • 在多个数学推理数据集上超越GRPO等基线方法

基于可验证奖励的强化学习(RLVR)已被证明能有效提升大语言模型(LLMs)的推理能力。基于蒙特卡洛树搜索(MCTS)的扩展方法通过树状推理回溯,实现细粒度的分段信用分配,优于原始的RLVR(如GRPO)。然而现有方法仍存在探索多样性不足和推理效率低的问题。为此,本文提出一种名为ROSE的增强型高效推理框架。该方法引入基于语义熵的分支策略,从已有推理路径中捕捉语义不确定性,选择高语义差异点进行分支以生成新的推理路径;同时采用ε-探索机制,随机从根节点启动推理回溯,防止搜索陷入局部最优。为提升效率,设计了长度感知的分段级优势估计算法,奖励简洁且正确的推理链,惩罚冗长无益的推理过程。在Qwen和Llama模型上,于多个数学推理基准测试中全面验证了ROSE的有效性与高效性。代码已开源:https://github.com/ZiqiZhao1/ROSE-rl。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) has proven effective in enhancing the reasoning of large language models (LLMs). Monte Carlo Tree Search (MCTS)-based extensions improve upon vanilla RLVR (e.g., GRPO) by providing tree-based reasoning rollouts that enable fine-grained and segment-level credit assignment. However, existing methods still suffer from limited exploration diversity and inefficient reasoning. To address the above challenges, we propose reinforced efficient reasoning via semantically diverse explorations, i.e., ROSE, for LLMs. To encourage more diverse reasoning exploration, our method incorporates a semantic-entropy-based branching strategy and an $\varepsilon$-exploration mechanism. The former operates on already sampled reasoning rollouts to capture semantic uncertainty and select branching points with high semantic divergence to generate new successive reasoning paths, whereas the latter stochastically initiates reasoning rollouts from the root, preventing the search process from becoming overly local. To improve efficiency, we design a length-aware segment-level advantage estimator that rewards concise and correct reasoning while penalizing unnecessarily long reasoning chains. Extensive experiments on various mathematical reasoning benchmarks with Qwen and Llama models validate the effectiveness and efficiency of ROSE. Codes are available at https://github.com/ZiqiZhao1/ROSE-rl.

大模型推理强化学习语义探索高效推理

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