Self-OPD:无需教师的流匹配模型同策略蒸馏

Hugging Face Daily Papers 论文

摘要

Self-OPD 引入了一种无需教师的同策略蒸馏框架,用于流匹配模型。该框架利用自探索的随机分支和归一化优势来优化速度场,在多目标对齐方面超越了先前的方法。

同策略蒸馏 (OPD) 利用预训练的专用教师模型提供密集监督信号,在大型语言模型 (LLMs) 中取得了显著成功,并最近被应用于流匹配模型。然而,这种范式存在两个主要问题:首先,为每个新目标训练单独的任务特定教师会带来高昂的计算成本。其次,教师和学生分布之间的差异通常会导致生成轨迹中的误差累积。本文中,我们介绍了 Self-OPD,一种用于流匹配模型的无需教师的 OPD 框架,它将学生自身的自探索转化为逐步监督。在每个时间步,Self-OPD 将确定性的下一状态预测分支为 K 个随机 SDE 候选,使用 ODE 采样器进行推演,并将其奖励与确定性自参考基线进行比较以获得归一化优势。速度场通过全分支推拉目标进行优化,其中高优势分支吸引学生,低优势分支在方向感知衰减和 SDE 方差归一化下排斥它。对于多目标对齐,Self-OPD 在奖励级别融合归一化分数,避免直接梯度冲突。在单奖励和混合奖励基准上的实验表明,Self-OPD 在没有任务特定教师的情况下,超越了先前的 RL 和 OPD 方法。
查看原文
查看缓存全文

缓存时间: 2026/08/28 03:24

论文页面 - Self-OPD:无需教师模型的流动匹配模型在策略蒸馏

来源:https://huggingface.co/papers/2608.26872 作者: ,

,

,

,

,

,

,

,

,

摘要

Self-OPD 通过使用自探索的随机分支与归一化优势函数来优化速度场,以实现多目标对齐,从而在流动匹配模型中无需任务特定的教师模型。

在策略蒸馏(https://huggingface.co/papers?q=On-policy%20distillation)(OPD)利用预训练的、专门的教师模型来提供密集监督信号,在大语言模型(LLMs)中取得了显著成功,并且最近已被应用于流动匹配(https://huggingface.co/papers?q=flow%20matching)模型。然而,这种范式存在两个主要问题:首先,为每个新目标训练一个单独的、任务特定的教师模型会导致高昂的计算成本。其次,教师与学生分布之间的差异通常会导致沿着生成轨迹产生累积误差。在本文中,我们引入了Self-OPD(https://huggingface.co/papers?q=Self-OPD),一个用于流动匹配(https://huggingface.co/papers?q=flow%20matching)模型的无教师OPD框架,它将学生自身的自探索转化为逐步监督。在每个时间步,Self-OPD(https://huggingface.co/papers?q=Self-OPD)将确定性的下一状态预测分支为K个随机SDE候选(https://huggingface.co/papers?q=SDE%20candidates),使用ODE采样器(https://huggingface.co/papers?q=ODE%20sampler)对其进行采样,并将它们的奖励与确定性的自参考基线进行比较,以获得归一化优势(https://huggingface.co/papers?q=normalized%20advantages)。速度场通过一个全分支拉推目标(https://huggingface.co/papers?q=pull-push%20objective)进行优化,其中高优势分支在方向感知衰减(https://huggingface.co/papers?q=direction-aware%20attenuation)和SDE方差归一化(https://huggingface.co/papers?q=SDE-variance%20normalization)下吸引学生,而低优势分支则排斥学生。为了实现多目标对齐(https://huggingface.co/papers?q=multi-objective%20alignment),Self-OPD(https://huggingface.co/papers?q=Self-OPD)在奖励层面融合归一化分数,避免了直接的梯度冲突。在单一和混合奖励基准上的实验表明,Self-OPD(https://huggingface.co/papers?q=Self-OPD)无需任务特定教师,其性能优于先前的强化学习和OPD方法。

查看arXiv页面(https://arxiv.org/abs/2608.26872)查看PDF(https://arxiv.org/pdf/2608.26872)添加到收藏(https://huggingface.co/login?next=%2Fpapers%2F2608.26872)

在你的代理中获取此论文:

hf papers read 2608\.26872

没有最新的CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash

引用此论文的模型0

无模型链接此论文

在模型README.md中引用arxiv.org/abs/2608.26872以从本页链接它。

引用此论文的数据集0

无数据集链接此论文

在数据集README.md中引用arxiv.org/abs/2608.26872以从本页链接它。

引用此论文的空间0

无空间链接此论文

在空间README.md中引用arxiv.org/abs/2608.26872以从本页链接它。

包含此论文的收藏集0

无收藏集包含此论文

将此论文添加到收藏集(https://huggingface.co/new-collection)以从本页链接它。

相似文章

Flow-OPD:用于流匹配模型的对策蒸馏

Hugging Face Daily Papers

Flow-OPD 是一篇研究论文,介绍了一种用于流匹配文生图模型的两阶段对策蒸馏框架。基于 Stable Diffusion 3.5 Medium,该框架显著提升了生成质量和对齐指标。

Any-OPD:基于表示空间桥接的异构同策略蒸馏用于流匹配模型

Hugging Face Daily Papers

Any-OPD 提出了首个在任意潜在流匹配生成器之间进行同策略蒸馏的框架,通过冻结的视觉表示进行桥接,实现了从 12B FLUX 模型到 2.5B SD3.5 模型的蒸馏。它将学生的 PickScore 从 0.846 提升到 0.884,在仅为教师模型五分之一规模的情况下与之媲美。

OPMD:超越输出监督的在线策略机制蒸馏

Hugging Face Daily Papers

本文提出了在线策略机制蒸馏(OPMD),在在线策略蒸馏过程中迁移与学生模型架构兼容的教师中间计算,在多种师生组合中均超越了仅基于输出监督的蒸馏基线。

DanceOPD:基于策略的生成场蒸馏

Hugging Face Daily Papers

DanceOPD提出了一种基于策略的生成场蒸馏框架,用于流匹配模型。该框架通过能力特定路由和基于速度的训练,统一了文本到图像生成、局部编辑和全局编辑,在保持基准生成质量的同时,提升了多能力组合。