arXiv:2509.21482cs.CLcs.AI2025-09被引 3

让大模型在连续混合词元空间推理,提升效率与准确性

Learning to Reason with Mixture of Tokens

  • 在连续混合词元空间中进行强化学习推理,保留概率分布信息
  • 在7个任务上比标准解码提升5%~35%,轨迹数减半仍达相近准确率
  • 适合需要高效高精度推理的场景,如复杂逻辑任务研究

基于可验证奖励的强化学习(RLVR)已成为提升大语言模型推理能力的主流方法。现有方法多采用分组相对策略优化,每步采样离散词元,忽略了模型概率分布中的丰富信息。本文提出在RLVR中使用混合词元生成(MoT-G),构建统一框架,将训练自由的方法扩展至连续混合空间,直接生成思维链。在Reasoning-Gym基准上测试两种MoT-G变体,使用Qwen2.5-1.5B模型,在10个任务中的7个取得5%~35%的性能提升,且仅需一半轨迹数即达相当准确率,表明训练效率显著提高。通过隐藏状态与词元级分析,发现MoT-G能维持更高隐藏状态熵,促进词元空间探索。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) has become a leading approach for improving large language model (LLM) reasoning capabilities. Most current methods follow variants of Group Relative Policy Optimization, which samples multiple reasoning completions, scores them relative to each other, and adjusts the policy accordingly. However, these approaches invariably sample discrete tokens at each reasoning step, discarding the rich distributional information in the model's probability distribution over candidate tokens. While preserving and utilizing this distributional information has proven beneficial in non-RL settings, current RLVR methods seem to be unnecessarily constraining the reasoning search space by not using this information. To address this limitation, we investigate mixture-of-token generation (MoT-G) in RLVR. We present a unified framework that generalizes existing MoT-G approaches, including existing training-free methods that construct mixture embeddings as weighted sums over token embeddings, and extend RLVR to operate directly in this continuous mixture space for generating chain-of-thought. Evaluating two MoT-G variants on Reasoning-Gym, a suite of reasoning-intensive language tasks, we find that MoT--G methods achieve substantial improvements (5--35 \% gains on 7 out of 10 tasks) compared to standard decoding with the Qwen2.5-1.5B model, while reaching comparable accuracy with half the number of trajectories, suggesting improved training efficiency. Through comprehensive hidden-state and token-level analyses, we provide evidence that MoT--G's benefits may stem from its ability to maintain higher hidden-state entropy throughout the reasoning process and promote exploration in token space.

大模型推理强化学习混合词元

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