arXiv:2410.16128cs.AIcs.LG2024-10被引 4

让大模型一次就选对推理策略,不用反复试错。

SMART: Self-learning Meta-strategy Agent for Reasoning Tasks

  • 用强化学习让模型自己学最优推理策略
  • GSM8K上准确率提升15点,仅需一次推理
  • 适合追求高效精准推理的场景

需要多步演绎推理的任务往往需要自适应策略,如生成中间推理过程或程序,因为单一方法并非始终最优。尽管语言模型可通过迭代自精炼和策略调整提升输出,但常在首次尝试中未能采用最有效策略。这引发一个问题:能否让语言模型在首次尝试中就选出最优策略,无需后续修正?为此,我们提出SMART(Self-learning Meta-strategy Agent for Reasoning Tasks),一种新型框架,使语言模型能自主学习并选择适用于各类推理任务的最佳策略。我们将策略选择建模为马尔可夫决策过程,并利用强化学习驱动的持续自改进机制,让模型找到解决给定任务的合适策略。与依赖多次推理或外部反馈的传统自精炼方法不同,SMART使模型能内化自身推理结果并相应调整策略,力求首次推理即得正确答案。我们在多个推理数据集及不同模型架构上实验表明,SMART显著提升了模型在无外部指导下的最优策略选择能力,在GSM8K数据集上准确率提升15点。通过单次推理实现更高准确率,SMART不仅提升性能,还降低基于精炼策略的计算成本,为语言模型的更高效、智能推理铺平道路。

原文摘要 · Abstract (English)

Tasks requiring deductive reasoning, especially those involving multiple steps, often demand adaptive strategies such as intermediate generation of rationales or programs, as no single approach is universally optimal. While Language Models (LMs) can enhance their outputs through iterative self-refinement and strategy adjustments, they frequently fail to apply the most effective strategy in their first attempt. This inefficiency raises the question: Can LMs learn to select the optimal strategy in the first attempt, without a need for refinement? To address this challenge, we introduce SMART (Self-learning Meta-strategy Agent for Reasoning Tasks), a novel framework that enables LMs to autonomously learn and select the most effective strategies for various reasoning tasks. We model the strategy selection process as a Markov Decision Process and leverage reinforcement learning-driven continuous self-improvement to allow the model to find the suitable strategy to solve a given task. Unlike traditional self-refinement methods that rely on multiple inference passes or external feedback, SMART allows an LM to internalize the outcomes of its own reasoning processes and adjust its strategy accordingly, aiming for correct solutions on the first attempt. Our experiments across various reasoning datasets and with different model architectures demonstrate that SMART significantly enhances the ability of models to choose optimal strategies without external guidance (+15 points on the GSM8K dataset). By achieving higher accuracy with a single inference pass, SMART not only improves performance but also reduces computational costs for refinement-based strategies, paving the way for more efficient and intelligent reasoning in LMs.

推理优化强化学习自适应策略

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