ActionOPD:蒸馏决策,而非token

Hugging Face Daily Papers 论文

摘要

ActionOPD提出一种动作级而非token级的on-policy蒸馏方法,通过语义动作对齐将大型工具使用agent蒸馏到Qwen3-4B/8B等小模型,在四个OOD基准上平均提升3.3-6.9个百分点,且无需教师模型的token概率或相同分词器。

将强大的语言模型agent蒸馏到更小的模型中,需要使学生模型的行为与教师模型对齐。这引发了一个根本性问题:应该在哪些状态以及在什么粒度上进行这种对齐?离线模仿提供了稠密但基于token的监督,且可能遭受状态分布不匹配的问题。Token级的on-policy蒸馏(OPD)解决了这种不匹配,但通常需要教师模型的token概率以及兼容的分词器。我们认为,语义动作是agent蒸馏中一种自然的监督单元。我们提出ActionOPD,一种动作级的on-policy蒸馏方法,其动机是最大化教师动作在学生所遇到状态上的期望似然。在每个这样的状态下,ActionOPD分别从学生模型和黑盒教师模型中采样动作,并根据学生动作与采样教师动作的语义一致性给予奖励。通过策略优化,ActionOPD将学生模型的动作级行为与教师的决策对齐,而无需复现教师撰写的推理过程或访问教师的token概率。使用两个教师模型在多样且精选的工具使用任务上训练Qwen3-4B和Qwen3-8B,在四个分布外基准上平均提升3.3-6.9个百分点。ActionOPD随学生模型规模和教师模型能力的提升而良好扩展,且不要求教师和学生属于同一模型家族或使用相同的分词器。在τ^2-bench上,ActionOPD取得了与离线蒸馏和token级OPD相竞争的性能。我们进一步探索了使用同模型教师和特权上下文的Self-ActionOPD。这些结果确立了ActionOPD作为一种有前景的agent蒸馏方法的地位。
查看原文
查看缓存全文

缓存时间: 2026/10/15 07:03

论文页面 - ActionOPD:蒸馏决策,而非Token

来源:https://huggingface.co/papers/2610.19820

摘要

将强大的语言模型智能体蒸馏到更小的模型中,需要对齐学生模型与教师模型的行为。这引出了一个核心问题:应在哪些状态上、以何种粒度进行这种对齐?离线模仿提供稠密但仅限token级别的监督,且可能遭受状态分布不匹配的问题。Token级别的在线策略蒸馏(OPD)解决了这一不匹配问题,但通常需要教师模型的token概率分布以及兼容的分词器。我们认为,语义动作(semantic action)为智能体蒸馏提供了一种自然的监督单元。我们提出了ActionOPD,一种动作级别的在线策略蒸馏方法,其核心动机是在学生模型所访问的状态上最大化教师动作的期望似然。在每个这样的状态上,ActionOPD分别从学生模型和一个黑盒教师模型中采样动作,并根据学生动作与采样教师动作之间的语义一致性给予奖励。通过策略优化,ActionOPD在不要求复现教师撰写的推理过程、也不需要访问教师token概率的情况下,将学生的动作级别行为与教师的决策对齐。我们在多样化的精选工具使用任务上,以两个教师模型训练Qwen3-4B和Qwen3-8B,在四个分布外(out-of-distribution)基准测试上平均获得了3.3至6.9个百分点的提升。ActionOPD在学生模型规模和教师模型能力方面均表现出良好的可扩展性,且不要求教师和学生共享同一模型系列或分词器。在τ²-bench上,ActionOPD取得了与离线蒸馏和Token级别OPD相当的性能。我们还进一步探索了使用同模型教师和特权上下文的Self-ActionOPD。这些结果确立了ActionOPD作为智能体蒸馏一种有前景的方法。

查看 arXiv 页面 查看 PDF 项目页面 GitHub 添加到收藏

在您的智能体中获取此论文:

hf papers read 2610.19820

没有最新版 CLI?curl -LsSf https://hf.co/cli/install.sh | bash

引用此论文的模型:0

暂无模型关联此论文

在模型的 README.md 中引用 arxiv.org/abs/2610.19820,即可从此页面关联该论文。

引用此论文的数据集:0

暂无数据集关联此论文

在数据集的 README.md 中引用 arxiv.org/abs/2610.19820,即可从此页面关联该论文。

引用此论文的 Spaces:1

包含此论文的收藏:0

暂无收藏包含此论文

将此论文添加到收藏,即可从此页面关联该论文。

相似文章

GraphOPD:面向 LLM Agent 的图增强同策略蒸馏

arXiv cs.LG

GraphOPD 提出首个将图结构增强引入 on-policy 蒸馏的 agent 训练方法,利用环境状态变化构建步骤依赖图并结合随机游走平稳分布进行信用分配,在 ALFWorld、WebShop、SearchQA 上较最强基线提升最高 5.8 个百分点。

面向智能体在线策略蒸馏的随机教师干预

arXiv cs.CL

提出STI-OPD框架,通过基于教师-学生策略差异的随机教师干预和重要性加权反向KL目标,解决多轮agent任务中on-policy蒸馏的轨迹漂移问题,在工具推理与长程交互基准上超越先前最强OPD基线。