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

Hugging Face Daily Papers Papers

Summary

Self-OPD introduces a teacher-free on-policy distillation framework for flow matching models that uses self-explored stochastic branches and normalized advantages to optimize velocity fields, outperforming prior methods in multi-objective alignment.

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 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.
Original Article
View Cached Full Text

Cached at: 08/28/26, 03:24 AM

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

Source: https://huggingface.co/papers/2608.26872 Authors:

,

,

,

,

,

,

,

,

,

Abstract

Self-OPD eliminates task-specific teachers in flow matching by using self-explored stochastic branches and normalized advantages to optimize the velocity field for multi-objective alignment.

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 toflow matchingmodels. 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 introduceSelf-OPD, a teacher-free OPD framework forflow matchingmodels that turns the student’s own self-exploration into step-wise supervision. At each timestep,Self-OPDbranches the deterministic next-state prediction into K stochasticSDE candidates, rolls them out with theODE sampler, and compares their rewards against a deterministic self-reference baseline to obtainnormalized advantages. The velocity field is optimized with an all-branchpull-push objective, where high-advantage branches attract the student and low-advantage branches repel it underdirection-aware attenuationandSDE-variance normalization. Formulti-objective alignment,Self-OPDfuses normalized scores at the reward level, avoiding direct gradient conflict. Experiments on single and mixed reward benchmarks show thatSelf-OPDoutperforms prior RL and OPD methods without task-specific teachers.

View arXiv pageView PDFAdd to collection

Get this paper in your agent:

hf papers read 2608\.26872

Don’t have the latest CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

Models citing this paper0

No model linking this paper

Cite arxiv.org/abs/2608.26872 in a model README.md to link it from this page.

Datasets citing this paper0

No dataset linking this paper

Cite arxiv.org/abs/2608.26872 in a dataset README.md to link it from this page.

Spaces citing this paper0

No Space linking this paper

Cite arxiv.org/abs/2608.26872 in a Space README.md to link it from this page.

Collections including this paper0

No Collection including this paper

Add this paper to acollectionto link it from this page.

Similar Articles

Flow-OPD: On-Policy Distillation for Flow Matching Models

Hugging Face Daily Papers

Flow-OPD is a research paper introducing a two-stage on-policy distillation framework for Flow Matching text-to-image models, significantly improving generation quality and alignment metrics using Stable Diffusion 3.5 Medium.

DanceOPD: On-Policy Generative Field Distillation

Hugging Face Daily Papers

DanceOPD proposes an on-policy generative field distillation framework for flow-matching models that unifies text-to-image generation, local editing, and global editing via capability-specific routing and velocity-based training, improving multi-capability composition while preserving anchor generation quality.