通过智能选择关键步骤和重用反馈,让扩散模型更高效地学习人类偏好。
Selective Timestep Weighting and Advantage-Based Replay for Sample-Efficient Diffusion RLHF

- 按重要性加权去噪步骤,聚焦有效信息进行优化。
- 重用高价值轨迹,减少重复获取人类反馈次数。
- 在相同设置下,采样效率提升最高6倍,适合资源有限场景。
基于人类反馈的强化学习(RLHF)已成为对齐生成模型与人类偏好的强大范式。然而,将RLHF应用于扩散模型仍面临严重的反馈效率问题,现有方法通常需要大量人类或奖励模型评估,限制了其在真实场景中的实用性。本文提出两种互补策略,显著提升扩散模型RLHF的反馈效率,同时保持对未见提示的泛化能力。核心观察是:扩散轨迹中的奖励信息分布不均,不同去噪步骤和轨迹对学习的贡献不一致。通过聚焦有信息量的步骤和轨迹,可获得更有效的梯度更新。首先,引入逐步骤权重机制,在策略优化中重新加权去噪步骤,理论上关联到近端策略优化(PPO)的最优收敛性,并通过实验近似其权重趋势。其次,提出一种优先重放机制,选择高信息量轨迹,使模型能复用历史样本,避免重复查询新奖励。二者结合后,在相同超参数设置下,相比广泛使用的扩散模型RLHF基线,样本效率最高提升6倍。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback (RLHF) has emerged as a powerful paradigm for aligning generative models with human preferences. However, applying RLHF to diffusion models remains highly feedback inefficient, as existing approaches typically require large amounts of human or reward model evaluations. This limitation reduces the practicality of diffusion RLHF in realworld settings where feedback is the primary bottleneck. In this paper, we propose two complementary strategies that substantially improve the feedback efficiency of diffusion RLHF while preserving generalization to unseen prompts. Our key observation is that reward information in diffusion trajectories is unevenly distributed: not all denoising timesteps or trajectories contribute equally to learning from a reward signal. By emphasizing informative timesteps and trajectories during optimization, we obtain more effective gradient updates. First, we introduce a per-timestep weighting scheme that reweights denoising steps during policy optimization. We theoretically connect this weighting to the optimal convergence properties of proximal policy optimization (PPO) and approximate the resulting weighting trend empirically. Second, we introduce a replay mechanism that prioritizes informative trajectories, enabling the model to reuse past samples instead of repeatedly querying new rewards. Together, these strategies significantly improve the feedback efficiency of diffusion RLHF. Under identical hyperparameter settings, our approach achieves up to a 6$\times$ improvement in sample efficiency compared to widely used diffusion RLHF baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。