arXiv:2507.05557cs.CL2025-07被引 7

用分层检索增强大模型推理,提升测试时计算资源的利用效率。

Enhancing Test-Time Scaling of Large Language Models with Hierarchical Retrieval-Augmented MCTS

  • 分两层检索:先找问题模板,再找解题步骤,辅助推理
  • 在数学题上比基线最高提升16%准确率
  • 适合需要精准逻辑推理的AI应用开发者

测试时扩展已成为语言建模中一种有前景的方法,通过在推理阶段引入额外计算资源来提升模型性能。本文提出R2-LLMs,一种新型通用的分层检索增强推理框架,用于提升大语言模型(LLM)的测试时扩展能力,无需依赖更先进模型的思维链(CoT)训练数据进行蒸馏。R2-LLMs通过双层检索增强上下文学习:(1) 粗粒度层面,从复杂推理问题中提取抽象模板,并检索相似的问题-答案对,促进高层上下文学习;(2) 细粒度层面,在蒙特卡洛树搜索(MCTS)过程中,高效检索参考数学题数据集中的类似中间解题步骤,结合过程奖励模型(PRM)对解题步骤评分,优化逐步推理。R2-LLMs是一种鲁棒的分层推理增强方法,可在不改变原有结构的前提下,无缝集成到步骤级树搜索中。利用PRM,它同时优化候选生成与决策过程,提升推理准确性。在MATH500、GSM8K和OlympiadBench-TO数据集上的实证评估显示,使用LLaMA-3.1-8B模型时,相对基线最高提升达16%,验证了该方法在复杂推理任务中的有效性。

原文摘要 · Abstract (English)

Test-time scaling has emerged as a promising paradigm in language modeling, leveraging additional computational resources at inference time to enhance model performance. In this work, we introduce R2-LLMs, a novel and versatile hierarchical retrieval-augmented reasoning framework designed to improve test-time scaling in large language models (LLMs) without requiring distillation from more advanced models to obtain chain-of-thought (CoT) training data. R2-LLMs enhances inference-time generalization by integrating dual-level retrieval-based in-context learning: (1) At the coarse level, our approach extracts abstract templates from complex reasoning problems and retrieves similar problem-answer pairs to facilitate high-level in-context learning; (2) At the fine level, during Monte Carlo Tree Search (MCTS), R2-LLMs efficiently retrieves analogous intermediate solution steps from reference mathematical problem datasets, refining step-wise reasoning with the aid of a process reward model (PRM) for scoring. R2-LLMs is a robust hierarchical reasoning-augmentation method that enhances in-context-level reasoning while seamlessly integrating with step-level tree search methods. Utilizing PRM, it refines both candidate generation and decision-making for improved reasoning accuracy. Empirical evaluations on the MATH500, GSM8K, and OlympiadBench-TO datasets achieve substantial relative improvement with an increase of up to 16% using LLaMA-3.1-8B compared to the baselines, showcasing the effectiveness of our approach in complex reasoning tasks.

大模型推理分层检索数学推理MCTS

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