arXiv:2509.26314cs.CL2025-09被引 15

让大模型在隐空间中自我纠错,提升推理效率与准确性。

Latent Thinking Optimization: Your Latent Reasoning Language Model Secretly Encodes Reward Signals in Its Latent Thoughts

  • 用隐空间中的思维模式区分对错,构建可监督的隐式奖励模型。
  • 在多个任务上,该方法使错误推理模式检测准确率达90%以上。
  • 无需修改模型结构,即可通用提升各类大模型的推理能力。

大型语言模型通过自然语言生成思维链进行问题求解,但该方式计算成本高且易过度思考。近期工作提出隐空间思维架构Huginn-3.5B,以隐向量序列表示中间推理步骤,但其缺乏可解释性且难以监督,引发对推理正确性的担忧。本文系统研究了Huginn-3.5B在隐空间中的推理模式,发现正确与错误答案对应的隐思维具有显著可区分特征,且隐分类器可直接从隐思维预测答案正确性。基于此,提出隐空间思维优化(LTO)算法,利用隐分类器作为隐式奖励模型(LRM),对隐思维过程进行概率优化。跨多种推理任务的实验表明,LRM在检测错误隐思维模式方面效果显著,且可在不同领域间泛化;LTO可无缝应用于通用大模型以改进其推理过程。相比显式语言思维,本方法证明了在隐空间中进行奖励建模与测试时思维扩展是可行的,展现了其作为通用、高效、领域无关的推理优化方案的潜力。

原文摘要 · Abstract (English)

Large Language Models (LLMs) excel at problem solving by generating chain of thoughts in natural language, but such verbal thinking is computationally costly and prone to overthinking. A recent work instead proposes a latent thinking architecture, Huginn-3.5B, which represents intermediate reasoning steps as a sequence of latent representations. However, latent thoughts lack interpretability and are difficult to supervise, raising concerns about the correctness and reliability of the model's latent thinking processes. In this paper, we provide a systematic study of how Huginn-3.5B thinks in the latent space and how external supervision signals can improve its latent thinking processes. We show that latent thoughts leading to correct versus incorrect answers exhibit highly distinguishable patterns, and that a latent classifier can reliably predict answer correctness directly from latent thoughts. Leveraging these insights, we propose Latent Thinking Optimization (LTO), a probabilistic algorithm that employs the latent classifier as a Latent Reward Model (LRM) to optimize the latent thinking processes. Extensive experiments across diverse reasoning tasks demonstrate that LRM is highly effective in detecting incorrect latent thinking patterns, and LTO can significantly improve the latent thinking processes. Furthermore, we show that LRM can generalize across diverse domains, and LTO can be seamlessly applied to general LLMs to improve their thinking processes. In contrast to verbal thinking, our method demonstrates that reward modeling and scaling test-time thinking with supervision can be performed directly in the latent space, highlighting its potential as a general, efficient, and domain-agnostic approach to improving the thinking processes of LLMs.

大模型推理隐空间优化奖励建模思维链

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