FuguReport

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

Authors Shiyi Zhang, Mushui Liu, Yunze Tong, Wanggui He, Siyu Zou, Jinlong Liu, Yunlong Yu, Jian Song, Hao Jiang, Pipei Huang, Bo Zheng
Affiliations Alibaba / Zhejiang University / Tsinghua University
Categories Method / Self-Distillation / On-policy distillation framework, Evaluation / Benchmarking / Performance comparison without teacher, Application / Flow Matching Models / RL methods without task-specific teacher
License CC BY 4.0

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

  1. Self-OPD turns local SDE branching combined with a deterministic self-reference rollout into dense per-step supervision without requiring a task-specific teacher.
  2. 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.
  3. 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

This page was created using generative AI such as GPT-5, Claude Opus 4, Gemini 3, Gemini 3.1 Flash Image, and their higher-end successor versions. No guarantee can be made regarding its contents.