Self-OPD: On-Policy Distillation for Flow Matching Models without Teacher
Abstract Overview
This paper presents Self-OPD, a teacher-free on-policy distillation framework for flow matching models. Instead of relying on a separate task-specific teacher, the method branches the model’s deterministic next-state prediction into multiple stochastic SDE candidates, rolls them out, and scores them against a deterministic self-reference baseline. The resulting normalized branch advantages provide dense step-wise supervision through an all-branch pull-push distillation objective that attracts the model toward better local trajectories and repels it from worse ones with direction-aware attenuation. The paper also introduces reward-level fusion for multi-objective alignment, combining multiple rewards for branch ranking without directly mixing parameter-space gradients.
Novelty
The key novelty is eliminating external teacher models in OPD by converting the student's own local SDE exploration and a deterministic self-reference baseline into dense step-wise supervision. This is paired with an all-branch pull-push distillation objective featuring direction-aware attenuation and reward-level fusion that avoids gradient conflicts during multi-objective alignment.
Results
Across single-reward and mixed-reward benchmarks spanning text rendering, compositional generation, and preference alignment, Self-OPD outperforms prior teacher-free RL baselines and matches or exceeds teacher-based OPD methods. In the mixed-reward setting, Self-OPD achieves 0.9521/0.9691 on GenEval strict/continuous, 0.9597 on OCR, and 23.87 PickScore with 0.3214 HPSv2 on the same test images, outperforming DiffusionOPD on these joint preference metrics while reducing overall wall-clock training time.
Key Points
- Self-OPD turns local SDE branching combined with a deterministic self-reference rollout into dense per-step supervision without requiring a task-specific teacher.
- The all-branch pull-push loss optimizes over both positive and negative branches, incorporating KL-motivated transition variance normalization and direction-aware attenuation to prevent gradient interference.
- Reward-level fusion handles multi-objective alignment by ranking sampled trajectories with composite normalized rewards, achieving strong simultaneous performance across GenEval, OCR, and preference benchmarks.
References
- arXiv: https://arxiv.org/abs/2608.26872v1
- Fugu-MT: https://fugumt.com/fugumt/paper_check/2608.26872v1
- Hugging Face Papers: https://huggingface.co/papers/2608.26872
- GitHub: https://github.com/Shiy-Zhang/Self-OPD