arXiv:2502.11183cs.CL2025-02ACL被引 15

优化大模型推理搜索,提升效率与准确率。

Don't Get Lost in the Trees: Streamlining LLM Reasoning by Overcoming Tree Search Exploration Pitfalls

  • 用语义聚类合并重复推理路径,减少冗余计算。
  • 改进验证器降低评分波动,减少无效路径切换。
  • 兼容多种搜索算法,适合追求高效推理的开发者。

近期基于验证器引导的树搜索算法显著提升了大语言模型的推理能力,但代价是计算资源消耗大幅增加。本文识别出两大效率瓶颈:因语义重复状态导致的过度探索,以及验证器评分方差大引发的探索不足。为此,提出FETCH框架,一种灵活可插拔的树搜索优化系统。通过微调SimCSE模型获取文本嵌入,并采用凝聚聚类合并语义相似状态,有效缓解过度探索;在训练中引入时序差分学习并调整λ-回报,降低验证器评分方差,并在推理阶段使用验证器集成聚合分数,以应对探索不足问题。在GSM8K、GSM-Plus和MATH数据集上的实验表明,该方法在四种不同树搜索算法上均显著提升推理准确率与计算效率,为大模型推理的实际应用铺平道路。代码已公开于https://github.com/Soistesimmer/Fetch。

原文摘要 · Abstract (English)

Recent advancements in tree search algorithms guided by verifiers have significantly enhanced the reasoning capabilities of large language models (LLMs), but at the cost of increased computational resources. In this work, we identify two key challenges contributing to this inefficiency: $\textit{over-exploration}$ due to redundant states with semantically equivalent content, and $\textit{under-exploration}$ caused by high variance in verifier scoring leading to frequent trajectory switching. To address these issues, we propose FETCH, an e$\textbf{f}$fici$\textbf{e}$nt $\textbf{t}$ree sear$\textbf{ch}$ framework, which is a flexible, plug-and-play system compatible with various tree search algorithms. Our framework mitigates over-exploration by merging semantically similar states using agglomerative clustering of text embeddings obtained from a fine-tuned SimCSE model. To tackle under-exploration, we enhance verifiers by incorporating temporal difference learning with adjusted $λ$-returns during training to reduce variance, and employing a verifier ensemble to aggregate scores during inference. Experiments on GSM8K, GSM-Plus, and MATH datasets demonstrate that our methods significantly improve reasoning accuracy and computational efficiency across four different tree search algorithms, paving the way for more practical applications of LLM-based reasoning. The code is available at https://github.com/Soistesimmer/Fetch.

大模型推理树搜索效率优化

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