解决扩散模型强化学习中的奖励欺骗问题,提升生成多样性与质量。
GARDO: Reinforcing Diffusion Models without Reward Hacking
- 仅对高不确定性样本施加选择性正则化,避免全局约束影响探索。
- 动态更新参考模型以匹配在线策略能力,保持正则化有效性。
- 通过奖励增强高质量且多样化的样本,防止模式崩溃,适合图像生成研究者。
通过在线强化学习微调扩散模型在提升文本到图像对齐方面展现出巨大潜力。然而,由于视觉任务的真实目标难以精确定义,模型通常依赖部分反映真实目标的代理奖励进行优化,导致奖励欺骗——代理分数上升但图像质量下降、生成多样性丧失。现有方法常添加对参考策略的正则化以缓解此问题,但会牺牲采样效率并阻碍对新颖高奖励区域的探索,因参考策略通常次优。为应对采样效率、有效探索与奖励欺骗缓解之间的矛盾,本文提出通用框架GARDO(Gated and Adaptive Regularization with Diversity-aware Optimization),兼容多种强化学习算法。核心思想是正则化无需普遍应用,而应选择性地惩罚高不确定性样本。为解决探索难题,GARDO引入自适应正则化机制,定期更新参考模型以匹配在线策略能力,确保正则化目标相关性。为缓解强化学习中的模式崩溃,GARDO放大高质量且高多样性的样本奖励,在不破坏优化过程的前提下促进模式覆盖。在多种代理奖励和未见评估指标上的大量实验一致表明,GARDO有效缓解奖励欺骗,提升生成多样性,且不牺牲采样效率或探索能力,彰显其有效性与鲁棒性。
原文摘要 · Abstract (English)
Fine-tuning diffusion models via online reinforcement learning (RL) has shown great potential for enhancing text-to-image alignment. However, since precisely specifying a ground-truth objective for visual tasks remains challenging, the models are often optimized using a proxy reward that only partially captures the true goal. This mismatch often leads to reward hacking, where proxy scores increase while real image quality deteriorates and generation diversity collapses. While common solutions add regularization against the reference policy to prevent reward hacking, they compromise sample efficiency and impede the exploration of novel, high-reward regions, as the reference policy is usually sub-optimal. To address the competing demands of sample efficiency, effective exploration, and mitigation of reward hacking, we propose Gated and Adaptive Regularization with Diversity-aware Optimization (GARDO), a versatile framework compatible with various RL algorithms. Our key insight is that regularization need not be applied universally; instead, it is highly effective to selectively penalize a subset of samples that exhibit high uncertainty. To address the exploration challenge, GARDO introduces an adaptive regularization mechanism wherein the reference model is periodically updated to match the capabilities of the online policy, ensuring a relevant regularization target. To address the mode collapse issue in RL, GARDO amplifies the rewards for high-quality samples that also exhibit high diversity, encouraging mode coverage without destabilizing the optimization process. Extensive experiments across diverse proxy rewards and hold-out unseen metrics consistently show that GARDO mitigates reward hacking and enhances generation diversity without sacrificing sample efficiency or exploration, highlighting its effectiveness and robustness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。