DR-Venus:仅用 10K 开放数据训练的前沿边缘级深度研究智能体 DR-Venus: Towards Frontier Edge-Scale Deep Research Agents with Only 10K Open Data
用 10K 开放数据和两阶段训练打造前沿 4B 边缘级深度研究智能体
前置知识
深度研究智能体
深度研究智能体是基于大语言模型构建的 agent 系统,需在长程交互中反复调用搜索/浏览工具,逐步收集证据并合成最终答案。这类任务通常涉及上百轮工具调用,要求稳定的规划和工具使用能力。
本文所有方法都围绕长程深度研究任务的执行可靠性展开,不理解长程交互特性就无法理解 IGPO 等设计动机。
GRPO 与策略优化
GRPO 是一种基于分组采样的强化学习算法,对同一 query 采样一组轨迹,用组内相对优势代替 critic。它源自 PPO 思想,去掉了 value network,靠同组样本的回报均值和方差做基线估计。
本文提出的 IGPO 建立在 GRPO 框架上,将轨迹级稀疏奖励扩展为 turn-level 密集奖励,理解 GRPO 是看懂 IGPO 改进点的前提。
监督微调 (SFT) 与数据清洗
SFT 是用标注好的 (输入, 输出) 序列对预训练模型进行有监督训练的标准范式。在 agent 场景中,SFT 数据通常是完整的工具调用轨迹,需先做格式对齐、错误过滤、去重等清洗才能用于训练。
本文 70% 的工程努力花在 SFT 数据清洗和重采样上,不了解 SFT 数据对模型敏感性的影响就难以理解为何小模型必须做严格清洗。
信息增益 (Information Gain)
信息增益衡量一个观察/动作使模型对正确答案后验概率增加的程度。在 RL 中常用于构造中间步骤的密集奖励,避免只依赖稀疏的最终结果反馈。
IGPO 的核心奖励 r^IG_{i,t} = log pi_theta(g | h_{i,<=t}) - log pi_theta(g | h_{i,<=t-1}) 正是基于信息增益定义,是理解本文 RL 算法的钥匙。
研究动机
现有深度研究智能体几乎都构建在 30B 以上的大模型之上(如 Tongyi-DR-30B、WebSailor-V2-30B-A3B、REDSearcher-30B-A3B),并且依赖封闭数据或复杂训练管线,部署成本高、延迟大、隐私风险高,难以在边缘设备落地。同时,作者观察到小模型对噪声轨迹、格式错误和工具调用冗余格外敏感:小模型 RL rollout 经常出现整组无成功轨迹的情况,导致 advantage collapse,训练效率极低。表 1 也显示在 ≤9B 区间内,多数基线在 BrowseComp 上仅 5–25 分,性能远落后于实际需求。
本文的目标是本文希望回答一个具体问题:在严格受限的开放数据条件下(仅约 10K SFT 轨迹 + 1K RL 查询),能否训练出接近甚至比肩 30B 系统的 4B 深度研究智能体?具体目标包括:(1) 设计一套两阶段训练流水线,先用清洗+长程重采样的 SFT 打底能力,再在 IGPO 框架下用 turn-level 密集奖励做 RL;(2) 让 4B 模型在六个深度研究基准(BrowseComp、BrowseComp-ZH、GAIA、xBench-DS-2505/2510、DeepSearchQA)上同时取得 SOTA 小模型成绩,并显著缩小与 30B 系统的差距。
与已有工作不同的是,本文的独特切入角度是把“小模型 + 开放数据”的成功关键归结为“数据质量 + 数据利用率”两个杠杆,而不是堆参数或堆闭源数据。具体差异有三:第一,与 REDSearcher-30B-A3B 仅做格式对齐不同,本文提出 turn-aware resampling(1×/2×/5× 加权),将 >50 turn 的轨迹占比从 60.28% 拉到 80.15%,从而让 4B 模型在长程任务上也能拿到足够监督;第二,与 GRPO 仅做轨迹级稀疏奖励不同,本文改造 IGPO 引入 browse-aware IG 分配、turn-level format penalty、IG-Scale 重平衡等机制,让 RL 在 200 turn 量级的长程任务中真正能学到东西;第三,论文通过 Pass@K 揭示了小模型 capability ceiling 远高于 Pass@1 的结论,为边缘部署 + test-time scaling 的方向提供了实证支撑。
核心方法
DR-Venus 的整体思路是:先用严格清洗后的高质量 SFT 数据给 4B 模型打底工具使用与长程推理能力,再用基于信息增益的 turn-level RL 精调执行可靠性。直觉上,作者认为小模型的根本瓶颈不在模型容量本身,而在有限数据被浪费和噪声干扰严重——所以方法路线也分两步走:第一步把数据“挤干”,第二步把监督信号“加稠”。SFT 阶段的关键是环境对齐、剔除 Python-Interpreter 等不允许的工具、去除重复的 browse 调用、用 Qwen3-235B-A22B-Instruct-2507 做正确性过滤、并按 turn 数加权重采样长程轨迹;RL 阶段的关键是把 GRPO 的轨迹级奖励替换为 IGPO 的 turn-level 信息增益奖励,并加入格式惩罚和 IG-Scale 重平衡机制。
核心创新在于把“小模型难以训练”这个工程难题分解为两个可独立优化的子问题——数据利用率和奖励信号密度——并分别用 turn-aware resampling 和 IGPO 给出具体解法。与已有方法的本质区别是:(1) 不同于 REDSearcher 仅做格式对齐或 DeepDive 等小模型系统直接堆数据,本文把 >100 turn 轨迹重采样 5×,让 18,745 个 SFT 实例集中覆盖长程交互;(2) 不同于 Tongyi-DR、DeepMiner 等依赖稀疏结果奖励的 GRPO 训练,本文用 $r^{IG}_{i,t} = \log \pi_\theta(g | h_{i,\leq t}) - \log \pi_\theta(g | h_{i,\leq t-1})$ 在每个 turn 给出密集监督,并通过 $\bar{R}_{i,t} = \sum_{k=t}^{T_i} \gamma^{k-t} \bar{r}_{i,k}$ 做折扣累积,$\gamma=0.95$;(3) 加入 browse-aware IG 分配(仅在 browse turn 计算 IG 并向前回填到上一个 browse 之间的所有 search turn)和 $\lambda_{\text{fmt}}=1.0$ 的 turn-level 格式惩罚,避免轨迹级惩罚误伤正常 turn。
方法步骤详情
完整流程分两大阶段。**阶段一:SFT 数据构建与训练。** 输入是 10,001 条原始 REDSearcher 轨迹,经过四步清洗:(a) 环境对齐,把所有轨迹转换为推理时使用的同一消息格式、system prompt、tool-call 参数,10,001 条全部成功;(b) 禁用工具剪枝与去重,由于运行时只有 search 和 browse 两个工具,把 Python-Interpreter 等调用在 turn 级删掉(影响 1,064 条,去除 3,378 次不允许工具调用)并删除重复的 search/browse 对(影响 6,821 条,去除 15,728 次重复);(c) 正确性过滤,用 Qwen3-235B-A22B-Instruct-2507 做 judge,保留 9,365 条;(d) Turn-aware resampling,按 0–50/51–100/>100 turn 分别赋 1×/2×/5× 权重,把数据集从 9,365 拉到 18,745 实例。训练时用 Qwen3-4B-Thinking-2507 底座,8 张 A100,verl FSDP,最大长度 200K token,全局 batch=32,单卡 micro-batch=1,学习率 1e-5,单 epoch,assistant token 计算 loss、observation token mask 掉。**阶段二:Agentic RL。** 输入是 1K 来自 REDSearcher 的 query-answer 对,从 SFT 模型继续训练。每条 query 采样 G=8 条轨迹,温度 1.0,每 turn 最多生成 8,192 token,总 context 上限 256K token,batch=16。奖励设计按公式 $\hat{r}_{i,t}=\begin{cases} r_{i,t} & \text{格式正确}\\ -\lambda_{\text{fmt}} & \text{格式错误}\end{cases}$ 做格式掩码,按公式 5 在 rollout 组内分别对 IG 和 outcome 奖励做 $(\hat{r}-\mu)/\sigma$ 归一化,再用 IG-Scale(公式 7,$\eta=0.3$、$s_{\max}=10$)把 IG 量纲对齐到 outcome 量纲,最后按公式 9 用 $\gamma=0.95$ 做 turn-level 折扣累积 $\tilde{R}_{i,t}=\sum_{k=t}^{T_i}\gamma^{k-t}\bar{r}_{i,k}$。优化目标直接套用 IGPO 公式 10,clip 阈值 $\epsilon$、KL 系数 $\beta$ 沿用 GRPO 默认设置。
技术新颖性
技术新颖性体现在三个层次。第一,**数据利用层面**的 turn-aware resampling 是一个简单但被忽视的杠杆,仅靠把 13.29% 的超长程轨迹升到 33.21%,BrowseComp 上就能从 22.8 提到 26.8(+4.0),BrowseComp-ZH 从 33.9 提到 35.7(+1.8),证明长程监督密度比样本总量更重要。第二,**奖励设计层面**的 IGPO 改造把“轨迹级稀疏”问题改造成“turn-level 密集”问题,并通过 browse-aware 分配避免在低信息量的 search turn 上浪费监督信号;这与 Tongyi-DR 等使用结果奖励 + judge 模型的工作形成对比,IGPO 是直接在策略概率空间上做差分,不需要额外 reward model。第三,**训练稳定性层面**的 IG-Scale 和格式惩罚是工程上很关键的细节:在 200 turn 量级的任务里 outcome reward 经常全 0,没有 IG-Scale 就会变成纯 IG 主导的更新,容易陷入局部最优;$\lambda_{\text{fmt}}=1.0$ 把格式错误精确定位到具体 turn,避免传统的“一条轨迹格式错就整条惩罚”的粗糙信号。这三个层次叠加,使得 4B 模型能在 RL 中获得稳定且持续的增益(BrowseComp +2.3、BrowseComp-ZH +2.0),而同等条件下 GRPO 反而出现 -1.5 的退化。
实验结果
实验在六个基准上全面对比 DR-Venus-4B 与闭源大模型、≥30B 开源 agent 和 ≤9B 小模型,主要发现可归纳为四点。**(1) SFT 已经很强。** DR-Venus-4B-SFT 在 BrowseComp 拿到 26.8、BrowseComp-ZH 35.7、GAIA 65.4、xBench-DS-2505 69.0、xBench-DS-2510 35.3、DeepSearchQA 37.7,比之前的 SOTA 小模型 AgentCPM-Explore-4B 在六个基准上分别领先 +2.7/+6.6/+1.5/+1.3(N/A 上 SFT 高出 +4.9 DeepSearchQA),证明仅靠 SFT 数据清洗+重采样就能让 4B 模型压过同尺寸基线。**(2) RL 把 SFT 进一步抬升。** DR-Venus-4B-RL 在 BrowseComp 29.1 (+2.3)、BrowseComp-ZH 37.7 (+2.0)、xBench-DS-2505 74.7 (+5.7)、xBench-DS-2510 40.7 (+5.4)、DeepSearchQA 39.6 (+1.9),仅 GAIA 微跌 -1.0;消融实验中把 IGPO 换成 GRPO 后 BrowseComp 直接掉到 25.3 (-1.5),验证了密集 turn-level 奖励在小模型上的必要性。**(3) 与 30B 系统对比并不虚。** 在 xBench-DS-2505 上 DR-Venus-4B-RL 拿到 74.7,已逼近 Tongyi-DR-30B 的 75.0;在 BrowseComp-ZH 上 DR-Venus-4B-SFT 直接超越 OpenResearcher-30B-A3B(26.3 → 35.7)。**(4) Pass@K 揭示 capability ceiling 远高于 Pass@1。** BrowseComp-ZH 上 SFT 模型 Pass@16 高达 78.5,超过 Tongyi-DR-30B(46.7)、Gemini-3-Pro(66.8)、GPT-5 High(65.0),说明小模型的真实能力被严重低估,test-time scaling 是边缘级部署的关键武器。工具使用分析(图 3)也显示 RL 把整体 browse 占比从 17.49% 拉到 22.46%,正确轨迹的 browse 占比从 23.71% 升到 28.96%,并在 xBench-DS-2510 上把“SFT 错误轨迹 browse 更多”的反常现象翻转过来,进一步证明 RL 让工具使用与任务成功对齐。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| BrowseComp(英文长程网页检索) | Accuracy | DR-Venus-4B-SFT 26.8 / DR-Venus-4B-RL 29.1 | AgentCPM-Explore-4B 24.1、WebExplorer-8B-RL 15.7、Tongyi-DR-30B 43.4 | 比同尺寸 SOTA 高 +5.0;比 30B Tongyi-DR-30B 低 14.3 但仅为 1/8 参数;RL 比 SFT 高 +2.3 |
| BrowseComp-ZH(中文长程网页检索) | Accuracy | DR-Venus-4B-SFT 35.7 / DR-Venus-4B-RL 37.7 | AgentCPM-Explore-4B 29.1、WebExplorer-8B-RL 32.0、OpenResearcher-30B-A3B 26.3 | 比 4B SOTA 高 +8.6;超过 30B OpenResearcher-30B-A3B 达 +11.4 |
| GAIA(Text-Only,多步推理) | Accuracy | DR-Venus-4B-SFT 65.4 / DR-Venus-4B-RL 64.4 | AgentCPM-Explore-4B 63.9、DeepSeek-V3.2 75.1、GPT-5 High 76.4 | SFT 阶段领先 4B SOTA +1.5;RL 阶段微跌 -1.0,可能与 IG-Scale 在 outcome reward 已较密集时偏保守有关 |
| xBench-DS-2505(深度搜索) | Accuracy | DR-Venus-4B-SFT 69.0 / DR-Venus-4B-RL 74.7 | AgentCPM-Explore-4B 70.0、Tongyi-DR-30B 75.0、SMTL-30B-300 82.0 | RL 比 SFT 提升 +5.7;逼近 30B Tongyi-DR-30B 的 75.0 |
| xBench-DS-2510(深度搜索更新版) | Accuracy | DR-Venus-4B-SFT 35.3 / DR-Venus-4B-RL 40.7 | AgentCPM-Explore-4B 34.0、Kimi-K2.5 46.0、Gemini-3-Pro 53.0 | SFT 阶段领先 4B SOTA +1.3;RL 阶段大幅 +5.4,验证 IGPO 对长程任务稳定性贡献明显 |
| DeepSearchQA(多步深度研究) | Accuracy | DR-Venus-4B-SFT 37.7 / DR-Venus-4B-RL 39.6 | AgentCPM-Explore-4B 32.8、WebExplorer-8B-RL 17.8、Claude-4.5-Opus 80.0 | SFT 阶段领先 4B SOTA +4.9;RL 再 +1.9,但与闭源大模型仍有 40+ 分差距 |
局限与改进
作者明确承认的局限有两点:(1) RL 训练数据全部为英文,导致 BrowseComp-ZH 在 Pass@8/16 出现轻微下滑(SFT 74.0/78.5 vs RL 73.0/76.5),是 distribution mismatch 引起的;(2) 当前评测不允许额外的 context management 或 test-time scaling 技巧,因此 RE-TRAC-4B、Marco-DR-8B、MiroThinker-v1.0 等系统被显式排除,作者承认这是为了公平对比而非否认其价值。此外从结果数据自身观察还可指出三点:(a) DR-Venus-4B-RL 在 GAIA 上相对 SFT 出现 -1.0 的回落,提示 IGPO 在 outcome reward 已经较密集的短程任务上可能反而偏保守;(b) 与 Claude-4.5-Opus(DeepSearchQA 80.0)、Gemini-3-Pro(BrowseComp-ZH 66.8)等闭源模型仍有显著差距,说明纯开放数据 + 小模型仍存在 capability ceiling;(c) 评测对工具服务(Serper+Jina)和 summarization 模型(Qwen3-30B-A3B-Instruct-2507)的依赖意味着复现时外部基础设施质量会直接影响结果。
独立分析的弱点
从独立分析角度看,本文至少有四个值得改进的弱点。**弱点一:SFT 与 RL 数据均来自同一来源 REDSearcher**,存在严重的 source bias,模型在 BrowseComp 这类与 REDSearcher 数据分布接近的任务上提升明显(+5.7 xBench-DS-2505),但在分布外任务如 DeepSearchQA 上提升有限(+1.9),且被 Claude-4.5-Opus 等闭源模型大幅超越;改进方向是引入跨源 SFT 数据或自生成 query 做数据扩展。**弱点二:Browse-aware IG 分配仍属于粗粒度启发式**,把所有 search turn 平摊到下一个 browse turn 之间的 IG,没有考虑不同 search turn 本身的信息量差异,可能把真正的“关键 search”和噪声 search 等量齐观;改进方向是基于 query-context 相似度或 attention 权重做更细粒度的 IG 归因。**弱点三:缺少对模型本身 capability ceiling 的理论分析**,Pass@K 曲线只是经验观察,没有回答“为什么 4B 能在 Pass@16 接近 Tongyi-DR-30B”——这可能是底层 Qwen3-4B-Thinking 的预训练知识 + REDSearcher 分布匹配的共同作用,而非 DR-Venus 方法本身的功劳;改进方向是消融底座模型或在不同底座上重复实验来剥离方法贡献。**弱点四:评测依赖固定 Serper/Jina 工具栈**,没有报告工具调用失败率、网页访问成功率等系统级指标,工具故障可能掩盖或放大模型本身的性能差异;改进方向是引入工具故障模拟或使用更稳健的浏览器接口(CDP/WebArena)。
未来方向
作者在结论部分给出的未来方向比较克制,主要强调“释放模型、代码、关键 recipes 支持复现研究”,但根据本文成果至少可以延伸出四条研究方向。**方向一:test-time scaling 范式落地**。Pass@K 显示 4B 在 K=16 时几乎接近 Tongyi-DR-30B(BrowseComp-ZH 78.5 vs 46.7),下一步可探索 Best-of-N、Self-Consistency、Process Reward Model 等与 IGPO 兼容的解码策略,把 capability ceiling 转化为可部署的 Pass@1 性能。**方向二:多语种 RL 数据扩展**。BrowseComp-ZH 在大 K 值上 RL 出现轻微退化,根因是 RL 数据全英文;引入中英混合或纯中文 RL 数据(可借鉴 Tongyi-DR 的中英混训策略)有望让 RL 在中文场景也保持单调提升。**方向三:模型规模-数据效率 trade-off 的系统研究**。本文只测了 4B,可延展到 1B/2B/7B/9B 同一 REDSearcher 数据集,看 turn-aware resampling 的增益曲线是否随模型变小而放大,并据此设计“数据利用率为先”的训练范式。**方向四:与 RAG / memory 模块的协同**。当前 4B 模型在长程任务中 context 长度已经逼近 256K 上限,未来可探索把 IGPO 的 turn-level 信用分配思想迁移到 memory-augmented agent,让模型主动决定何时写入/读取长期记忆,从而突破纯 in-context 推理的瓶颈。
复现评估
复现评估整体良好。代码、模型和关键 recipe 均承诺开源:模型在 HuggingFace 集合 inclusionAI/dr-venus 下发布,代码仓库为 github.com/inclusionAI/DR-Venus;SFT 训练数据 REDSearcher_SFT_10K(10K 原始轨迹)和 RL 训练数据 REDSearcher_RL_1K 也都公开。算力需求方面,SFT 阶段 8×A100、单 epoch、最长 200K token 序列并行 size=8,verl FSDP 训练器;RL 阶段 16×A100、rollout 上限 256K context、batch=16、group size=8、verl vLLM 引擎,单次完整 RL 训练成本较高但仍属于中小团队可承受范围。底座 Qwen3-4B-Thinking-2507 与 judge 模型 Qwen3-235B-A22B-Instruct-2507 均开源权重。评测侧对外部基础设施的依赖(Serper 搜索 API、Jina 浏览 API、Qwen3-30B-A3B-Instruct-2507 摘要)是潜在变数,搜索结果和网页快照的稳定性可能影响 BrowseComp 等长程基准的可复现性,文中对每条 query 允许最多 200 interaction step、temperature 1.0、top-p 0.95、top-k 20、presence penalty 1.1、最大 token 256K 的解码设置也必须严格一致才能复现论文中的具体数字。总体而言方法论层面可复现性高,但完整复现仍需对外部工具栈投入工程精力。
论文图表
散点对比图,纵轴为 BrowseComp 准确率、横轴为 BrowseComp-ZH 准确率,把 DR-Venus-4B-SFT、DR-Venus-4B-RL 与 Qwen3-4B-Thinking-2507、WebSailor-7B、WebSailor-V2-30B-A3B、DeepDive-9B/32B、Tongyi-DR-30B 等开源模型放在同一象限对比。DR-Venus-4B-RL 落在象限右上角,明显领先所有 ≤9B 小模型,并接近 30B Tongyi-DR-30B。
这张图是论文 hero figure,一眼展示 4B 模型可以同时跑赢同尺寸基线并接近 30B 系统,是 motivation 的最直观证据;放在 motivation 章节最能体现论文想解决的“边缘级能不能打”问题。