首页 最新 热门 推荐

  • 首页
  • 最新
  • 热门
  • 推荐

字节ReTool:大模型也要学会善于利用工具

  • 25-04-24 05:41
  • 2577
  • 8780
juejin.cn

导语

  • 标题:ReTool: Reinforcement Learning for Strategic Tool Use in LLMs
  • 链接:arxiv.org/pdf/2504.11…

摘要

尽管强化学习训练的推理模型在纯文本推理任务中表现突出,但在需要精确计算或符号操作的结构化问题上仍显不足。为此,本文提出 ReTool 框架,通过将实时代码执行与自然语言推理交叉集成,并采用结果驱动的强化学习策略,让模型自主学习何时、如何调用计算工具。训练过程分为两阶段:一是利用合成数据进行代码增强的监督微调,二是在沙箱环境中以任务正确性为奖励,迭代优化工具使用策略。在国际数学竞赛基准 AIME 上的实验显示,ReTool 在训练效率和最终准确率上均大幅领先于纯文本强化学习和多种竞争基线,并在模型中观察到诸如代码自我修正等新兴行为,表明该方法在复杂数学推理场景中具有显著潜力。

image.png

引言

本文探讨如何将工具调用集成到大型语言模型(LLM)中,通过强化学习(RL)训练让模型在推理过程中能灵活调用外部工具,从而弥补传统文本推理方法的不足。具体来说,ReTool是一个新型的强化学习框架,通过集成代码解释器,能够提升模型在处理复杂数学推理任务时的表现,尤其是在那些需要精确计算和符号操作的任务中。

image.png

核心创新点

ReTool的核心创新点在于其通过强化学习,训练大型语言模型在推理过程中智能地调用外部工具(如代码解释器)。传统的语言模型依赖文本推理来解题,但在面对需要高度精确计算或复杂方程求解时,纯文本推理往往会产生错误或无法覆盖所有解空间。而通过ReTool,模型能够实时生成代码并与外部计算工具交互,帮助其验证中间步骤或进行精确计算,从而弥补传统方法的不足。

具体来说,ReTool有两个关键特性:

  1. 动态交错的实时代码执行:模型在推理过程中可以实时生成并执行代码,通过反馈结果来调整推理方向。
  2. 自动化的RL训练范式:模型通过强化学习的反馈机制,学习何时以及如何调用外部工具,并在多轮推理中优化工具的使用策略。

这种设计使得模型能够逐步掌握何时调用工具、如何选择合适的工具,并且在推理过程中进行自我修正,最终实现更高效、更精确的推理过程。

研究方法和思路

ReTool的设计分为两个主要阶段:

  1. 冷启动SFT阶段:通过构建一个高质量的数据集,训练模型理解何时和如何调用工具。这个数据集由多来源的数学推理任务组成,经过专家审核和筛选,确保每个示例都能提供准确的推理路径和工具调用信息。
  2. 强化学习阶段:在此阶段,模型通过与代码解释器的交互进行强化学习,探索最佳的推理策略。模型通过动态生成代码、调用外部工具、获取执行结果,并根据反馈优化策略。这一过程帮助模型逐渐学会在复杂任务中应用工具。

训练过程

  • 训练算法:PPO算法

image.png

  • 奖励设置:使用正确性评估奖励,从指定格式\boxed{}中提取最终答案

image.png

实验结果

在挑战性的AIME奥赛基准上,ReTool的表现令人惊艳。经过仅400步训练,ReTool在AIME2024测试集上达到了67%的准确率,这一结果大大超过了传统文本型RL方法(仅40%的准确率,训练步数超过1000步)。此外,ReTool在AIME2025上也展现了出色的性能,达到了49.3%的准确率,远超其他模型。

image.png

更重要的是,ReTool在训练过程中展示了新的行为模式。例如,模型能够自我修正代码错误并通过迭代优化推理过程。这种自适应的工具使用和修正能力,为未来更高效的推理模式奠定了基础。

对比与分析

ReTool的最大优势在于其灵活的工具使用策略。在传统的文本型RL方法中,模型只能依赖于静态的文本推理,很难应对需要精确计算或复杂符号操作的任务。而ReTool通过引入工具集成,能够动态生成代码、与外部计算工具交互,极大地提高了推理的准确性和效率。

image.png

本文对训练过程中的一些指标进行了详细分析,具体如下:

  • Token效率(图3a):训练后平均响应长度减少约 40%,表明代码介入可精简冗长计算。
  • 代码使用趋势(图3b):随着训练推进,代码调用比例和代码行数持续上升,且调用时机逐渐提前。

image.png

  • 自我修正能力(图4):在遇到执行错误时,模型能够根据反馈识别问题并迭代生成正确代码。

image.png

  • 用途演化(图5):最初代码主要用于简单计算与验证,后期用途更趋多样,涵盖复杂逻辑与自适应工具选择。
  • 案例对比(图6):典型题目展示了纯文本推理易出错、过程冗长,而代码驱动推理则以简洁且精确的方式完成计算,进一步支持整体策略判断。

image.png

总结

ReTool通过创新性地将代码执行与强化学习结合,为大型语言模型的推理能力提供了新的提升路径。通过将计算工具集成到推理过程中,ReTool显著提高了模型在数学推理任务中的表现,并为未来更高效的工具增强型推理开辟了新的方向。随着研究的深入,ReTool可能会在更广泛的应用领域中展现其强大的潜力,为解决复杂问题提供更多的智能解决方案。

注:本文转载自juejin.cn的Q同学的文章"https://juejin.cn/post/7495420228388241448"。版权归原作者所有,此博客不拥有其著作权,亦不承担相应法律责任。如有侵权,请联系我们删除。
复制链接
复制链接
相关推荐
发表评论
登录后才能发表评论和回复 注册

/ 登录

评论记录:

未查询到任何数据!
回复评论:

分类栏目

后端 (14832) 前端 (14280) 移动开发 (3760) 编程语言 (3851) Java (3904) Python (3298) 人工智能 (10119) AIGC (2810) 大数据 (3499) 数据库 (3945) 数据结构与算法 (3757) 音视频 (2669) 云原生 (3145) 云平台 (2965) 前沿技术 (2993) 开源 (2160) 小程序 (2860) 运维 (2533) 服务器 (2698) 操作系统 (2325) 硬件开发 (2492) 嵌入式 (2955) 微软技术 (2769) 软件工程 (2056) 测试 (2865) 网络空间安全 (2948) 网络与通信 (2797) 用户体验设计 (2592) 学习和成长 (2593) 搜索 (2744) 开发工具 (7108) 游戏 (2829) HarmonyOS (2935) 区块链 (2782) 数学 (3112) 3C硬件 (2759) 资讯 (2909) Android (4709) iOS (1850) 代码人生 (3043) 阅读 (2841)

热门文章

109
人工智能
关于我们 隐私政策 免责声明 联系我们
Copyright © 2020-2025 蚁人论坛 (iYenn.com) All Rights Reserved.
Scroll to Top