Ensembling Large Language Models with Process Reward-Guided Tree Search for Better Complex Reasoning

TL;DR

LE-MCTS以PRM引导多模型MCTS,在MATH达45.2%、MQA达71.1%。

cs.CL 🔴 高级 2024-12-20 29 次浏览
Sungjin Park Xiao Liu Yeyun Gong Edward Choi
大语言模型集成 蒙特卡洛树搜索 过程奖励模型 复杂数学推理 强化搜索

核心发现

方法论

LE-MCTS将逐步推理建模为MDP:状态是已有推理路径,动作是从LLaMA-3、Gemma-2、DeepSeek-Math或Rho-Math中选择模型生成下一行。算法采用UCT选择、贪心扩展、Math Shepherd PRM评分,并以乐观回传用子节点最大价值更新路径。

关键结果

  • 在GSM8K、MATH500、SVAMP、ASDiv和MQA上,Top-3 LE-MCTS平均准确率为73.8%,比第二名BoE的69.9%高3.9个百分点;MATH为45.2%,较第二名提高3.6个百分点。
  • LE-MCTS在MQA达到71.1%,较第二名提高4.3个百分点;在SVAMP达到84.0%,与最佳方法基本持平,并在GSM8K和ASDiv分别达到84.1%和84.4%。
  • 乐观回传在五个数据集上提升0.1–1.6%;UCT常数C=0.5更适合MATH/MQA,C=1.0或1.414更适合简单数据集;迭代次数从10增至200总体改善性能。

研究意义

论文把模型集成从“拼接概率”或“比较完整答案”推进到“搜索推理过程”。这解决了单一开源模型能力不均、完整候选全部出错以及不同词表无法对齐等问题。结果表明,模型多样性只有在步骤级搜索和可靠过程评分共同作用下,才可转化为复杂推理能力。

技术贡献

核心技术是跨模型统一推理树:不同架构只需生成自然语言步骤,无需共享词表或参数。UCT公式平衡探索与利用;扩展阶段按换行切分步骤;PRM直接评估中间步骤,避免昂贵rollout;乐观回传使用子节点最大值,符合“只需一条可行后继路径”的集成目标。

新颖性

LE-MCTS提出了面向复杂推理的过程级语言模型集成框架,将多个模型的局部推理步骤放入同一MCTS树中。据论文定位,它区别于BoE、EBS、LLM-Blender、MoA和EVA,不是在答案层融合,而是以PRM驱动跨模型的逐步组合与前瞻搜索。

局限性

  • 200次迭代在MATH平均每题342.2分钟,明显高于EBS的47.2分钟,性能提升伴随巨大推理成本。
  • 简单任务中深度搜索收益有限;ASDiv上BoE/EBS更高效,说明固定使用LE-MCTS可能造成资源浪费。
  • 效果依赖Math Shepherd PRM的校准质量;PRM错误可能把搜索引向表面合理但最终错误的路径。

未来方向

未来可研究自适应迭代次数和UCT常数、按题目难度动态选择模型,并改进PRM校准与多步信用分配。还可探索并行树搜索、缓存共享、模型专长路由及非数学任务上的过程级集成。

AI 总览摘要

复杂数学推理要求模型在多个中间步骤上持续正确,而不是只生成一个看似合理的答案。论文指出,传统token级集成依赖词表和架构匹配,输出级方法如LLM-Blender、MoA只能在完整答案间选择或融合;若所有候选都错,集成也无能为力。

作者提出LE-MCTS(Language model Ensemble with Monte Carlo Tree Search),把每条推理链视为树路径。节点记录中间步骤,动作是在预定义模型池中选择一个模型生成下一步;UCT负责探索与利用,Math Shepherd的过程奖励模型逐步打分,乐观回传则沿最有希望的子路径传播价值。这样,一条最终解答可以由多个模型接力完成。

实验覆盖GSM8K、MATH500、SVAMP、ASDiv和MQA。Top-3配置平均准确率73.8%,超过BoE的69.9%;MATH达到45.2%,提升3.6个百分点,MQA达到71.1%,提升4.3个百分点。代价是计算量较高:MATH上200次迭代平均342.2分钟。因此,LE-MCTS最适合高难度、允许额外推理预算的场景,而简单题可采用BoE或EBS。

深度分析

研究背景

开源模型如LLaMA-3、Gemma-2、DeepSeek-Math和Rho-Math能力互补。既有研究包括token概率融合、EVA词表投影,以及LLM-Blender和MoA等输出融合;CoT、MCTS和PRM则推动了过程级推理。但这些方向尚未有效结合模型多样性与逐步搜索。

核心问题

给定问题q和模型集合{π1,…,πL},第k步可由任一模型生成。所有组合数量为∏k|Pk|,随模型数和推理长度指数增长。完整答案排序无法修复共同错误,token融合又受词表、维度和架构限制,因此需要可扩展的步骤级搜索。

核心创新

  • ��将跨模型推理形式化为MDP,状态为中间轨迹、动作为模型选择和下一步生成。

  • ��用MCTS在统一自然语言推理树中搜索,而非要求参数或词表兼容。

  • ��用PRMϕ(q,pk)直接评分每一步,并提出乐观回传,以最大子节点价值代表父节点,突出至少一条可行路径。

方法详解

  • ��初始化:根节点为问题q,每个节点保存轨迹、价值vs和访问次数Ns。

  • ��选择:对非终止子节点计算UCT,U(s)=vs+C√(ln Nparent/Ns),选择最大者。

  • ��扩展:随机选模型πl,按pk,t=argmaxw πl(w|pk,<t;q,p1:k−1)生成至换行;最多nchild个孩子,并用ε规则鼓励深入。

  • ��评估:Math Shepherd PRM输出rk=ϕ(q,pk),不做rollout;作者称ORM会使时间增加约5–10倍。

  • ��回传:Ns加一,以max子节点价值更新vs;200次迭代后,对终止轨迹按PRM排序并输出最高者。

实验设计

数据集为GSM8K、MATH500、SVAMP、ASDiv、MQA,指标为准确率。基线包括Greedy、Self-Consistency、Beam Search、Best-of-N、BoE、EBS、LLM-Blender、MoA和EVA。模型池含LLaMA-3 8B、Gemma-2 9B、DeepSeek-Math 7B、Rho-Math 7B;默认niter=200,并消融C、回传策略和迭代次数。

结果分析

Top-3 LE-MCTS在五项任务平均73.8%,高于BoE 69.9%。各项为GSM8K 84.1%、MATH 45.2%、SVAMP 84.0%、ASDiv 84.4%、MQA 71.1%。乐观回传提升0.1–1.6%;niter=10到200总体上升。效率上,MATH使用200次迭代需342.2分钟,显示准确率与成本的明确权衡。

应用场景

可用于竞赛数学、GRE/GMAT解题、定理推导、代码规划和需要多步验证的代理系统。部署前应准备多个能力互补的模型、可靠PRM及足够GPU预算;简单任务可优先使用BoE/EBS以降低延迟。

局限与展望

LE-MCTS假设推理能自然切分为换行步骤,并依赖PRM准确评估局部正确性;错误评分会系统性误导树搜索。200次搜索资源昂贵,且论文只评估数学数据集。未来需要难度自适应搜索、并行与缓存优化、更强PRM,以及在科学、代码和开放式规划上的验证。

通俗解读 非专业人士也能看懂

把解题想成一支厨房团队做复杂菜。传统方法要么把几位厨师的每个动作混在一起,要么等每人做完整菜后再选最好的一盘;如果大家都把菜做坏了,就没有补救机会。LE-MCTS让厨师轮流接手:先看当前菜做到哪一步,再让不同厨师提出下一步,并由一位检查员判断这一步是否靠谱。搜索系统会保留很多可能路线,更多尝试看起来有希望的路线,同时偶尔检查尚未尝试的路线。最后,它选择检查员认为最好的完整做法。优点是某位厨师前半段好、另一位后半段强时,可以组合他们的长处;缺点是要反复试做,复杂菜尤其耗时。

简单解释 像给14岁少年讲一样

想象你和四个很会做数学题的同学组队打游戏。一个同学擅长基础题,另一个擅长竞赛题,但谁都有可能在某一步翻车。普通做法是让每个人独立写完答案,再投票;如果大家都错在同一个地方,投票也救不了。

LE-MCTS像一张闯关地图:每走一步就留下一个分叉。系统可以让不同同学接力完成同一道题,而不是强迫一个人从头做到尾。每个分叉都会由“步骤检查员”打分,判断这一步是否合理;搜索器会更常走高分路线,也会偶尔探索没走过的路线。

它还使用一种聪明的记分方式:只要某条后续路线特别好,前面的路就仍然值得保留,不会因为其他分叉很差而被平均拉低。这很像游戏里一条隐藏通关路线足够强,就继续投资它。

结果很亮眼:MATH达到45.2%,MQA达到71.1%,分别比第二名高3.6和4.3个百分点。不过它像反复练习副本,200次搜索在MATH平均要342.2分钟。所以难题值得用,简单题则应选择更快的方法。

术语表

Monte Carlo Tree Search (蒙特卡洛树搜索)

一种通过反复选择、扩展、评估和回传来搜索决策树的方法。它在有限预算下优先探索高价值路径。

LE-MCTS用它搜索不同模型生成的推理步骤。

Process Reward Model (过程奖励模型)

逐步判断推理过程质量的模型,而非只判断最终答案。论文使用Math Shepherd PRM计算ϕ(q,pk)。

它为每个新推理步骤提供搜索奖励。

UCT

结合当前价值与探索奖励的树搜索准则,公式为vs+C√(ln Nparent/Ns)。C控制探索和利用的平衡。

LE-MCTS在选择阶段用UCT挑选子节点。

Optimistic Backpropagation (乐观回传)

用子节点最大价值更新父节点,而不是平均所有子节点奖励。它保留“至少一条成功后继路径”的可能性。

该策略使搜索聚焦高潜力推理链。

Process-level Ensembling (过程级集成)

在中间推理步骤层面组合多个模型,而不是融合token概率或完整答案。它允许不同模型接力完成一条链。

这是LE-MCTS区别于EVA、MoA和LLM-Blender的核心。

开放问题 这项研究留下的未解疑问

  • 1 PRM在长链、含隐蔽错误或跨领域推理中的校准可靠性仍不清楚;需要人工标注和更细粒度误差分析。
  • 2 模型选择目前主要随机进行,尚未回答如何依据题型、历史步骤和模型专长动态路由。

应用场景

近期应用

竞赛数学辅助解题

教育平台可组合多个开源数学模型,用Math Shepherd PRM逐步检查并搜索解题链。适合MATH、GRE和GMAT类难题,但需配置GPU预算和答案验证器。

多模型推理代理

在代码规划、工具调用或科学问答中,把每个计划步骤作为树节点,由不同专长模型提出候选,再由过程检查器筛选,降低一次性生成完整计划的风险。

远期愿景

通用可靠推理系统

未来可建立跨数学、代码、科学和规划的过程奖励模型,并让搜索预算随题目难度自动变化,使高风险任务获得深度验证、普通任务保持低延迟。

原文摘要

Despite recent advances in large language models, open-source models often struggle to consistently perform well on complex reasoning tasks. Existing ensemble methods, whether applied at the token or output levels, fail to address these challenges. In response, we present Language model Ensemble with Monte Carlo Tree Search (LE-MCTS), a novel framework for process-level ensembling of language models. LE-MCTS formulates step-by-step reasoning with an ensemble of language models as a Markov decision process. In this framework, states represent intermediate reasoning paths, while actions consist of generating the next reasoning step using one of the language models selected from a predefined pool. Guided by a process-based reward model, LE-MCTS performs a tree search over the reasoning steps generated by different language models, identifying the most accurate reasoning chain. Experimental results on five mathematical reasoning benchmarks demonstrate that our approach outperforms both single language model decoding algorithms and language model ensemble methods. Notably, LE-MCTS improves performance by 3.6% and 4.3% on the MATH and MQA datasets, respectively, highlighting its effectiveness in solving complex reasoning problems.

cs.CL