arXiv:2510.18353cs.CV2025-10NeurIPS被引 3

不用奖励模型,用排序学习优化图像生成质量

Ranking-based Preference Optimization for Diffusion Models from Implicit User Feedback

  • 将偏好学习转为排序问题,简化训练目标
  • 结合离线专家数据与在线负样本,提升生成多样性
  • 在新提示下表现更优,适合图像生成研究者

直接偏好优化(DPO)方法通过成对比较训练,在对齐文本到图像扩散模型与人类偏好方面展现出潜力。这些方法避免了REINFORCE算法带来的训练不稳定性,但仍面临图像概率估计不准的问题,源于sigmoid函数的非线性特性以及离线数据集多样性有限。本文提出扩散去噪排序优化(Diffusion-DRO),一种基于逆强化学习的新偏好学习框架。Diffusion-DRO通过将偏好学习转化为排序问题,摆脱了对奖励模型的依赖,将训练目标简化为去噪形式,克服了先前方法中的非线性估计难题。此外,Diffusion-DRO创新性地融合了离线专家演示与在线策略生成的负样本,有效捕捉人类偏好并弥补离线数据局限。大量实验表明,Diffusion-DRO在多种挑战性及未见提示下均实现更高生成质量,优于当前最优基线,在定量指标和用户评测中表现突出。代码与预训练模型已开源。

原文摘要 · Abstract (English)

Direct preference optimization (DPO) methods have shown strong potential in aligning text-to-image diffusion models with human preferences by training on paired comparisons. These methods improve training stability by avoiding the REINFORCE algorithm but still struggle with challenges such as accurately estimating image probabilities due to the non-linear nature of the sigmoid function and the limited diversity of offline datasets. In this paper, we introduce Diffusion Denoising Ranking Optimization (Diffusion-DRO), a new preference learning framework grounded in inverse reinforcement learning. Diffusion-DRO removes the dependency on a reward model by casting preference learning as a ranking problem, thereby simplifying the training objective into a denoising formulation and overcoming the non-linear estimation issues found in prior methods. Moreover, Diffusion-DRO uniquely integrates offline expert demonstrations with online policy-generated negative samples, enabling it to effectively capture human preferences while addressing the limitations of offline data. Comprehensive experiments show that Diffusion-DRO delivers improved generation quality across a range of challenging and unseen prompts, outperforming state-of-the-art baselines in both both quantitative metrics and user studies. Our source code and pre-trained models are available at https://github.com/basiclab/DiffusionDRO.

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

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