arXiv:2601.20218cs.CV2026-01中稿 · ICLR被引 17

提出密集奖励机制,让扩散模型每一步生成都获得精准反馈。

DenseGRPO: From Sparse to Dense Reward for Flow Matching Model Alignment

  • 用ODE方法预测每步去噪的奖励增量,实现细粒度奖励
  • 在多个基准上显著提升文本到图像生成与人类偏好对齐效果
  • 适合关注生成质量精细调控的研究者和开发者

基于流匹配模型的GRPO方法在文本到图像生成中已显著提升人类偏好对齐效果。然而,仍存在稀疏奖励问题:整个去噪轨迹的终端奖励被应用于所有中间步骤,导致全局反馈信号与各步骤的精细贡献不匹配。为此,本文提出DenseGRPO框架,通过密集奖励实现人类偏好对齐。核心包括:(1) 提出预测逐步奖励增益作为每步的密集奖励,利用基于ODE的方法对中间清晰图像施加奖励模型,确保反馈信号与各步骤贡献一致,促进有效训练;(2) 基于估计的密集奖励,发现现有GRPO方法中均匀探索设置与随时间变化的噪声强度之间存在不匹配,导致探索空间不当,因此提出奖励感知方案,通过自适应调整SDE采样器中的时间步特异性随机注入强度,确保各时间步均有合适探索空间。在多个标准基准上的大量实验验证了DenseGRPO的有效性,并凸显了有效密集奖励在流匹配模型对齐中的关键作用。

原文摘要 · Abstract (English)

Recent GRPO-based approaches built on flow matching models have shown remarkable improvements in human preference alignment for text-to-image generation. Nevertheless, they still suffer from the sparse reward problem: the terminal reward of the entire denoising trajectory is applied to all intermediate steps, resulting in a mismatch between the global feedback signals and the exact fine-grained contributions at intermediate denoising steps. To address this issue, we introduce \textbf{DenseGRPO}, a novel framework that aligns human preference with dense rewards, which evaluates the fine-grained contribution of each denoising step. Specifically, our approach includes two key components: (1) we propose to predict the step-wise reward gain as dense reward of each denoising step, which applies a reward model on the intermediate clean images via an ODE-based approach. This manner ensures an alignment between feedback signals and the contributions of individual steps, facilitating effective training; and (2) based on the estimated dense rewards, a mismatch drawback between the uniform exploration setting and the time-varying noise intensity in existing GRPO-based methods is revealed, leading to an inappropriate exploration space. Thus, we propose a reward-aware scheme to calibrate the exploration space by adaptively adjusting a timestep-specific stochasticity injection in the SDE sampler, ensuring a suitable exploration space at all timesteps. Extensive experiments on multiple standard benchmarks demonstrate the effectiveness of the proposed DenseGRPO and highlight the critical role of the valid dense rewards in flow matching model alignment.

扩散模型奖励对齐生成质量

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