← 首页|学术|Self-OPD: Teacher-free On-Policy Distillation
cs.CV · 2608.26872 · 2026-08-27

Self-OPD: On-Policy Distillation for Flow Matching Models without Teacher

Shiyi Zhang, Mushui Liu, Yunze Tong, Wanggui He, Siyu Zou, Jinlong Liu, Yunlong Yu, Jian Song, Hao Jiang, Pipei Huang, Bo Zheng
Flow MatchingOn-Policy DistillationRL训练
💬 不需要单独训练教师模型:让学生自己探索、自己打分、自己蒸馏自己,flow matching 模型的 on-policy distillation 摆脱了教师依赖。

🎯 背景

On-policy distillation 在 LLM 和 flow matching 模型上都很有效,但要为每个新目标单独训一个任务专用教师,成本很高;教师和学生分布不匹配还会导致生成轨迹上的误差累积。

🔬 方法

Self-OPD 让确定性的下一状态预测分叉出 K 个随机 SDE 候选,用 ODE 采样器 rollout,再和一个确定性自参照基线比较获得归一化优势;速度场用「全分支拉-推」目标优化,高优势分支吸引学生、低优势分支在方向感知衰减和 SDE 方差归一化下排斥学生。多目标对齐时在 reward 层面融合分数,避免梯度直接冲突。

📊 结果与意义

在单目标和混合 reward 基准上,Self-OPD 全面超过此前依赖任务专用教师的 RL/OPD 方法,且不需要额外训练教师模型。
▶ 原文摘要 Abstract
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.
来源:arXiv:2608.26872 · 精读基于摘要与 arXiv HTML/abs 页信息生成,未解析 PDF 全文