DIVE:以证据驱动的多样性扩展提升工具使用智能体的泛化能力 DIVE: Scaling Diversity in Agentic Task Synthesis for Generalizable Tool Use
证据优先合成智能体任务,扩展多样性以提升工具使用泛化能力
前置知识
工具使用智能体(Tool-Use Agent)
一种基于大语言模型构建的智能体,通过交替执行推理与工具调用来完成开放式任务。在每一步,模型先生成思考(thought),再选择动作(action,通常为函数调用)作用于外部工具,环境返回观察值(observation),形成"思考-调用-观察"的循环。代表性协议如 ReAct、OpenAI Function Calling。
DIVE的核心场景是工具使用智能体的后训练,理解智能体与工具之间"思考-调用-观察"的循环是阅读整篇论文的认知基础,否则无法理解合成任务中轨迹(trajectory)、拒绝采样(rejection sampling)等关键概念。
任务合成(Task Synthesis)
指利用大模型自动生成用于训练其他模型的任务-答案对的过程。常见范式有查询优先(query-first,先生成查询再构造工具轨迹)和证据优先(evidence-first,先执行工具获得证据再反向生成任务)两种。DIVE 属于后者,通过收集器 $\mathcal{F}_{\mathrm{col}}$ 与派生器 $\mathcal{F}_{\mathrm{der}}$ 协同产出 $(Q, A)$。
本文的差异化贡献完全建立在任务合成范式之上,需要理解为什么"证据优先"能从根本上避免传统查询优先方法中常见的"任务不可执行、答案不可验证"问题。
监督微调与强化学习(SFT + RL)
智能体后训练的两阶段范式:SFT 阶段用教师模型做拒绝采样式轨迹模仿学习得到冷启动 checkpoint;RL 阶段(如 GRPO)让策略与环境交互,通过奖励 $R = \alpha R_{\mathrm{format}} + R_{\mathrm{correct}}$ 进一步优化鲁棒性和泛化性。本文中学习率分别设为 $\eta_{\mathrm{SFT}} = 1 \times 10^{-5}$、$\eta_{\mathrm{RL}} = 5 \times 10^{-6}$。
DIVE 的训练管线正是 SFT 冷启动 + GRPO 强化学习的标准两阶段方案,理解奖励函数与学习率设置是看懂实验细节的前提。
分布外泛化(OOD Generalization)
指模型在训练分布之外的任务、工具集、调用协议或执行环境上仍能保持性能的能力。论文将 OOD 分解为五个维度:任务分布(Task)、工具池(Pool)、工具集(Set)、协议(Proto)、环境(Env)。DIVE 在全部 9 个 OOD 基准上平均提升 +22.2 分(RL)即为该能力的直接度量。
DIVE 全部的实验设计都围绕"如何让智能体在新工具和新任务上不掉点"展开,理解 OOD 的多维定义是判断实验结论含金量的关键。
研究动机
近期以 WebExplorer、WebSailor、WebDancer 等为代表的智能体后训练工作,主要通过合成"深研究"类任务并配合 search/browse/code execution 等通用工具来训练工具调用大模型。然而这类方法存在显著的"窄分布陷阱":任务形态高度集中在"信息检索+整合"模式,工具集被锁死在通用搜索引擎和浏览器上,导致模型在训练分布内表现良好,但一旦切换到医学诊断(如需要 PatientLookup 工具)、金融分析(FHIR/EDGAR API)等专业领域时,常出现严重的负迁移——例如 WebExplorer-8B 在 SWE-bench 上仅得 7.0 分,比基础 Qwen3-8B 还低 3.8 分,在 Toolathlon 上更只剩 0.3 分。更深层的张力在于,任务多样化与训练有效性之间存在矛盾:训练阶段需要任务可验证(verifiable,用于轨迹过滤和奖励计算)和可执行(executable,至少存在一条成功路径),而多样性的提升往往同时牺牲这两条性质。
本文的目标是DIVE 旨在构建一个自动化、可扩展的任务合成管线,在保持所有训练任务可验证、可执行的前提下,系统性地扩展训练数据的结构性多样性——具体包括工具类型覆盖(tool-pool coverage)、每任务工具集组合(per-task toolset variety)以及多步工具调用模式(tool-use patterns,如 retrieval-only、retrieval-then-process 等)。最终目标是在 Qwen3-8B 这种小规模基座上,仅使用 48k SFT + 3.2k RL 的数据,就在 9 个分布外基准(含 5 个需要专业工具集的 L3 基准)上超越专门化大模型和量级远大于自身的 frontier 模型。
与已有工作不同的是,现有合成方法存在三种固有缺陷,任一都难以兼顾多样性与可验证性:一是专用管线抽取(如 Liu et al. 2025a)成本高昂,每扩展一种任务类型都要重新设计管线;二是用 LLM 或通用工具模拟环境(如 AgentInstruct、SynthTools)面临模拟工具不可靠导致的不可验证风险;三是查询优先范式(如 ToolLLM、TaskBench)直接在真实工具上合成查询,又会产生大量基于假想查询的不可解任务,需要昂贵的人工质检。本文的核心洞察是——"顺序倒置":不再先生成查询再尝试执行,而是先调用真实工具拿到可复现的执行轨迹,再反向推导出严格由证据蕴含的查询-答案对。这种"构造性接地"使得可执行性和可验证性同时成为合成过程的内在属性,而非事后补救目标。
核心方法
DIVE 的整体思路可以概括为"证据优先、倒序合成":先把多样化的真实世界工具跑一遍拿到执行轨迹(evidence),再从轨迹里反向提炼出有据可查的问题和答案(query-answer),从而在合成阶段就一次性解决可执行性和可验证性两个老大难问题。具体技术上,DIVE 先通过 Crawl-Validate 管线构造一个由 373 个真实工具构成的"工具池",再从维基百科、PubMed、NCBI、Yahoo Finance 等源头抽取约 5000 个种子实体和一批 query-only exemplars,三者构成可独立采样的"工具-种子-示例"三元组资源库。在合成阶段,每轮随机采样得到一个合成配置 $C = \{\mathcal{T}, S, \mathcal{X}\}$(15-50 个工具的子集 + 一个种子 + 3-5 个示例),然后驱动一个 evidence collector agent 在 $\mathcal{T}$ 上做最多 6 步的工具调用,收集真实观察 $E_k$;接着一个 task generator LLM 依据 exemplars 提供的结构先验,把 $E_k$ 反向组装成 $(Q_k, A_k)$。这一收集-派生循环迭代 $K=3$ 次,逐步扩展证据集和任务多样性,最终得到的 $\mathcal{D}_{\mathrm{task}} = \{(Q_K^{(i)}, A_K^{(i)}, \mathcal{T}^{(i)})\}$ 直接进入两阶段训练:SFT 用 GPT-OSS-120B 做教师做拒绝采样得到 48k 轨迹,RL 在 3.2k 边界任务上跑 GRPO 进一步抬升泛化上限。
DIVE 与已有工作的本质区别在于"任务与证据的生成顺序":传统 query-first 范式把任务当成因、轨迹当成果,需要事后打补丁保证任务可达;DIVE 把轨迹当成因、任务当成果,直接消除可达性问题。落到数据结构上,本文创新性地把合成资源解耦为"工具池 + 种子池 + 示例池"三个独立采样空间,从而让多样性可以"按维度独立扩展":想要更多工具类型就扩充工具池,想要更多任务形式就扩充示例池,想要更稀有的实体语义就扩充种子池,三者的随机组合可以指数级扩大任务分布。这种"资源解耦 + 顺序倒置"的组合,使得 DIVE 在只使用 12k 数据(四分之一)的情况下,就能稳定击败仅做量级扩缩的 Gen-DR baseline(48k),首次以受控实验证明了"多样性扩展"在智能体训练中的边际收益远高于"数量扩展"。
方法步骤详情
DIVE 的合成流程分为四个紧密衔接的步骤。Step 1:工具池构建。从 search/browse/code_execution 三个通用原子出发(分别对应 Retrieval 与 Processing 两种工具原语),再爬取并封装 Finance、Biology、Medicine、Academia 四个专家领域的公开 API,对每个候选工具跑正确性、并发安全性、响应一致性三类单元测试,过滤后得到 373 个稳定工具。Step 2:种子与示例池构建。从 Wikipedia、PubMed、NCBI、Yahoo Finance 四个数据源用 LLM 抽取约 5000 个领域实体作为"语义锚点",并从异构任务族(如数据库查询、数学计算、跨域比对等)收集 query-only exemplars,每个 exemplar 仅携带 query 措辞和隐含的 tool-use pattern 两类结构先验,不携带执行轨迹。Step 3:配置采样与证据收集。在每个合成周期随机抽取 $\mathcal{T} \in [15, 50]$ 个工具、1 个种子、3-5 个 exemplars,初始化当前询问 $Q_0 = S$,证据集 $E_0 = \emptyset$;然后 evidence collector agent 在工具集 $\mathcal{T}$ 限定下做最多 $T_{\max} = 6$ 步推理-调用循环,产出 $\tau_k = (r_1, a_1, o_1, \ldots)$,把其中成功的 $(a_t, o_t)$ 对累加得到 $E_k = \mathcal{F}_{\mathrm{col}}(Q_{k-1}, E_{k-1} \mid \mathcal{T})$。Step 4:任务派生与闭环迭代。task generator LLM 接收 $(Q_{k-1}, E_k, \mathcal{X})$,在一次推理中产出 $(Q_k, A_k) = \mathcal{F}_{\mathrm{der}}(Q_{k-1}, E_k \mid \mathcal{X})$,其中 $Q_k$ 的措辞受 exemplar 启发、$A_k$ 必须严格由 $E_k$ 中的工具输出构成;该 $(Q_k, A_k)$ 立即作为下一轮 $Q$ 的种子,循环 $K=3$ 轮后输出 $(Q_K, A_K, \mathcal{T})$ 三元组进入训练集。由于 $A_K$ 由真实工具返回值直接派生,verifier 通过简单的字符串/数值比对即可实现,极大降低了奖励计算的工程开销。
技术新颖性
DIVE 在三个层面具有新颖性。第一,范式层面:首次系统提出"evidence-first synthesis"这一任务合成范式,并形式化其"构造性接地"性质——executable 来自真实工具轨迹,verifiable 来自工具返回值,两者从合成源头被保证而非事后修补。第二,资源建模层面:把合成资源分解为三个解耦的池(tool / seed / exemplar),并通过独立采样实现组合爆炸式多样性,这是以往固定管线方法无法达到的扩展性。第三,实验层面:首次在受控条件下量化了"多样性扩展 vs 数量扩展"的优劣,发现用 12k 多样性数据即可击败 48k 数量扩展数据,且 RL 阶段会进一步放大多样性带来的泛化收益(4 domain 时 RL-SFT gap 增长到 +5.6,而 1 domain 时仅 +4.6)。此外,本文在结构多样性度量上也颇具新意,首次把 tool-call 抽象为 Retrieval/Processing 拓扑,并构建了 222 类的 R/P 拓扑分类体系用于评估。
实验结果
实验在 3 个层级的 9 个 OOD 基准上全面评估 DIVE-8B,核心发现可归纳为四点。发现一:DIVE 在所有 9 个 OOD 基准上均稳定提升,Qwen3-8B 基础模型平均得分仅 13.0,DIVE-SFT 提升到 35.4 (+22.4),DIVE-RL 进一步到 42.5 (+29.5),平均每个 OOD 基准提升 +22.2 分(RL),整体相对最强 8B baseline 提升 +68%。具体而言,GAIA 上 DIVE-RL 达到 61.2,超过 WebExplorer-8B (50.0) 11.2 分;FAB (金融专业) 上达到 34.0,超过 GPT-OSS-120B (22.0) 12 分;Toolathlon (零样本 + 604 MCP 工具 + 有状态容器) 上从 Qwen3-8B 的 0.9 提升至 8.3,接近 Gemini-2.5-Pro 的 10.5。发现二:DIVE 即使对专门化模型也能以"通用身份"取得优势。WebExplorer-8B (专门为深研究微调) 在 L3 专业工具集上出现负迁移,SWE-bench 仅 7.0、Toolathlon 仅 0.3,反而比 base Qwen3-8B 还低,而 DIVE-RL 在两者上分别为 18.3 和 8.3,展现真正的跨域泛化能力。发现三:多样性扩展显著优于数量扩展。受控实验中,Diversity-only (固定 12k 数据,工具池从 1 domain 扩到 4 domains) 在所有 OOD 基准上都稳定优于 Quantity-only (固定 Gen-DR 任务族,数据从 12k 扩到 48k),且仅用 1/4 数据量就取得更好成绩;Pool-Expansion+Variety 又比单纯 Toolset-Variety 提升更明显,说明扩展新工具类型比单纯增加工具集组合的边际收益更高。发现四:RL 阶段放大多样性红利。在 24 条扩缩路径上,SFT 阶段多样性收益已显现 (mean 从 29.1 → 34.8),RL 进一步将均值抬升且收窄了四分位间距,平均 RL-SFT gap 从 1 domain 时的 +4.6 扩大到 4 domain 时的 +5.6,说明 RL 的"探索能力"能与 SFT 的"模仿能力"在多样性维度上形成正反馈。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| DIVE-Eval (L1 分布内) | 成功率 (%) | 42.5 | WebExplorer-8B: 19.1; Qwen3-8B base: 13.0 | +23.4 相对 base |
| GAIA (L2 通用深研究) | 成功率 (%) | 61.2 | WebExplorer-8B: 50.0; Claude-4-Sonnet: 63.7 | +11.2 相对 WebExplorer |
| HLE (L2 通用深研究) | 成功率 (%) | 17.8 | WebExplorer-8B: 17.3; Claude-4-Sonnet: 20.8 | +0.5 相对 WebExplorer |
| BROWSECOMP (L2 通用深研究) | 成功率 (%) | 16.4 | WebExplorer-8B: 15.7 | +0.7 |
| XBENCH-DeepSearch (L2 通用深研究) | 成功率 (%) | 58.1 | WebExplorer-8B: 53.7 | +4.4 |
| FinSearchComp T2 (L2 金融深研究) | 成功率 (%) | 67.3 | WebExplorer-8B: 35.9; Claude-4-Sonnet: 60.2 | +31.4 |
| FinSearchComp T3 (L2 金融深研究) | 成功率 (%) | 37.3 | WebExplorer-8B: 18.1; Claude-4-Sonnet: 33.3 | +19.2 |
| Finance Agent Benchmark (L3 金融专业) | 成功率 (%) | 34.0 | GPT-OSS-120B: 22.0; EnvScaler-8B: 14.0 | +12 |
| MedAgentBench (L3 医学专业) | 成功率 (%) | 57.3 | EnvScaler-8B: 56.6; GPT-OSS-120B: 64.3 | +18.9 相对 base |
| SWE-bench Verified (L3 软件工程) | 成功率 (%) | 18.3 | EnvScaler-8B: 11.5; WebExplorer-8B: 7.0 | +7.5 |
| Toolathlon (L3 零样本通用) | 成功率 (%) | 8.3 | WebExplorer-8B: 0.3; EnvScaler-8B: 2.2; Claude-4-Sonnet: 29.9 | +7.4 相对 base |
局限与改进
作者在论文中坦诚承认若干限制,值得读者注意。第一,DIVE 的合成依赖 Claude-4-Sonnet 作为 evidence collector 和 task generator,这意味着合成数据的质量与教师模型能力高度耦合;在更弱的教师下,evidence 的多样性可能塌缩,evidence-first 优势会被削弱。第二,373 个工具覆盖了 5 个领域,但仍远未穷尽真实场景——尤其在需要私有 API、企业内部系统、或需要长时间状态保持的复杂环境(如 TOOLATHLON 的 stateful Docker)上,DIVE 的覆盖度明显不足,这也是 Toolathlon 上 8.3 分仍显著低于 Claude-4-Sonnet 29.9 分的主因。第三,K=3 轮迭代、6 步工具调用的上限限制了单个任务的深度,导致论文 Example 中虽然出现了 50 calls / 26 available 的复杂金融任务,但更多任务停留在 3-5 步的多跳推理,对于需要 20+ 步的工具链式推理仍是开放问题。第四,论文未深入分析 RL 阶段在 L3 高难度基准(如 Toolathlon、SWE-bench)上增益明显小于 L2 的原因,推测与有状态环境、HTTP 协议差异有关,但缺少消融。从我自己的观察看,DIVE 仍未解决"评测协议异构性"问题——例如 MedAgentBench 使用 FHIR GET/POST HTTP 协议,而训练时全部采用 OpenAI Function Calling 协议,两者间的协议转换成本可能占据了大量模型 capacity,若能直接在合成阶段引入多种协议模板,可能会进一步压缩与 frontier 模型的差距。
独立分析的弱点
独立审视 DIVE,我看到四个值得改进的弱点。弱点一:工具池的领域覆盖仍偏静态。373 个工具来自一次性 Crawl-Validate,无法自动发现新 API;在工具生态快速演进的当下(MCP 协议下每周都有新 server 上线),DIVE 缺少一个持续 ingest 新工具的机制,容易陷入"训练时工具集-推理时工具集"的时间漂移问题。改进方向:引入基于 MCP registry 的自动化监控 + 周期性 mini-validation pipeline,把工具池做成"持续生长"的活体资源。弱点二:多样性受限于 exemplars 的质量与多样性。task generator 完全依赖 exemplars 提供的"query 措辞 + 隐含 tool-use pattern",而 exemplars 是人工/半自动收集的 query-only 数据,本身可能存在长尾覆盖不足。改进方向:在 task derivation 阶段引入 few-shot bootstrapping——用已有 (Q, A, τ) 三元组自动挖掘新的隐含模式,代替人工 curation 的 exemplars。弱点三:K=3 轮迭代深度的代价收益不清晰。论文 Appendix C 提到迭代能增加多样性,但缺少对 K=1, 2, 4, 5 等的消融,无法判断 3 是工程甜点还是真正最优;更深的迭代也可能让 evidence 过度膨胀,导致 task derivation 的 grounding 信号被稀释。改进方向:对 K 做系统性消融,并引入 evidence size 自适应停止准则(例如当新加入的 $(a_t, o_t)$ 与已有证据的 Jaccard 相似度高于阈值时停止)。弱点四:RL 阶段奖励过于稀疏。$R = \alpha R_{\mathrm{format}} + R_{\mathrm{correct}}$ 只在最终答案层面给信号,缺少对中间步骤(如某步工具调用是否必要、是否冗余)的细粒度反馈,这可能是 Toolathlon 上仅得 8.3 分的关键瓶颈。改进方向:引入 process reward 或 step-level credit assignment,例如对工具调用图的每个节点(检索-处理-整合)单独打分。
未来方向
作者在 Conclusion 中提出了一些方向,基于本文成果可延伸的研究包括:(1) 把 DIVE 范式扩展到多模态工具——例如图像理解、代码沙箱、PDF 解析等,把"证据"从纯文本工具返回扩展到多模态 observation,这对开放域 agent 落地尤为关键;(2) 与 active learning 结合,让 evidence collector 在合成阶段就显式挑选"对当前模型最难的任务"来合成,而非均匀采样配置 $C$,进一步提升 RL 阶段的样本效率;(3) 探索 DIVE 与 retrieval-augmented synthesis 的结合:从外部知识库(如 Wikipedia dump)自动挖掘实体关系作为种子,有望把种子池规模从 5000 扩展到数十万。从我自己延伸的角度,我认为 DIVE 的 evidence-first 思路天然适合与 self-play 框架结合——让 evidence collector 主动与 task generator 博弈,前者尝试"制造难题",后者尝试"生成可解的任务",可能把任务合成的上限推到接近自演化水平。此外,DIVE 当前聚焦单智能体单任务,扩展到多智能体协作场景(一个智能体调度多个 sub-agents 调用不同工具集)是一个自然的下一步,这能让 toolset variety 从"扁平集合"升级为"层级化能力栈"。
复现评估
DIVE 的复现性整体良好但工程门槛不低。资源层面,373 个真实工具分布在通用、Finance、Biology、Medicine、Academia 五个领域,涉及 NCBI/PubMed、Yahoo Finance、CrossRef、OpenAlex、EDGAR 等公共 API,绝大多数无需付费或仅需免费 key,但部分金融 API 可能存在 rate limit 和稳定性问题,作者通过 Crawl-Validate pipeline 做了单元测试过滤,但复现者仍需自行处理实时 API 变更。代码与数据层面,论文公布了项目主页(https://sheep333c.github.io/DIVE/),但是否完全开源 48k SFT + 3.2k RL 训练数据、teacher trajectories、合成脚本仍需确认;Claude-4-Sonnet 作为合成阶段的 teacher 本身是闭源商业 API,这部分成本不可忽视(每轮 K=3 迭代都需要多次 LLM 调用)。算力层面,Qwen3-8B 的 SFT (300 steps, batch 64, lr 1e-5) 和 RL (100 steps, batch 512, lr 5e-6) 都在论文中明示,推算大致需要 32-64 张 A100 训练 1-3 天,对于有成熟 RL 基建(如 verl/OpenRLHF)的实验室是可负担的,但对个人研究者门槛较高。复现难度评估为"中等偏难":核心算法思路清晰,工程实现细节(如 exemplar 来源、Crawl-Validate 的具体规则、exemplar pool 来源)依赖论文正文外的补充材料,建议复现者先在 Finance 单 domain 上做 K=1 的最小闭环验证再扩展。
论文图表
Figure 1 用三段式对比呈现 DIVE 的动机:顶部展示两种主流但有缺陷的合成路径(固定工具集合成、深研究任务池化),中部展示另两种受限路径(模拟工具环境、查询优先合成),底部展示 DIVE 的 evidence-first 流程(执行真实工具→收集证据→反向派生任务)。右下角的雷达图用三色直观对比:base 模型在所有维度都很弱,深研究数据训练的模型在分布内强但 OOD 弱,DIVE 训练在所有维度都更均衡。
这是理解 DIVE 立意的核心图,它同时回答了"现有方法哪里不行"和"DIVE 怎么做"两个关键问题,雷达图也提供了视觉化的"diversity scaling > quantity scaling"的初步证据。