arXiv:2602.11683cs.AIcs.CL2026-02被引 3

通过动态切换思维空间提升大模型推理效率与准确率

ThinkRouter: Efficient Reasoning via Routing Thinking between Latent and Discrete Spaces

  • 低置信度时转至离散符号空间,高置信度时保留潜在空间推理
  • 在多个基准上平均提升19.70分,生成长度减少最多15.55%
  • 适用于需要高效高精度推理的复杂任务场景

近期研究尝试用潜在空间中的连续表示替代显式思维轨迹以提升推理效率,但效果因场景而异。分析发现,错误答案对应的思维轨迹中低置信度步骤更少。我们提出,多个低置信度思维路径聚合产生的软嵌入可能引入并传播噪声,导致不可靠路径出现高置信度。为此,提出ThinkRouter:一种推理时的置信度感知路由机制,在模型置信度低时路由至离散标记空间,否则保留潜在空间。在多样大型推理模型的STEM推理与编码基准上,ThinkRouter在准确率上优于显式思维链(CoT)、随机路由及潜在推理基线,平均提升Pass@1达19.70分,生成长度最多减少15.55%。全面分析显示,该方法可校正显式CoT与潜在推理的错误,并通过全局降低模型置信度加速结束思维标记生成。

原文摘要 · Abstract (English)

Recent work explores latent reasoning to improve reasoning efficiency by replacing explicit reasoning trajectories with continuous representations in a latent space, yet its effectiveness varies across settings. Analysis of model confidence dynamics under latent reasoning reveals that thinking trajectories ending in incorrect answers contain fewer low-confidence steps than those ending in correct answers. Meanwhile, we suggest that soft embeddings aggregated by multiple low-confidence thinking alternatives may introduce and propagate noise, leading to high confidence in unreliable reasoning trajectories. Motivated by these observations, ThinkRouter, an inference-time confidence-aware routing mechanism is proposed to avoid high confidence and noise for efficient reasoning. ThinkRouter routes thinking to the discrete token space when model confidence is low, and to the latent space otherwise. Extensive experiments on STEM reasoning and coding benchmarks across diverse large reasoning models demonstrate that ThinkRouter outperforms explicit CoT, random routing, and latent reasoning baselines in terms of accuracy, achieving an average improvement of 19.70 points in Pass@1, while reducing generation length by up to 15.55%. Further comprehensive analysis reveals that ThinkRouter can calibrate errors arising from explicit CoT and latent reasoning, and accelerates end-of-thinking token generation by globally lowering model confidence.

推理优化思维路由大模型

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