On-Policy蒸馏与Off-Policy微调对比

作者:暴富20212026.11.04 10:19浏览量:0

简介:本文围绕大模型后训练中的两类核心方法展开对比:On-Policy蒸馏与Off-Policy监督微调。读者将理解二者在奖励密度、分布偏移、训练效率与适用场景上的关键差异,并掌握面向小模型专业化训练的选型依据、迁移成本与使用边界,为实际训练方案设计提供可操作的判断框架。

On-Policy蒸馏与Off-Policy微调:大模型后训练方法的系统对比

在大语言模型的能力构建过程中,训练通常被划分为几个前后衔接的阶段:预训练负责建立通用语言能力、广泛推理能力与世界知识;中训练注入领域知识,例如代码、医疗数据库或企业内部文档;后训练则引导出目标行为,例如指令遵循、数学推理或对话交互。当模型进入特定专业领域后,一个经过更强训练的较小模型往往能够胜过规模更大的通用模型。小模型的价值在于可本地部署以满足隐私与安全要求、更容易持续训练和更新、推理成本更低。但要兑现这些优势,后训练阶段的方法选择至关重要。

本文对比的两类对象是:On-Policy蒸馏与Off-Policy监督微调(SFT)。前者从学生模型自身采样的轨迹出发,由教师模型提供密集的过程奖励;后者依赖外部来源的目标输出,让学生模型学习模仿。二者共同服务于“将强模型能力迁移到紧凑模型”的目标,但在反馈密度、分布一致性与工程代价上存在本质差异。

一、对比背景:稀疏奖励为何成为瓶颈

以训练一个紧凑模型解决数学问题为例。若采用强化学习式的On-Policy训练,学生模型对每道题采样轨迹,再根据最终答案是否正确给予评分。这种评分可由人工完成,也可由能够可靠得出正确答案的教师模型完成。其优势在于:学生在自己产生的样本上学习,能更直接地学会规避错误路径。

但强化学习的主要缺点是反馈稀疏。无论一个训练回合消耗多少token,每个回合所传授的信息量基本固定。学生知道“21”是错误答案,于是更新参数以避免重复该轨迹,却无法得知究竟错在运算顺序还是算术本身。这种稀疏性使强化学习在许多应用中效率偏低——错误被惩罚了,但错误的结构没有被解释。

Off-Policy训练通常通过监督微调完成:在一组精选的、特定任务的标注示例上训练,示例可来自在当前任务上表现良好的教师模型。进一步地,可以采用蒸馏,训练学生匹配教师的输出分布,即在教师轨迹(包含中间思考步骤的完整token序列)上训练,使用教师每一步的完整下一token分布(logit蒸馏),或仅采样给定序列。这条路线信息密度高,但引入了分布偏移风险:学生被训练去模仿它自己未必会产生的轨迹,推理时一旦偏离教师分布,就可能出现SFT风格的OOD冲击。

On-Policy蒸馏的提出正是为了调和这一矛盾:把教师模型当作过程奖励模型,在學生自身推演过程中提供密集奖励,同时避免纯SFT式的分布外冲击。

二、对象定义与相同点

On-Policy蒸馏:训练数据来自学生模型自身的rollout,教师模型对学生轨迹的每一步或每个中间状态给出细粒度评价,形成密集奖励信号,学生据此更新。其技术逻辑是“在自己的分布上接受专家点评”。

Off-Policy监督微调:训练数据来自外部教师或人工标注的目标输出,学生通过最大似然或分布匹配去拟合这些输出。其技术逻辑是“在专家的分布上模仿专家行为”。

二者的相同点包括:目标一致,都是把更强模型的能力迁移到更小或更专用的模型;都依赖教师信号,只是信号形式不同;都处于后训练阶段,需建立在中训练获得的领域知识之上;都可用于指令遵循、数学推理、领域问答等任务;都需要关注数据质量、训练稳定性与过拟合风险。

三、核心差异分析

反馈密度与信息效率。Off-Policy SFT的每个token都有明确目标,监督信号密集;On-Policy强化学习只有回合级奖励,信号稀疏。On-Policy蒸馏介于两者之间:它保留On-Policy的采样分布,但把教师评价下沉到过程级,使每个回合携带的比特数显著提升。这意味着在相同token预算下,On-Policy蒸馏可能获得更高的训练效率,但前提是教师能给出可靠的过程评价。

分布一致性与OOD风险。Off-Policy训练的数据分布由教师决定,学生推理时自回归产生的分布若与教师轨迹偏离,就会出现暴露偏差,表现为SFT风格的OOD冲击。On-Policy蒸馏的训练分布由学生自己产生,教师只负责评价,因此训练与推理的分布更一致,避免“模仿未经历状态”的问题。代价是采样成本落在学生侧,且需要教师在线或准在线参与。

实现复杂度与工程依赖。SFT实现简单:准备数据、定义损失、离线训练即可,对训练框架要求低。On-Policy蒸馏需要rollout生成、教师前向评价、奖励聚合与策略更新的协同,工程链路更长,对推理吞吐和教师调用成本更敏感。若教师以API方式提供,还需考虑延迟与配额。

能力覆盖与使用边界。SFT擅长传授格式、风格、既定解题模式和可验证输出;对“为何错、错在哪”的过程性知识传递较弱。On-Policy蒸馏擅长纠正常见错误路径、提升推理鲁棒性,但依赖教师的过程评价能力;若教师只能判断最终答案,则退化为稀疏奖励。二者都不是万能的:SFT不适合需要探索新策略的任务,On-Policy蒸馏不适合教师信号不可靠或采样成本极高的场景。

成本结构。SFT的成本主要在数据构建与离线训练;On-Policy蒸馏的成本主要在在线采样与教师评价,长期看可能因样本效率更高而摊薄,但初期工程投入更大。二者都需考虑持续训练与版本更新带来的维护成本。

四、对比表格

维度 On-Policy蒸馏 Off-Policy监督微调
数据来源 学生自身rollout 教师或人工标注输出
奖励形式 过程级密集奖励 token级目标分布
分布一致性 高,训练即推理分布 存在暴露偏差风险
信息效率 较高,依赖教师过程评价 高,但过程知识弱
实现复杂度 较高,需采样与评价闭环 较低,离线训练为主
主要成本 在线采样与教师调用 数据构建与离线算力
典型风险 教师评价不可靠、采样开销 OOD冲击、模仿表面模式
适用任务 推理纠错、鲁棒性提升 格式对齐、风格与模式传授

五、典型场景与选型建议

若团队需要快速让紧凑模型掌握既定输出格式、客服话术或结构化抽取模式,且已有高质量标注或教师输出,Off-Policy SFT是接入成本更低的起点。若目标是提升数学推理、代码修复等过程的正确率,且教师能够对学生中间步骤给出可靠评价,On-Policy蒸馏更有利于纠正错误路径。若教师只能判断最终答案,可先以SFT建立基础能力,再以On-Policy蒸馏做过程级强化,形成分阶段组合。若采样预算紧张或教师调用受限,应优先评估SFT或离线蒸馏的可行性。选型不应绝对化:在高并发推理部署诉求下,小模型本身是共同前提,差异在于训练侧的资源结构。

六、迁移与使用注意事项

从SFT迁移到On-Policy蒸馏,需评估:教师评价接口的稳定性与延迟;rollout采样对推理资源的占用;奖励聚合逻辑是否正确处理中间步骤;训练与推理框架是否支持在线闭环;权限与数据隔离是否满足合规要求。反向迁移则需关注教师数据授权、数据清洗与去重、以及离线数据与学生分布不匹配带来的OOD风险。无论哪种路线,都应保留验证集上的过程指标与最终指标,避免仅看训练损失。

七、总结

On-Policy蒸馏与Off-Policy监督微调的核心差异在于“在谁的分布上学习、以多密的信号学习”。SFT以密集目标换取实现简单,但承担分布偏移风险;On-Policy蒸馏以在线采样换取分布一致与过程纠错能力,但工程链路更长。决策时应围绕教师信号可靠性、采样成本、任务是否依赖过程推理、团队运维能力四个条件展开,而非简单判断孰优孰劣。