arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2610.13481cs.AI
收藏

分支式在线策略蒸馏

Branching On-Policy Distillation

Tsinghua University(清华大学);SenseTime Research(商汤科技研究院);Peking University(北京大学)

Dongyang Chen, Shudong Liu, Shubo Li, Duoxun Tang, Jinwei Liang, Zehuan Wu, Lewei Lu

AI总结:

本研究针对在线策略蒸馏监督覆盖不足的问题,提出BranchOPD方法,通过分支采样与共享前缀复用降低成本,在数学任务和ALFWorld、WebShop等智能体任务上均优于基线方法。

摘要:

在线策略蒸馏(OPD)为学生生成的轨迹提供教师监督,但单条轨迹在每个决策点仅实现一种后续动作,在有限采样预算下限制了监督覆盖范围。我们提出BranchOPD(分支式在线策略蒸馏),该方法利用学生不确定性和师生分歧选择信息丰富的决策点,从共享前缀重新采样学生策略的后续动作,并对主轨迹和分支应用 token 级教师监督。BranchOPD 适用于数学领域的 token 级决策以及智能体任务的回合内决策。与7条独立轨迹相比,共享前缀复用使学生生成的 token 成本在 ALFWorld 上降低约19%,在 WebShop 上降低约31%。我们还引入协议边界修复(PBR),该方法在协议约束下使用针对性监督以缓解重复和无效生成。BranchOPD 在所有评估基准上均优于普通 OPD 和多 rollout OPD(n=7),相较于普通 OPD,其在16个采样答案上的平均数学准确率(avg@16)提升18.55个百分点,在 ALFWorld 分布内/分布外平均成功率提升13.8个百分点。机制分析和选择 ablation 实验表明,在不确定决策点进行分支可扩大后续行为覆盖范围,且优于随机选择和回合初始分支。

英文摘要:

On-policy distillation (OPD) provides teacher supervision on student-generated trajectories, but a single trajectory realizes only one continuation at each decision, limiting supervision coverage under a finite sampling budget. We propose BranchOPD (Branching On-Policy Distillation), which selects informative decision points using student uncertainty and teacher-student disagreement, resamples student-policy continuations from shared prefixes, and applies token-level teacher supervision to both main trajectories and branches. BranchOPD operates at token-level decisions in mathematics and within-turn decisions in agent tasks. Shared-prefix reuse reduces student-generated token costs by approximately 19% on ALFWorld and 31% on WebShop relative to seven independent trajectories. We also introduce protocol-boundary repair (PBR), which uses targeted supervision under protocol constraints to mitigate repetition and wasted generation. BranchOPD outperforms Vanilla OPD and Multi-Rollout OPD (n=7) across all evaluated benchmarks, with gains over Vanilla OPD of 18.55 percentage points in mean mathematics accuracy over 16 sampled answers (avg@16) and 13.8 points in mean ALFWorld ID/OOD success. Mechanistic analyses and selection ablations show that branching at uncertain decisions expands subsequent behavior coverage and outperforms random selection and turn-initial branching.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑