arXiv:2509.19170cs.CLcs.AI2025-09被引 17

用强化学习训练大模型连续思维链,提升推理多样性与泛化能力。

Soft Tokens, Hard Truths

  • 通过强化学习直接训练连续思维链,无需依赖离散参考路径。
  • 在数学推理任务上,连续思维链的 pass@32 显著优于离散版本。
  • 训练后可转换为标准离散模型部署,且更保留学界外任务预测能力。

近年来,基于连续思维链(CoT)的大型语言模型推理方法受到关注,其核心思想是通过连续混合离散标记来同时模拟多种推理路径。理论证明连续标记具有更强表达能力,能更高效解决特定问题。然而,实际应用受限于训练难度:此前工作要么仅在推理阶段使用连续标记,要么需从真实离散思维链中蒸馏,计算开销大,导致思维链长度受限。本文首次提出一种可扩展的强化学习方法,无需蒸馏即可学习连续思维链。采用“软”标记机制,即在输入嵌入中引入噪声以实现探索。计算开销极小,支持数百个标记的连续思维链训练。在 Llama 与 Qwen 模型(最大 8B)的数学推理基准测试中,连续思维链的 pass@1 与离散版本相当,pass@32 则显著超越,表明推理路径多样性更高。系统对比显示,最优方案为:训练时使用连续标记,推理时转为离散标记,实现“软”模型向标准部署的平滑过渡。此外,连续思维链训练更能保留基模型在域外任务上的预测结果,对基模型影响更温和。

原文摘要 · Abstract (English)

The use of continuous instead of discrete tokens during the Chain-of-Thought (CoT) phase of reasoning LLMs has garnered attention recently, based on the intuition that a continuous mixture of discrete tokens could simulate a superposition of several reasoning paths simultaneously. Theoretical results have formally proven that continuous tokens have much greater expressivity and can solve specific problems more efficiently. However, practical use of continuous tokens has been limited by strong training difficulties: previous works either just use continuous tokens at inference time on a pre-trained discrete-token model, or must distill the continuous CoT from ground-truth discrete CoTs and face computational costs that limit the CoT to very few tokens. This is the first work introducing a scalable method to learn continuous CoTs via reinforcement learning (RL), without distilling from reference discrete CoTs. We use "soft" tokens: mixtures of tokens together with noise on the input embedding to provide RL exploration. Computational overhead is minimal, enabling us to learn continuous CoTs with hundreds of tokens. On math reasoning benchmarks with Llama and Qwen models up to 8B, training with continuous CoTs match discrete-token CoTs for pass@1 and surpass them for pass@32, showing greater CoT diversity. In systematic comparisons, the best-performing scenario is to train with continuous CoT tokens then use discrete tokens for inference, meaning the "soft" models can be deployed in a standard way. Finally, we show continuous CoT RL training better preserves the predictions of the base model on out-of-domain tasks, thus providing a softer touch to the base model.

思维链强化学习连续标记大模型推理

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