arXiv:2511.03317cs.CV2025-11被引 6

解决扩散模型对齐中偏好优化的副作用问题,提升生成质量

Diffusion-SDPO: Safeguarded Direct Preference Optimization for Diffusion Models

  • 提出自适应梯度缩放机制,保护优选输出不被劣化
  • 在多个基准上实现美学、提示对齐等指标显著提升
  • 方法简单通用,适合现有偏好学习框架直接集成

文本到图像扩散模型生成质量高,但与人类偏好对齐仍具挑战。我们重新审视基于扩散的直接偏好优化(Diffusion-DPO),发现增大偏好间距并不必然提升生成质量:标准目标会导致优/劣分支重建误差同时增加,劣质输出严重退化时,优选分支也会受损。为此,我们提出 Diffusion-SDPO,通过自适应调整劣分支梯度强度,使其与优分支梯度对齐,从而保护优选输出。一阶分析得到闭式缩放系数,确保每步优化中优选输出误差非递增。该方法简单、模型无关,兼容现有 DPO 框架,计算开销极小。在标准文本到图像基准上,相较于基线,本方法在自动化偏好、美学和提示对齐等指标上均取得一致提升。

原文摘要 · Abstract (English)

Text-to-image diffusion models deliver high-quality images, yet aligning them with human preferences remains challenging. We revisit diffusion-based Direct Preference Optimization (DPO) for these models and identify a critical pathology: enlarging the preference margin does not necessarily improve generation quality. In particular, the standard Diffusion-DPO objective can increase the reconstruction error of both winner and loser branches. Consequently, degradation of the less-preferred outputs can become sufficiently severe that the preferred branch is also adversely affected even as the margin grows. To address this, we introduce Diffusion-SDPO, a safeguarded update rule that preserves the winner by adaptively scaling the loser gradient according to its alignment with the winner gradient. A first-order analysis yields a closed-form scaling coefficient that guarantees the error of the preferred output is non-increasing at each optimization step. Our method is simple, model-agnostic, broadly compatible with existing DPO-style alignment frameworks and adds only marginal computational overhead. Across standard text-to-image benchmarks, Diffusion-SDPO delivers consistent gains over preference-learning baselines on automated preference, aesthetic, and prompt alignment metrics. Code is publicly available at https://github.com/AIDC-AI/Diffusion-SDPO.

扩散模型偏好优化生成质量

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