Open-MOPD:诊断与修复多教师在策略蒸馏中的能力不平衡
摘要
本文介绍了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:隐空间多教师同策略蒸馏
论文提出了 Latent-MOPD,这是首个面向大语言模型的表示层面多教师同策略蒸馏方法,无需额外的教师训练,即可同时通过输出分布和隐藏状态整合多个专业教师模型;在涵盖数学、代码和逻辑推理的九个基准测试中,其表现优于仅基于 token、仅基于表示以及平均融合等基线方法。
超越教师分配:领域归一化多教师在策略蒸馏
本文提出了领域归一化多教师在策略蒸馏(DN-MOPD),以解决合并专家语言模型时反馈不平衡的问题,并在数学和指令跟随等基准测试中展示了性能提升。
OPMD:超越输出监督的在线策略机制蒸馏
本文提出了在线策略机制蒸馏(OPMD),在在线策略蒸馏过程中迁移与学生模型架构兼容的教师中间计算,在多种师生组合中均超越了仅基于输出监督的蒸馏基线。
超越策略差异:能力感知的多教师在线蒸馏
本文提出 CA-MOPD,通过合并强化学习专家并利用验证器估算的师生能力差距来调整多教师在线蒸馏的监督强度,在数学、代码生成和指令遵循六个基准上超越现有 MOPD 方法。
基于同伴成功与失败的多 rollout 在策略蒸馏
提出多 rollout 在策略蒸馏 (MOPD),一种将教师条件化于同伴成功和失败的 rollout 以提供更密集的 token 级监督进行语言模型后训练的方法,在多个基准上提升了性能。