论文精选72°

Entropy-Cut MH:用熵识别推理决策点,采样提升推理能力

Reasoning with Sampling: Cutting at Decision Points

精选理由

这项研究为无需 RL 训练即可提升模型推理能力提供了实用采样方法,做推理模型优化或采样策略的开发者可以直接尝试 Entropy-Cut MH,它比均匀截断更高效。

AI 摘要

前沿推理模型通常通过强化学习后训练得到。近期研究挑战了这一范式,表明从基础模型的幂分布中采样即可获得可比推理能力,无需额外训练。但高效采样是关键挑战。本文提出 Entropy-Cut Metropolis-Hastings 算法,利用模型下一 token 的熵作为代理,识别推理轨迹中的关键决策点(如证明策略或算法选择),并仅从这些位置重新采样,而非均匀随机截断。理论证明该方法混合时间与决策点数量成正比,而非 token 数量。在 MATH500、HumanEval、GPQA Diamond 和 AIME26 基准上,该方法一致优于基线及强化学习训练模型。

原文 · arXiv cs.AI

Reasoning with Sampling: Cutting at Decision Points

Frontier reasoning models are produced by posttraining base language models with reinforcement learning. Recent work has challenged this by showing that sampling from a sharpened version of the base model's distribution, a so-called power distribution, elicits comparable reasoning without additional training, curated datasets, or verifiers. However, making this method practical requires efficiently sampling from the power distribution. A sampler needs to "mix" to the power distribution, which necessitates moving between modes of the target distribution; intuitively, e.g., trying different reasoning strategies. The samplers proposed in prior works repeatedly select a "cut" position in the current reasoning trace uniformly at random and resample the suffix from that position onward. However, reasoning traces typically contain a few consequential decisions (e.g., the choice of proof strategy or algorithm), and we observe that a uniformly chosen cut tends to rewrite local details rather than revisit decision points. We introduce an algorithm (Entropy-Cut Metropolis-Hastings) that uses the base model's next-token entropy as a proxy to identify key decision points and resample from those positions. We empirically verify that entropy jumps are a useful proxy for decision points and, in a stylized model of reasoning, prove that our method's mixing time scales with the number of decisions in a trace rather than with the number of tokens, which can be much larger. Across MATH500, HumanEval, GPQA Diamond, and AIME26, our method consistently improves over baselines and RL-trained models.