arXiv:2511.07806cs.CV2025-11

用轻量分类器让扩散模型生成更符合人类偏好的图像。

PC-Diffusion: Aligning Diffusion Models with Human Preferences via Preference Classifier

  • 用可训练的偏好分类器直接判断图像优劣,不需全模型微调。
  • 相比DPO,训练成本更低,且不依赖参考模型,结果更稳定。
  • 适合需要高效、稳定图像生成优化的研究与应用。

扩散模型在条件图像生成中表现优异,但输出常与人类偏好不符。现有基于直接偏好优化(DPO)的方法虽有效,但存在两大缺陷:1)需全模型微调,计算开销大;2)对参考模型质量敏感,易引入不稳定性和偏差。为此,我们提出PC-Diffusion框架,通过一个轻量级可训练的偏好分类器,直接建模样本间的相对偏好。该方法将偏好学习限制在分类器中,实现生成模型与偏好对齐解耦,无需全模型微调和参考模型。理论分析表明:1)偏好引导分布可在时间步间一致传播;2)偏好分类器训练目标等价于DPO,但无需参考模型;3)所提偏好修正可逐步引导生成趋向偏好对齐区域。实验显示,PC-Diffusion在偏好一致性上媲美DPO,同时显著降低训练成本,实现高效稳定的偏好引导生成。

原文摘要 · Abstract (English)

Diffusion models have achieved remarkable success in conditional image generation, yet their outputs often remain misaligned with human preferences. To address this, recent work has applied Direct Preference Optimization (DPO) to diffusion models, yielding significant improvements.~However, DPO-like methods exhibit two key limitations: 1) High computational cost,due to the entire model fine-tuning; 2) Sensitivity to reference model quality}, due to its tendency to introduce instability and bias. To overcome these limitations, we propose a novel framework for human preference alignment in diffusion models (PC-Diffusion), using a lightweight, trainable Preference Classifier that directly models the relative preference between samples. By restricting preference learning to this classifier, PC-Diffusion decouples preference alignment from the generative model, eliminating the need for entire model fine-tuning and reference model reliance.~We further provide theoretical guarantees for PC-Diffusion:1) PC-Diffusion ensures that the preference-guided distributions are consistently propagated across timesteps. 2)The training objective of the preference classifier is equivalent to DPO, but does not require a reference model.3) The proposed preference-guided correction can progressively steer generation toward preference-aligned regions.~Empirical results show that PC-Diffusion achieves comparable preference consistency to DPO while significantly reducing training costs and enabling efficient and stable preference-guided generation.

扩散模型偏好对齐DPO图像生成

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