arXiv:2606.02521cs.LGcs.CV2026-06

让单步图像生成模型更懂人类偏好,无需反向传播。

Drifting Preference Optimization for One-Step Generative Models

论文配图:Drifting Preference Optimization for One-Step Generative Models
图 1 · 摘自论文原文
  • 用高分低分样本合成更新方向,不依赖梯度或可微奖励。
  • 在多个评测集上提升对齐效果,训练计算量减少3.51倍。
  • 适合部署受限场景,支持黑箱或不可导的奖励函数。

单步文本到图像生成器因仅需一次前向传播即可生成图像而具备部署优势,但对其进行偏好微调仍具挑战:传统对齐方法常依赖策略似然、去噪轨迹、可微奖励梯度或测试时优化。本文提出漂移偏好优化(DrPO),一种针对确定性单步生成器的在线偏好微调方法。对于每个提示词,DrPO从当前生成器采样候选图像,利用目标奖励进行排序,并用高分与低分样本合成特征空间的更新方向。该更新由非参数偶极偏好场与基于冻结基础生成器估计的参考漂移构成,通过解耦的特征空间回归目标进行优化。目标奖励仅用于排序,因此DrPO可在大容量、黑箱或不可导奖励下训练,而推理仍保持单次生成调用。我们在SD-Turbo和SDXL-Turbo上评估了DrPO,使用多种目标奖励与基准,包括HPSv3和GenEval。结果表明,DrPO在无奖励梯度的单步偏好基线中实现更好的对齐性能,并在匹配有效批量设置下将HPSv3训练计算量降低3.51倍,因去除了奖励模型反向传播。初步离线实验表明,基于样本的梯度合成也可拓展至其他场景。

原文摘要 · Abstract (English)

One-step text-to-image generators are attractive for deployment because they generate an image with a single forward pass, but preference finetuning them remains difficult: standard alignment methods often rely on policy likelihoods, denoising trajectories, differentiable reward gradients, or test-time optimization. We propose Drifting Preference Optimization (DrPO), an online preference-finetuning method for deterministic one-step generators. For each prompt, DrPO samples candidates from the current generator, ranks them with a target reward, and uses high- and low-scoring samples to synthesize a feature-space update direction. The update is a non-parametric dipole preference field plus a reference drift estimated from the frozen base generator, and is optimized through a detached feature-space regression target. The target reward is used only for ranking, so DrPO can train with large, black-box, or non-differentiable rewards while inference remains a single generator call. We evaluate DrPO on SD-Turbo and SDXL-Turbo with multiple target rewards and benchmarks, including HPSv3 and GenEval. DrPO improves alignment over reward-gradient-free one-step preference baselines and reduces HPSv3 training computation by $3.51\times$ under the matched effective-batch setting by removing reward-model backpropagation. Initial offline experiments suggest that sample-based gradient synthesis can also be used beyond online reward ranking.

图像生成偏好优化单步生成无梯度训练

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