提出LOOP方法,让文本生成图像模型更高效地对齐黑盒目标。
A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuning
- 结合多动作采样与基线修正,降低策略梯度方差。
- 在多种黑盒目标上提升性能,样本效率优于传统REINFORCE。
- 适合追求高效微调的扩散模型研究者和开发者使用。
基于强化学习的微调已成为对齐扩散模型与黑盒目标的强大方法。近端策略优化(PPO)虽在性能和样本效率上表现优异,但对超参数敏感且计算开销大。而REINFORCE虽简化实现、内存占用低,却因高方差和严重的样本低效性难以应用。尽管通过多动作采样与基线校正可降低方差,其样本效率仍不足。为此,本文系统分析了REINFORCE与PPO在样本效率与效果间的权衡,提出一种新方法:留一法PPO(LOOP)。LOOP融合了REINFORCE中的多动作采样与基线校正,同时保留PPO的剪裁机制与重要性采样以增强鲁棒性和样本效率。实验表明,LOOP在多个黑盒目标上显著提升了扩散模型性能,实现了样本效率与最终效果的更好平衡。
原文摘要 · Abstract (English)
Reinforcement learning (RL)-based fine-tuning has emerged as a powerful approach for aligning diffusion models with black-box objectives. Proximal policy optimization (PPO) is a popular choice of method for policy optimization. While effective in terms of performance and sample complexity, PPO is highly sensitive to hyper-parameters and involves substantial computational overhead. REINFORCE, on the other hand, mitigates some implementation complexities such as high memory overhead and sensitive hyper-parameter tuning, but has suboptimal performance due to high variance and crucially sample inefficiency, which is the primary notion of efficiency we study in this work. While the variance of the REINFORCE can be reduced by sampling multiple actions per input prompt and using a baseline correction term, it still suffers from sample inefficiency. To address these challenges, we systematically analyze the sample efficiency-effectiveness trade-off between REINFORCE and PPO, and propose leave-one-out PPO ( LOOP), a novel RL for diffusion fine-tuning method. LOOP combines variance reduction techniques from REINFORCE, such as sampling multiple actions per input prompt and a baseline correction term, with the robustness and sample efficiency of PPO via clipping and importance sampling. Our results demonstrate that LOOP effectively improves diffusion models on various black-box objectives, and achieves a better balance between sample efficiency and final performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。