通过马尔可夫过程建模思维链,发现搜索能显著提升推理效率。
Metastable Dynamics of Chain-of-Thought Reasoning: Provable Benefits of Search, RL and Distillation
- 将思维链视为具有稳定与非稳定状态的马尔可夫过程,区分易难推理步骤。
- 搜索奖励稀疏连接可减少到达不同推理状态的平均步数,提升推理速度。
- 结果可用于微调模型或压缩为更高效的小模型,适合优化推理系统。
提升大语言模型推理能力的关键方法是在推理时分配更多计算资源,对验证器或奖励模型进行搜索。该过程可用于优化预训练模型或将其推理模式提炼为更高效的模型。本文将思维链生成视为一种亚稳态马尔可夫过程:简单推理步骤(如代数运算)形成密集连接的簇,而复杂推理步骤(如应用相关定理)则在簇间形成稀疏、低概率的边,导致长时尺度下的相变现象。在此框架下,我们证明了采用奖励稀疏边的搜索协议能降低到达不同簇的期望步数。相反,若仅依赖预训练图的局部信息,则推理能力存在上限。此外,我们展示搜索所得信息可用于改进模型:(1) 可通过策略梯度方法直接微调预训练模型以偏好稀疏边;(2) 可将推理动态的压缩亚稳态表示蒸馏至更小、更高效的模型中。
原文摘要 · Abstract (English)
A key paradigm to improve the reasoning capabilities of large language models (LLMs) is to allocate more inference-time compute to search against a verifier or reward model. This process can then be utilized to refine the pretrained model or distill its reasoning patterns into more efficient models. In this paper, we study inference-time compute by viewing chain-of-thought (CoT) generation as a metastable Markov process: easy reasoning steps (e.g., algebraic manipulations) form densely connected clusters, while hard reasoning steps (e.g., applying a relevant theorem) create sparse, low-probability edges between clusters, leading to phase transitions at longer timescales. Under this framework, we prove that implementing a search protocol that rewards sparse edges improves CoT by decreasing the expected number of steps to reach different clusters. In contrast, we establish a limit on reasoning capability when the model is restricted to local information of the pretrained graph. We also show that the information gained by search can be utilized to obtain a better reasoning model: (1) the pretrained model can be directly finetuned to favor sparse edges via policy gradient methods, and moreover (2) a compressed metastable representation of the reasoning dynamics can be distilled into a smaller, more efficient model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。