arXiv:2501.05752cs.AIcs.CL2025-01ACL被引 8

用自适应门控减少大模型推理冗余,提升效率与准确率

Semantic Exploration with Adaptive Gating for Efficient Problem Solving with Language Models

  • 根据简单推理的置信度动态决定是否进行深度搜索
  • 在GSM8K等数据集上准确率提升4.3%,计算成本降至31%
  • 适合需要高效复杂推理的应用场景,如教育问答

大语言模型在多步推理任务中展现出巨大潜力,常依赖树搜索探索多种推理路径。但现有方法存在计算效率低、路径重复问题:一方面忽略任务难度差异,对简单任务也做过度搜索;另一方面忽视推理路径语义,导致语义相同路径重复探索。为此,我们提出语义自适应门控探索(SEAG),通过自适应门控机制,依据前置简单推理的置信度动态决定是否启动树搜索。同时,其基于树的探索会合并语义相同的推理步骤,减少冗余。大量实验表明,在GSM8K和ARC等复杂推理基准上,SEAG平均准确率提升4.3%,计算成本仅为现有树搜索方法的31%,支持Llama2、Llama3、Mistral等多种模型。代码已开源。

原文摘要 · Abstract (English)

Recent advancements in large language models (LLMs) have shown remarkable potential in various complex tasks requiring multi-step reasoning methods like tree search to explore diverse reasoning paths. However, existing methods often suffer from computational inefficiency and redundancy. First, they overlook the diversity of task difficulties, leading to unnecessarily extensive searches even for easy tasks. Second, they neglect the semantics of reasoning paths, resulting in redundant exploration of semantically identical paths. To address these limitations, we propose Semantic Exploration with Adaptive Gating (SEAG), a computationally efficient method. SEAG employs an adaptive gating mechanism that dynamically decides whether to conduct a tree search, based on the confidence level of answers from a preceding simple reasoning method. Furthermore, its tree-based exploration consolidates semantically identical reasoning steps, reducing redundant explorations while maintaining or even improving accuracy. Our extensive experiments demonstrate that SEAG significantly improves accuracy by 4.3% on average while requiring only 31% of computational costs compared to existing tree search-based methods on complex reasoning benchmarks including GSM8K and ARC with diverse language models such as Llama2, Llama3, and Mistral. Our code is available at https://github.com/ml-postech/SEAG-semantic-exploration-with-adaptive-gating .

大模型推理语义去重自适应搜索

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