无需教师模型的流匹配训练框架,用自探索实现逐步优化。
Self-OPD: On-Policy Distillation for Flow Matching Models without Teacher

- 用学生自身探索生成多路径,通过奖励比较实现自监督
- 在单任务与多任务基准上超越已有强化学习与师生蒸馏方法
- 避免教师-学生分布差异,适合需要快速适配新目标的场景
基于策略蒸馏(OPD)的流匹配模型虽取得进展,但需为每个新任务训练专用教师模型,计算成本高;且师生分布差异易导致生成轨迹误差累积。本文提出自监督式OPD(Self-OPD),无需外部教师,将学生自身探索转化为逐时步监督信号。每一步将确定性预测分支为K个随机SDE候选路径,使用ODE采样器展开,并与确定性自参考基线比较获得归一化优势。通过全分支拉-推目标优化速度场:高优势分支吸引学生,低优势分支被排斥,结合方向感知衰减与SDE方差归一化。多目标对齐时,在奖励层面融合归一化得分,避免梯度冲突。在单任务与混合奖励基准测试中,Self-OPD优于现有强化学习与OPD方法,且无需任务专属教师。
原文摘要 · Abstract (English)
On-policy distillation (OPD), which leverages a pre-trained, specialized teacher model to provide dense supervisory signals, has achieved significant success in Large Language Models (LLMs) and has recently been adapted to flow matching models. However, this paradigm suffers from two major issues: First, training a separate, task-specific teacher for every new objective incurs high computational costs. Second, the discrepancy between teacher and student distributions often leads to compounding errors along the generation trajectory. In this paper, we introduce \textbf{Self-OPD}, a teacher-free OPD framework for flow matching models that turns the student's own self-exploration into step-wise supervision. At each timestep, Self-OPD branches the deterministic next-state prediction into $K$ stochastic SDE candidates, rolls them out with the ODE sampler, and compares their rewards against a deterministic self-reference baseline to obtain normalized advantages. The velocity field is optimized with an all-branch pull-push objective, where high-advantage branches attract the student and low-advantage branches repel it under direction-aware attenuation and SDE-variance normalization. For multi-objective alignment, Self-OPD fuses normalized scores at the reward level, avoiding direct gradient conflict. Experiments on single and mixed reward benchmarks show that Self-OPD outperforms prior RL and OPD methods without task-specific teachers.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。