用中间反馈提升流模型生成质量,解决训练不稳与奖励分配难题。
Fine-tuning Flow Matching Generative Models with Intermediate Feedback
- 设计奖赏塑形与双稳定性机制,稳定中间值学习与策略更新。
- 在Stable Diffusion 3上实现文本对齐最优性能,泛化至未见偏好模型。
- 适合需要高效微调生成模型的研究者与应用开发者。
基于流的生成模型在文本到图像生成中表现卓越,但使用中间反馈进行微调仍具挑战性,尤其是连续时间流匹配模型。现有方法多仅从最终奖励学习,难以解决信用分配问题;而通过直接回归累积奖励来学习评判器的方法,在在线设置下常出现训练不稳定和模型崩溃。本文提出AC-Flow,一种稳健的演员-评判器框架,包含三项关键创新:(1) 奖赏塑形提供归一化学习信号,实现稳定的中间值学习与梯度控制;(2) 新型双稳定性机制结合优势裁剪防止破坏性策略更新,并引入预热阶段使评判器成熟后再影响演员;(3) 可扩展的广义评判器加权方案,在保留模型多样性的同时,通过Wasserstein正则化扩展传统奖励加权方法。在Stable Diffusion 3上的大量实验表明,AC-Flow在文本到图像对齐任务中达到当前最佳性能,并能泛化至未见的人类偏好模型。结果表明,即使采用计算高效的评判器模型,也能在不牺牲生成质量、多样性和稳定性的情况下,稳健微调流模型。
原文摘要 · Abstract (English)
Flow-based generative models have shown remarkable success in text-to-image generation, yet fine-tuning them with intermediate feedback remains challenging, especially for continuous-time flow matching models. Most existing approaches solely learn from outcome rewards, struggling with the credit assignment problem. Alternative methods that attempt to learn a critic via direct regression on cumulative rewards often face training instabilities and model collapse in online settings. We present AC-Flow, a robust actor-critic framework that addresses these challenges through three key innovations: (1) reward shaping that provides well-normalized learning signals to enable stable intermediate value learning and gradient control, (2) a novel dual-stability mechanism that combines advantage clipping to prevent destructive policy updates with a warm-up phase that allows the critic to mature before influencing the actor, and (3) a scalable generalized critic weighting scheme that extends traditional reward-weighted methods while preserving model diversity through Wasserstein regularization. Through extensive experiments on Stable Diffusion 3, we demonstrate that AC-Flow achieves state-of-the-art performance in text-to-image alignment tasks and generalization to unseen human preference models. Our results demonstrate that even with a computationally efficient critic model, we can robustly finetune flow models without compromising generative quality, diversity, or stability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。