arXiv:2602.06584cs.CLcs.LG2026-02

让AI在推理时自我修正,用潜空间向量提升数学解题能力

Inference-Time Rethinking with Latent Thought Vectors for Math Reasoning

  • 将思维过程拆分为潜空间向量和生成解码器,实现可优化的推理结构
  • 30次重思迭代下,0.2B模型超越10-15倍参数量的基线模型
  • 适合需要高精度数学推理的场景,尤其关注推理机制而非参数规模

标准链式思维推理在单次前向传播中确定每个词元,无法纠正早期错误。本文提出推理时重思(Inference-Time Rethinking),一种生成框架,通过解耦声明性潜空间思维向量与过程性生成,实现迭代自校正。将推理分解为连续潜空间思维向量(思考什么)和基于该向量条件生成推理轨迹的解码器(如何思考)。潜空间向量作为声明性缓冲区,将推理结构压缩为连续表示,抽象掉表面词元差异,使基于梯度的推理策略优化成为可能。先验模型将无结构噪声映射到学习到的有效推理模式流形;测试时采用类似Gibbs的交替过程:生成候选推理轨迹并优化潜空间向量以更好解释该轨迹,从而在潜空间流形上导航,精炼推理策略。在GSM8K数据集上从头训练0.2B参数模型,经30次重思迭代后,性能超过拥有10至15倍参数的基线模型,包括一个3B模型。结果表明,有效的数学推理可源于复杂的推理时计算,而不仅依赖庞大的参数量。

原文摘要 · Abstract (English)

Standard chain-of-thought reasoning generates a solution in a single forward pass, committing irrevocably to each token and lacking a mechanism to recover from early errors. We introduce Inference-Time Rethinking, a generative framework that enables iterative self-correction by decoupling declarative latent thought vectors from procedural generation. We factorize reasoning into a continuous latent thought vector (what to reason about) and a decoder that verbalizes the trace conditioned on this vector (how to reason). Beyond serving as a declarative buffer, latent thought vectors compress the reasoning structure into a continuous representation that abstracts away surface-level token variability, making gradient-based optimization over reasoning strategies well-posed. Our prior model maps unstructured noise to a learned manifold of valid reasoning patterns, and at test time we employ a Gibbs-style procedure that alternates between generating a candidate trace and optimizing the latent vector to better explain that trace, effectively navigating the latent manifold to refine the reasoning strategy. Training a 0.2B-parameter model from scratch on GSM8K, our method with 30 rethinking iterations surpasses baselines with 10 to 15 times more parameters, including a 3B counterpart. This result demonstrates that effective mathematical reasoning can emerge from sophisticated inference-time computation rather than solely from massive parameter counts.

数学推理推理优化潜空间自修正

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