让扩散模型生成更真实,避免奖励欺骗。
Reward Sharpness-Aware Fine-Tuning for Diffusion Models
- 用平滑奖励模型的梯度优化生成过程。
- 在多个数据集上减少奖励欺骗,提升图像质量。
- 适合做可控生成与高质量图像合成的研究者。
基于人类反馈的强化学习(RLHF)在对齐大语言模型与人类偏好方面表现优异,推动了以奖励为中心的扩散强化学习(RDRL)的发展,实现类似对齐与可控性。尽管扩散模型能生成高质量输出,但RDRL仍易受奖励欺骗影响,即奖励分数上升却未带来感知质量提升。我们发现该问题源于奖励模型梯度的非鲁棒性,尤其当奖励景观随输入图像变化剧烈时。为此,我们提出利用经过鲁棒化处理的奖励模型梯度,无需重新训练。具体方法包括:通过扩散模型参数扰动及其生成样本扰动,获得平滑化的奖励模型梯度。实验表明,每种方法独立缓解奖励欺骗并增强鲁棒性,联合使用效果更显著。所提出的RSA-FT(Reward Sharpness-Aware Fine-Tuning)框架简单、兼容性强,持续提升RDRL的可靠性。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback (RLHF) has proven effective in aligning large language models with human preferences, inspiring the development of reward-centric diffusion reinforcement learning (RDRL) to achieve similar alignment and controllability. While diffusion models can generate high-quality outputs, RDRL remains susceptible to reward hacking, where the reward score increases without corresponding improvements in perceptual quality. We demonstrate that this vulnerability arises from the non-robustness of reward model gradients, particularly when the reward landscape with respect to the input image is sharp. To mitigate this issue, we introduce methods that exploit gradients from a robustified reward model without requiring its retraining. Specifically, we employ gradients from a flattened reward model, obtained through parameter perturbations of the diffusion model and perturbations of its generated samples. Empirically, each method independently alleviates reward hacking and improves robustness, while their joint use amplifies these benefits. Our resulting framework, RSA-FT (Reward Sharpness-Aware Fine-Tuning), is simple, broadly compatible, and consistently enhances the reliability of RDRL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。