Open-MOPD:诊断与修复多教师在策略蒸馏中的能力不平衡

Hugging Face Daily Papers 论文

摘要

本文介绍了Open-MOPD,这是一个通过平衡token级预算来诊断和修复多教师在策略蒸馏中能力不平衡的框架,通过动态分配和奖励刷新,将头空间恢复率从35.6%提高到83.4%。

多教师在策略蒸馏(M-OPD)已成为一个有前景的范式,通过密集的、token级奖励监督,将领域专业的强化学习(RL)专家整合到一个单一的通才学生模型中。尽管其在实际中取得成功,但控制多教师能力整合的优化动态仍知之甚少,开放且严格可复现的配方明显缺乏。在这项工作中,我们在SmolLM3-3B-Base上建立了一个受控的M-OPD基准,使用神谕路由,将能力整合与路由歧义隔离。我们的调查揭示了一个显著的能力整合差距:标准M-OPD仅捕获了相对于领域路由神谕集成35.6%的可用头空间,像指令遵循这样的简洁任务遭受严重退化和过早停滞。关键的是,我们表明这种失败并非源于梯度冲突,而是源于token级优化预算的严重错误分配。这种病理由三个正交因素驱动:跨领域的结构性序列长度差异、由于非均匀学习率导致的动态收敛漂移,以及异步策略更新引起的多步奖励过时。为了解决这些不平衡,我们引入了Open-MOPD,一个原则性框架,包含token共享平衡、差距感知动态预算分配和学生奖励刷新。这些机制共同系统地恢复了跨领域平衡,将头空间恢复率从35.6%提高到83.4%,在一个单一的可部署学生模型中。我们完全开源了我们的端到端后训练配方、训练轨迹和评估套件,基于学术可访问的硬件预算。
查看原文
查看缓存全文

缓存时间: 2026/08/27 07:20

论文页面 - Open-MOPD:诊断与修复多教师在策略蒸馏中的能力失衡问题

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

摘要

多教师在策略蒸馏存在跨领域令牌级预算分配不均的问题,本文通过平衡机制、动态分配与奖励刷新方法解决了这一问题,恢复了大部分预言机集成模型的能力。

多教师在策略蒸馏(https://huggingface.co/papers?q=Multi-teacher%20on-policy%20distillation)(M-OPD)已成为一个极具前景的范式,它通过密集的令牌级奖励监督(https://huggingface.co/papers?q=token-level%20reward%20supervision),将领域专家强化学习(https://huggingface.co/papers?q=reinforcement%20learning)(RL)模型整合为一个通用学生模型。尽管其实际效果显著,但支配多教师能力整合(https://huggingface.co/papers?q=capability%20integration)的优化动态仍鲜为人知,且明显缺乏开放、可严格复现的技术方案。本研究在 SmolLM3-3B-Base 模型上建立了一个受控的 M-OPD 基准测试,并使用预言机路由,将能力整合(https://huggingface.co/papers?q=capability%20integration)与路由歧义隔离开来。我们的研究发现了一个显著的能力整合差距:标准 M-OPD 仅能捕获相对于领域路由预言机集成 35.6% 的可用提升空间,尤其在简洁任务(如指令遵循)上表现出严重的性能下降和过早停滞。关键在于,我们证明这种失败并非源于梯度冲突(https://huggingface.co/papers?q=gradient%20conflict),而是由于令牌级优化预算的严重错配所致。这种病理现象由三个正交因素驱动:跨领域的结构性序列长度差异、非均匀学习率导致的动态收敛漂移,以及异步策略更新引起的多步奖励过时问题。为解决这些失衡问题,我们引入了 Open-MOPD,一个基于原则的框架,它整合了令牌份额平衡(https://huggingface.co/papers?q=token-share%20balancing)、基于差距感知的动态预算分配(https://huggingface.co/papers?q=gap-aware%20dynamic%20budget%20allocation)和学生奖励刷新(https://huggingface.co/papers?q=student%20reward%20refresh)机制。这些机制协同工作,系统地恢复了跨领域平衡,将单个可部署学生模型的提升空间恢复率从 35.6% 提高到了 83.4%。我们在学术可及的硬件预算内,完全开源了端到端的后训练方案、训练轨迹和评估套件。

查看 arXiv 页面 (https://arxiv.org/abs/2608.19098) 查看 PDF (https://arxiv.org/pdf/2608.19098) 项目主页 (https://bytedtsinghua-sia.github.io/Open-MOPD/) GitHub43 (https://github.com/BytedTsinghua-SIA/Open-MOPD) 添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2608.19098)

在代理中获取此论文:

hf papers read 2608\.19098

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

引用此论文的模型0

暂无模型链接此论文

在模型 README.md 中引用 arxiv.org/abs/2608.19098 以从此页面链接。

引用此论文的数据集0

暂无数据集链接此论文

在数据集 README.md 中引用 arxiv.org/abs/2608.19098 以从此页面链接。

引用此论文的 Space0

暂无 Space 链接此论文

在 Space README.md 中引用 arxiv.org/abs/2608.19098 以从此页面链接。

包含此论文的合集0

暂无合集包含此论文

将此论文添加到合集 (https://huggingface.co/new-collection) 以从此页面链接。

相似文章

Latent-MOPD:隐空间多教师同策略蒸馏

arXiv cs.LG

论文提出了 Latent-MOPD,这是首个面向大语言模型的表示层面多教师同策略蒸馏方法,无需额外的教师训练,即可同时通过输出分布和隐藏状态整合多个专业教师模型;在涵盖数学、代码和逻辑推理的九个基准测试中,其表现优于仅基于 token、仅基于表示以及平均融合等基线方法。

OPMD:超越输出监督的在线策略机制蒸馏

Hugging Face Daily Papers

本文提出了在线策略机制蒸馏(OPMD),在在线策略蒸馏过程中迁移与学生模型架构兼容的教师中间计算,在多种师生组合中均超越了仅基于输出监督的蒸馏基线。

超越策略差异:能力感知的多教师在线蒸馏

arXiv cs.LG

本文提出 CA-MOPD,通过合并强化学习专家并利用验证器估算的师生能力差距来调整多教师在线蒸馏的监督强度,在数学、代码生成和指令遵循六个基准上超越现有 MOPD 方法。

基于同伴成功与失败的多 rollout 在策略蒸馏

arXiv cs.LG

提出多 rollout 在策略蒸馏 (MOPD),一种将教师条件化于同伴成功和失败的 rollout 以提供更密集的 token 级监督进行语言模型后训练的方法,在多个基准上提升了性能。