统一扩散模型与流匹配的对齐方法,无需微调且大幅降本。
Alignment of Diffusion Model and Flow Matching for Text-to-Image Generation
- 基于奖励加权采样思想,统一处理两类生成模型的对齐。
- 一阶段生成性能媲美微调模型,计算成本降低60%以上。
- 适用于追求高效生成的图像合成场景,尤其适合资源受限部署。
扩散模型与流匹配在文本到图像生成中表现出色。现有对齐方法多依赖预训练模型的微调以最大化奖励函数,但需大量计算资源,且跨目标泛化能力差。本文提出一种新对齐框架,利用对齐问题的本质——从奖励加权分布中采样——并证明该方法同时适用于扩散模型(通过梯度引导)和流匹配模型(通过速度引导)。用于奖励加权分布的得分函数(速度场)可分解为预训练得分(速度场)加上奖励的条件期望。针对扩散模型,发现引导项具有对抗性,易引入伪影,因此提出免微调框架,训练一个引导网络估计奖励的条件期望。该方法在单步生成下性能接近微调模型,计算成本至少降低60%。针对流匹配,提出免训练框架,在不增加计算开销的前提下提升生成质量。
原文摘要 · Abstract (English)
Diffusion models and flow matching have demonstrated remarkable success in text-to-image generation. While many existing alignment methods primarily focus on fine-tuning pre-trained generative models to maximize a given reward function, these approaches require extensive computational resources and may not generalize well across different objectives. In this work, we propose a novel alignment framework by leveraging the underlying nature of the alignment problem -- sampling from reward-weighted distributions -- and show that it applies to both diffusion models (via score guidance) and flow matching models (via velocity guidance). The score function (velocity field) required for the reward-weighted distribution can be decomposed into the pre-trained score (velocity field) plus a conditional expectation of the reward. For the alignment on the diffusion model, we identify a fundamental challenge: the adversarial nature of the guidance term can introduce undesirable artifacts in the generated images. Therefore, we propose a finetuning-free framework that trains a guidance network to estimate the conditional expectation of the reward. We achieve comparable performance to finetuning-based models with one-step generation with at least a 60% reduction in computational cost. For the alignment on flow matching, we propose a training-free framework that improves the generation quality without additional computational cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。