将离散推理的采样重排技术移植到连续空间,发现效果受限于思维表示缺陷。
Towards Inference-time Scaling for Continuous Space Reasoning
- 通过丢弃采样生成多样化连续推理路径
- Pass@N分析显示性能提升潜力与离散空间相当
- 现有重排模型在连续空间改进有限,因缺乏判别性归纳偏置
通过多样本生成结合过程或结果奖励模型(PRM/ORM)重排,已在大语言模型的文本推理中证明有效。本文探究该方法能否成功适配至连续空间推理,以COCONUT(Hao et al. 2024)为基底模型。我们证明了基于丢弃采样的多样化推理路径生成可行性。对生成样本的Pass@N分析表明,连续空间存在类似离散空间的性能增益潜力。然而,我们指出在连续思维空间实现该增益面临独特挑战:在离散空间有效的数据生成与训练策略,在连续空间仅带来微弱改进。通过分析几何特性与轨迹动态,我们发现当前问题根源在于连续思维表征缺乏区分正确与错误推理的关键归纳偏置。因此,我们认为连续推理模型的训练框架不仅需优化准确率,还应显式引入可用于推理时判别的归纳偏置。
原文摘要 · Abstract (English)
Inference-time scaling through multiple sample generation in combination with Process- or Outcome-Reward Model (PRM or ORM) re-ranking has proven effective for text-based reasoning in large language models. This paper investigates whether such established techniques can be successfully adapted to reasoning in the continuous space, using COCONUT (Hao et al. 2024) continuous space reasoning LM as the backbone. We demonstrate the feasibility of generating diverse reasoning paths through dropout-based sampling. Our Pass@N analysis on the generated samples reveals the potential that could enable a significant gain in performance akin to observed gain in the discrete space. However, we highlight unique challenges faced for materializing this gain in the continuous thought space. In particular, working recipes for data generation and training PRM and ORM models in the discrete space unlocks only marginal improvements in the continuous space. Through probing various aspects including geometric properties and trajectory dynamics we identify the underlying reasons that prevent effective discrimination between correct and incorrect reasoning (essential for the functioning of PRM and ORM). Our findings reveal that current limitations stem from the absence of key inductive biases in continuous thought representations. We argue that the training frameworks for continuous reasoning LMs require not only to optimize for accuracy but also to explicitly incorporate inductive biases that could be utilized during inference-time for discrimination of correct and incorrect thoughts.\footnote{Our code and data will be publicly available.}
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。