arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2406.00633cs.LGcs.AIcs.CVstat.ML
收藏

改进GFlowNets以实现文本到图像扩散对齐

Improving GFlowNets for Text-to-Image Diffusion Alignment

Apple(苹果公司)

Dinghuai Zhang, Yizhe Zhang, Jiatao Gu, Ruixiang Zhang, Josh Susskind, Navdeep Jaitly, Shuangfei Zhai

AI总结:

本文提出DAG算法,利用GFlowNets以高概率生成高奖励图像而非直接最大化奖励,实现对文本到图像扩散模型的有效黑盒对齐。

摘要:

扩散模型已成为生成视觉数据的事实标准方法,其训练目标是匹配训练数据集的分布。此外,我们还希望控制生成过程以满足期望的属性,例如与文本描述的对齐,这可以通过黑盒奖励函数来指定。先前的工作通过基于强化学习的算法对预训练扩散模型进行微调以实现这一目标。然而,这些方法存在信用分配缓慢以及生成样本质量低等问题。在本工作中,我们探索的技术并非直接最大化奖励,而是以相对较高的概率生成高奖励图像——这是生成流网络(GFlowNets)框架下的自然场景。为此,我们提出了基于GFlowNet的扩散对齐(DAG)算法,利用黑盒属性函数对扩散模型进行后训练。在Stable Diffusion及多种奖励设定上的大量实验证实,我们的方法能够有效地将大规模文本到图像扩散模型与给定的奖励信息对齐。

英文摘要:

Diffusion models have become the de-facto approach for generating visual data, which are trained to match the distribution of the training dataset. In addition, we also want to control generation to fulfill desired properties such as alignment to a text description, which can be specified with a black-box reward function. Prior works fine-tune pretrained diffusion models to achieve this goal through reinforcement learning-based algorithms. Nonetheless, they suffer from issues including slow credit assignment as well as low quality in their generated samples. In this work, we explore techniques that do not directly maximize the reward but rather generate high-reward images with relatively high probability -- a natural scenario for the framework of generative flow networks (GFlowNets). To this end, we propose the Diffusion Alignment with GFlowNet (DAG) algorithm to post-train diffusion models with black-box property functions. Extensive experiments on Stable Diffusion and various reward specifications corroborate that our method could effectively align large-scale text-to-image diffusion models with given reward information.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑