游客发表

这项突破性研究由北京大学、北京小米大模型团队、大学多项香港大学及中国人民大学联合完成。小米相关论文已于2026年6月29日上传至预印本平台arXiv,联合编号为arXiv:2606.30406。研究院出读者可通过该编号获取完整技术细节。手让
你是掌握否曾设想:若将厨艺、修车与写作三位专家的北京技能融合于一人,该如何实现?大学多项直觉上,轮流或同时教学似乎可行,小米但现实往往是联合“学了新忘旧”或“样样通样样松”。这正是研究院出当前大型语言模型(LLM)面临的核心瓶颈。研究团队将此定义为“能力融合”(Capability Integration),手让并提出了全新解决方案——MOPD(多教师在线蒸馏,掌握Multi-Teacher On-Policy Distillation)。北京
现代AI训练主要依赖强化学习(RL):通过奖励机制引导模型学习。目前,数学推理、代码生成、指令遵循等任务通常拥有独立的“训练场”和“奖励规则”,能培养出各领域的“专科医生”级模型。然而,用户更需要的是无所不能的“全科大夫”。
在MOPD出现前,业界主要尝试过四种路径,但均存在显著缺陷:
MOPD的设计灵感源自吉他自学场景:学生即兴演奏(自主生成),老师逐音符纠正(密集反馈),而非背诵老师示范曲。
其两大核心技术优势:
1. 在线训练(On-Policy):学生基于自身生成的内容进行学习,训练状态与推理状态一致,消除暴露偏差。
2. 密集优化信号(Dense Optimization):提供逐词级别的反馈,而非仅给出最终对错,信号更丰富、更精准。
MOPD将训练过程拆解为三个精密协作的阶段:
使用涵盖多领域的宽泛数据对基础模型进行微调,获得一个具备基础能力的“底子”模型(SFT Checkpoint)。这是后续所有专家和学生模型的共同起点。
基于SFT模型,独立并行训练各领域专家:
* 数学专家:采用可验证答案的奖励机制。
* 软件工程专家:采用代码可执行沙箱奖励。
* 指令遵循专家:采用评分标准奖励。
* 特点:各专家独立训练,互不干扰,如同不同学科老师各自备课。
这是核心融合阶段:
* 角色设定:SFT模型为“学生”,冻结的各专家模型为“老师”。
* 流程:
1. 学生从题库随机抽取题目,生成回答并记录词级概率分布。
2. 回答分发至对应领域的专家老师,老师生成其概率分布。
3. 计算师生分布差异,以此信号更新学生模型。
* 结果:生成一个融合所有领域技能的统一模型。
MOPD使用反向KL散度(Reverse KL Divergence)衡量学生与老师的分布差异。优化目标是让学生在每个词的选择上无限接近老师。
研究团队提供了两种实现方式:
直觉上,多老师同时打分会增加延迟。MOPD通过异步打分机制解决此问题:
* 将专家模型部署为独立的后台“打分服务”。
* 学生生成下一批数据时,后台并行处理上一批数据的打分请求。
* 效果:打分时间被完全“隐藏”在学生生成的等待间隙中,实测几乎不增加额外训练耗时。
在Qwen3-30B-A3B模型上的对比实验(涵盖数学、指令遵循、软件工程)显示:
参数平均合并:0.328(几乎失败)
均衡性:MOPD在三个领域的分数集中在0.91-0.95,差异极小(0.044),实现了真正的“全面均衡”。相比之下,级联RL和离线微调在不同领域间波动巨大。
样本效率:
研究团队将MOPD应用于小米内部工业级模型MiMo-V2-Flash(约3090亿参数,混合专家架构),覆盖数学、代码、指令、工程、工具使用五大领域。
研究团队进行了两组关键对照实验:
由于师生分布高度重合(KL散度仅0.04),两种方法均能获得有效信号,收敛结果相似。
外部大模型作为老师的失败案例:
MOPD支持迭代升级:
1. 第一轮MOPD结束后,学生已接近专家水平。
2. 将该学生作为新起点,重新训练领域专家(专家水平进一步提升至1.030)。
3. 基于更强专家进行第二轮MOPD蒸馏。
4. 结果:学生归一化分数从0.937提升至0.986,形成“学生变强->老师变强->学生更强”的良性循环。
MOPD不仅提升了性能,更重构了AI研发流程:
* 传统模式:串行或强耦合,任一环节失误影响全局,团队间互相掣肘。
* MOPD模式:能力生产(专家训练)与能力融合(MOPD蒸馏)完全解耦。
* 各领域团队可独立并行迭代,使用各自最优算法和超参数。
* 单一领域失败不影响其他领域。
* 最终统一接入MOPD完成融合,大幅提升团队开发吞吐量。
核心洞察:真正的学习发生在学习者自主尝试并对照高标准修正的过程中,而非被动模仿。MOPD让每个专家专注极致,让学生通过在线练习内化精华,避免了多任务训练中的相互拉扯。
Q1:MOPD与直接平均合并专家权重有何区别?
A:参数平均合并无需训练但极不稳定(实验得分仅0.328)。MOPD在策略空间(而非权重空间)通过让学生接受多专家逐词指导完成融合,因此更稳定、效果更全面。
Q2:多专家同时运行是否会导致计算开销过大?
A:不会。MOPD采用异步打分机制,专家打分服务并行运行在学生生成数据的等待间隙中。实测显示,专家服务几乎不增加额外训练耗时,整体速度由学生采样速度决定。
Q3:为何不能用更强的外部大模型当老师?
A:实验证明,外部大模型与学生分布差异过大,导致学生生成内容落入老师认为“概率极低”的区域,引发大量惩罚信号,造成策略崩溃。只有“同根同源”的师生分布接近,才能提供清晰有效的优化信号。
随机阅读
热门排行
友情链接