arXiv:2608.24646cs.CV2026-08

将图像奖励转化为中间去噪目标,提升扩散模型对齐效率。

On-Policy Self-Distillation in Diffusion Models

  • 用自蒸馏构建中间预测的显式目标,实现在线策略优化。
  • 在20个场景中19次超越基准,最高提升44.0%。
  • 训练效率更高,节省40%~63% GPU时长,适合高效对齐任务。

强化学习可使扩散模型对齐人类偏好与任务目标,但终点奖励无法指导中间去噪预测的调整。本文提出DiffusionOPSD,一种基于策略的自蒸馏框架,将图像级奖励引导转化为采样查询处的干净输出预测显式目标。每个外循环中,冻结的行为策略生成轨迹并提供查询状态与锚点;奖励梯度在各锚点周围构建有界正负目标。可训练策略通过有限拟合,以分离监督方式逼近这些目标,随后通过指数移动平均更新行为策略。该设计允许独立评估目标构建与实际实现效果。同查询控制实验表明,更大的目标构建增益并不必然带来更大的实现增益。在SD 3.5-M和步长蒸馏的Z-Image-Turbo上,本方法在两种骨干网络、十名评估者共20个匹配奖励设置中,有19次取得最优保持分数。相比最强基线最高提升44.0%,在SD 3.5-M上减少40%的训练GPU小时,在Z-Image-Turbo上减少63%。结果表明,基于策略的自蒸馏是一种高效且可分析的扩散模型后训练对齐方法。

原文摘要 · Abstract (English)

Reinforcement learning can align diffusion models with human preferences and task-specific objectives, but endpoint rewards do not specify how an intermediate denoising prediction should change. We introduce DiffusionOPSD as an on-policy self-distillation framework that converts image-level reward guidance into explicit targets for clean-output predictions at sampled queries. At each outer iteration, a frozen behavior policy generates trajectories and supplies query states and anchors. Reward gradients construct bounded positive and negative targets around each anchor. The trainable policy fits these targets as detached supervision through finite fitting before an exponential moving average update refreshes the behavior policy. This setup lets us measure target construction and finite realization separately. Controlled same-query experiments show that larger target-construction gains do not necessarily translate into larger realized gains after a single fitting update. Across SD 3.5-M and the step-distilled Z-Image-Turbo, our approach achieves the best final held-out scores in 19 of 20 reward-matched settings across two backbones and ten evaluators. It outperforms the strongest competing method by up to 44.0% and reduces training GPU-hours relative to DiffusionNFT by 40% on SD 3.5-M and 63% on Z-Image-Turbo. These results support on-policy self-distillation as an efficient and analyzable approach to diffusion post-training by converting image-level reward guidance into explicit and continually refreshed intermediate supervision, thereby opening a path toward more efficient and diagnosable alignment.

扩散模型强化学习自蒸馏对齐优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。