提出GDRO方法,让扩散模型高效实现群体奖励对齐。
GDRO: Group-level Reward Post-training Suitable for Diffusion Models
- 设计离线优化框架,避免图像采样耗时
- 在OCR和GenEval上提升奖励分数且抗奖励作弊
- 适用于无需随机采样的确定性扩散模型
近期工作采用基于大语言模型的在线强化学习,对文本到图像的修正流扩散模型进行奖励对齐。群体奖励能有效对齐模型目标奖励,但存在效率低、依赖随机采样器及奖励作弊等问题。原因在于修正流模型与大语言模型本质不同:1)图像在线采样耗时极长,主导训练时间;2)一旦初始噪声固定,修正流即为确定性过程。针对此问题,受大语言模型中群体奖励效果启发,本文提出群体级直接奖励优化(GDRO)。GDRO是一种新型后训练范式,结合修正流模型特性,通过严格理论分析表明其支持全离线训练,大幅节省图像展开采样时间成本;同时独立于扩散采样器,无需通过ODE-to-SDE近似获取随机性。我们还实证研究了可能误导评估的奖励作弊陷阱,并引入修正评分——不仅考虑原始奖励,还考量奖励作弊趋势。大量实验表明,GDRO在OCR和GenEval任务上通过群体离线优化显著提升扩散模型奖励得分,且在缓解奖励作弊方面表现出强稳定性与鲁棒性。
原文摘要 · Abstract (English)
Recent advancements adopt online reinforcement learning (RL) from LLMs to text-to-image rectified flow diffusion models for reward alignment. The use of group-level rewards successfully aligns the model with the targeted reward. However, it faces challenges including low efficiency, dependency on stochastic samplers, and reward hacking. The problem is that rectified flow models are fundamentally different from LLMs: 1) For efficiency, online image sampling takes much more time and dominates the time of training. 2) For stochasticity, rectified flow is deterministic once the initial noise is fixed. Aiming at these problems and inspired by the effects of group-level rewards from LLMs, we design Group-level Direct Reward Optimization (GDRO). GDRO is a new post-training paradigm for group-level reward alignment that combines the characteristics of rectified flow models. Through rigorous theoretical analysis, we point out that GDRO supports full offline training that saves the large time cost for image rollout sampling. Also, it is diffusion-sampler-independent, which eliminates the need for the ODE-to-SDE approximation to obtain stochasticity. We also empirically study the reward hacking trap that may mislead the evaluation, and involve this factor in the evaluation using a corrected score that not only considers the original evaluation reward but also the trend of reward hacking. Extensive experiments demonstrate that GDRO effectively and efficiently improves the reward score of the diffusion model through group-wise offline optimization across the OCR and GenEval tasks, while demonstrating strong stability and robustness in mitigating reward hacking.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。