arXiv:2510.08892cs.CLcs.AI2025-10被引 8

通过不同温度控制推理与知识令牌,提升大模型推理能力。

Exploring Multi-Temperature Strategies for Token- and Rollout-Level Control in RLVR

  • 对推理令牌用高温度,知识令牌用低温度,实现生成阶段主动探索。
  • 在多个推理基准上性能显著提升,验证方法有效性。
  • 适合研究大模型推理优化或强化学习生成控制的读者。

强化学习已显著提升大语言模型(LLMs)的推理能力,应用广泛。近期研究发现,LLM中的令牌在推理任务中扮演不同角色,可分为高熵推理令牌和低熵知识令牌。以往方法通常通过限制更新间接鼓励探索,但未在生成阶段直接促进探索行为。本文提出一种互补方法:在采样阶段为不同类型的令牌设置不同温度,对推理令牌使用较高温度以主动鼓励探索,对知识令牌保留较低温度以保持事实正确性。同时,系统研究了多种多温度调度策略在强化学习环境中的影响。在多个推理基准上的实证评估表明,该方法显著提升了LLMs的推理性能。代码已公开于 https://github.com/zhmzm/Multi_Temperature_Verl.git。

原文摘要 · Abstract (English)

Reinforcement Learning has demonstrated substantial improvements in the reasoning abilities of Large Language Models (LLMs), exhibiting significant applicability across various domains. Recent research has identified that tokens within LLMs play distinct roles during reasoning tasks, categorizing them into high-entropy reasoning tokens and low-entropy knowledge tokens. Prior approaches have typically focused on restricting updates to indirectly encourage exploration, yet they do not explicitly facilitate exploratory behavior during the token generation stage itself. In this work, we introduce a complementary approach that explicitly promotes exploration during sampling by applying distinct temperature settings for different token types. Specifically, our method employs higher temperatures for reasoning tokens to actively encourage exploration, while retaining lower temperatures for knowledge tokens to maintain factual correctness. Furthermore, we systematically investigate various multi-temperature scheduling strategies and their impacts within reinforcement learning contexts. Empirical evaluations on several reasoning benchmarks demonstrate that our approach significantly enhances the reasoning performance of LLMs. The code is available at https://github.com/zhmzm/Multi_Temperature_Verl.git.

强化学习大模型推理优化温度控制

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