arXiv:2412.15904cs.AIcs.LG2024-12AAAI被引 25

研究发现,数学推理的步骤奖励模型主要依赖逻辑结构而非语言描述。

What Are Step-Level Reward Models Rewarding? Counterintuitive Findings from MCTS-Boosted Mathematical Reasoning

  • 用MCTS自动标注推理步骤偏好,无需语言描述仍有效
  • 奖励模型擅长判断数学语言的逻辑连贯性,对自然语言效果差
  • 适合想优化数学推理模型的开发者参考

步骤级奖励模型(SRMs)可通过过程监督或基于强化学习的步骤级偏好对齐显著提升数学推理能力。其性能至关重要,因为它们为推理每一步提供关键指导。近期,采用蒙特卡洛树搜索(MCTS)进行自动步骤级偏好标注的AlphaZero类方法表现尤为出色。然而,SRMs成功背后的机制仍不明确。本研究深入探讨了基于MCTS的SRMs的反直觉特性。结果表明,去除思维过程的语言描述对SRMs有效性影响极小;同时,SRMs能有效评估数学语言中的复杂逻辑连贯性,但在自然语言理解上表现不佳。这些发现揭示了数学推理中有效步骤奖励建模的核心要素,为构建更高效、简洁的SRMs提供了重要指导,只需聚焦数学推理的关键部分即可。

原文摘要 · Abstract (English)

Step-level reward models (SRMs) can significantly enhance mathematical reasoning performance through process supervision or step-level preference alignment based on reinforcement learning. The performance of SRMs is pivotal, as they serve as critical guidelines, ensuring that each step in the reasoning process is aligned with desired outcomes. Recently, AlphaZero-like methods, where Monte Carlo Tree Search (MCTS) is employed for automatic step-level preference annotation, have proven particularly effective. However, the precise mechanisms behind the success of SRMs remain largely unexplored. To address this gap, this study delves into the counterintuitive aspects of SRMs, particularly focusing on MCTS-based approaches. Our findings reveal that the removal of natural language descriptions of thought processes has minimal impact on the efficacy of SRMs. Furthermore, we demonstrate that SRMs are adept at assessing the complex logical coherence present in mathematical language while having difficulty in natural language. These insights provide a nuanced understanding of the core elements that drive effective step-level reward modeling in mathematical reasoning. By shedding light on these mechanisms, this study offers valuable guidance for developing more efficient and streamlined SRMs, which can be achieved by focusing on the crucial parts of mathematical reasoning.

数学推理奖励模型MCTS逻辑一致性

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