用扩散模型做评分器,分步评估生成过程,提升图像质量。
DRM: Diffusion-based Reward Model With Step-wise Guidance

- 用预训练扩散模型作评分骨干,可评估每一步生成中间状态。
- 提出分步强化学习,每步给奖励,解决传统方法奖励不准问题。
- 适合追求高质量图像生成的用户,尤其在艺术创作场景中有效。
当前主流的扩散模型与人类偏好对齐方法多采用基于视觉语言模型(VLM)的评分器,但这些模型因预训练目标侧重语义对齐,难以捕捉美学、构图、视觉和谐等关键感知属性。本文认为,高保真生成需深刻理解此类视觉特征。为此,提出扩散模型评分器(DRM),利用预训练扩散模型作为强大评估基础,具备评估任意生成阶段中间噪声潜在表示的能力。我们基于此能力设计两种机制:其一,提出分步GRPO算法,通过每步提供密集奖励,解决传统GRPO算法中奖励分配模糊的问题,实现更稳定有效的对齐;其二,提出分步采样策略,将DRM作为动态引导,在每步评估多个生成路径,引导过程趋向更高质量结果。大量实验表明,该方法显著提升生成图像质量。代码已开源。
原文摘要 · Abstract (English)
Current mainstream methods of aligning diffusion models with human preferences typically employ VLM-based reward models. However, these reward models, pre-trained for semantic alignment, struggle to capture the essential perceptual qualities-such as aesthetics, composition, and visual harmony. In this work, we argue that a model capable of high-fidelity generation must possess a profound understanding of these visual attributes. Based on this insight, we introduce the Diffusion-based Reward Model (DRM), a novel paradigm that use the pre-trained diffusion model as a powerful evaluative backbone. A key advantage of the DRM is its unique ability to assess not only the final image but also the noisy intermediate latents at any stage of the generative process. We leverage this step-wise evaluative capacity in two ways. First, we propose Step-wise GRPO, a reinforcement learning algorithm that provides dense, per-step rewards to resolve the imprecise credit assignment problem in GRPO algorithm, leading to more stable and effective alignment. Second, we introduce Step-wise Sampling, a novel inference strategy that employs the DRM as a dynamic guide to evaluate multiple generation paths at each step, steering the process towards higher-quality outcomes. Extensive experiments confirm that our approach significantly enhances the final quality of generated images. Code: https://github.com/jjaxonx/DRM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。