arXiv:2409.02392cs.LGstat.ML2024-09ICLR被引 68

用多轮偏好学习提升数学智能体解题能力,效果显著

Building Math Agents with Multi-Turn Iterative Preference Learning

  • 设计多轮直接偏好学习框架,融合代码执行反馈优化推理过程
  • 在GSM8K和MATH数据集上,模型准确率最高提升7.4个百分点
  • 适合研究数学推理、工具增强型大模型的开发者参考

近期研究表明,通过集成外部工具(如代码解释器)并采用多轮思维链(CoT)推理,可提升大语言模型的数学求解能力。现有方法多依赖合成数据生成与监督微调(SFT),本文探索了更直接的偏好学习路径。然而,现有偏好学习算法原为单轮对话设计,难以应对工具集成数学推理中的多轮复杂性。为此,本文提出适配该场景的多轮直接偏好学习框架,利用代码解释器反馈优化轨迹级偏好,包含多轮DPO与多轮KTO两种实现。通过在GSM8K和MATH数据集增强提示集上训练多种语言模型,验证了框架有效性:监督微调的Gemma-1.1-it-7B模型在GSM8K上从77.5%提升至83.9%,MATH从46.1%升至51.2%;Gemma-2-it-9B模型在GSM8K上从84.1%升至86.3%,在MATH上从51.0%升至54.5%。

原文摘要 · Abstract (English)

Recent studies have shown that large language models' (LLMs) mathematical problem-solving capabilities can be enhanced by integrating external tools, such as code interpreters, and employing multi-turn Chain-of-Thought (CoT) reasoning. While current methods focus on synthetic data generation and Supervised Fine-Tuning (SFT), this paper studies the complementary direct preference learning approach to further improve model performance. However, existing direct preference learning algorithms are originally designed for the single-turn chat task, and do not fully address the complexities of multi-turn reasoning and external tool integration required for tool-integrated mathematical reasoning tasks. To fill in this gap, we introduce a multi-turn direct preference learning framework, tailored for this context, that leverages feedback from code interpreters and optimizes trajectory-level preferences. This framework includes multi-turn DPO and multi-turn KTO as specific implementations. The effectiveness of our framework is validated through training of various language models using an augmented prompt set from the GSM8K and MATH datasets. Our results demonstrate substantial improvements: a supervised fine-tuned Gemma-1.1-it-7B model's performance increased from 77.5% to 83.9% on GSM8K and from 46.1% to 51.2% on MATH. Similarly, a Gemma-2-it-9B model improved from 84.1% to 86.3% on GSM8K and from 51.0% to 54.5% on MATH.

数学推理多轮推理偏好学习工具集成

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