​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

未分类4周前发布 tree
15 0 0
↑ 点击蓝字 关注极市平台
​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了
作者丨AI生成未来
来源丨AI生成未来
编辑丨极市平台

极市导读

 

本文介绍了一种新的文本到图像模型偏好优化方法,通过合成标注偏好数据集(Syn-Pic)和基于排名的偏好优化(RankDPO)技术,显著提升了模型的文本跟随能力和视觉质量,同时降低了数据收集成本并提高了数据集的可扩展性。这种方法在多个基准数据集上展示了改进效果,并在用户研究中得到了验证,为文本生成图像模型的性能和安全性提升提供了一个实用且可扩展的解决方案。 >>加入极市CV技术交流群,走在计算机视觉的最前沿

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

文章链接:https://arxiv.org/pdf/2410.18013

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

亮点直击

  • 合成标注偏好数据集(Syn-Pic):从不同的T2I模型生成图像,并通过多个预训练的奖励模型对其进行标注,这些模型可以估计人类偏好。因此,数据收集过程中无需人工标注,使得数据收集成本更低且易于扩展。通过聚合多个奖励模型的评分,减轻了奖励过度优化的问题。与传统的成对比较不同,为每个提示生成的图像构建一个排名。虽然聚合多个人工标注者的偏好并构建排名是可行的,但这相比本文的低成本方法会显著增加标注成本。
  • 基于排名的偏好优化(RankDPO):为利用排名中更丰富的信号,引入了一种排名增强的DPO目标函数RankDPO,借鉴了“学习排序”领域的广泛文献。它使用折扣累计增益(DCG)对偏好损失进行加权,使其与首选排名对齐。
  • 使用与Pick-a-Picv2相同的提示,SDXL和SD3-Medium模型的表现显著提升。展示了在多个基准数据集上的改进结果,包括GenEval(表1)、T2I-Compbench(表2)和DPG-Bench(表3),以及用户研究中的视觉对比(图1中的示例和图3)。
  • 与现有的偏好优化方法相比,取得了最新的效果,例如表3。更重要的是,仅使用了Pick-a-Picv2所需图像数量的三分之一,即表8。
  • 尽管SD3-Medium(20亿参数)已经通过3M个人偏好数据的DPO进行优化,但使用包含24万图像的Syn-Pic数据集仍能进一步实现显著提升,例如表1、2、3。
​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了
​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了
​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

总结速览

解决的问题
传统的直接偏好优化(DPO)方法依赖大量人工标注的数据集来对文本生成图像(T2I)模型进行对齐,但这种方法成本高昂,且数据容易过时,难以适应T2I模型快速提升的需求。

提出的方案
本研究提出一种完全合成的数据集收集方法,用于DPO训练。通过预训练的奖励函数生成成对图像的偏好数据,替代人工标注,极大地提升数据集的收集效率。同时,引入RankDPO方法,通过排名反馈增强DPO训练,提升模型的偏好学习效果。

应用的技术
采用预训练的奖励函数生成成对图像的偏好数据,并在SDXL和SD3-Medium模型上应用RankDPO技术。数据集“Syn-Pic”用于支持该方法,避免人工参与标注并提供更高效的偏好数据生成方式。

达到的效果
通过使用“Syn-Pic”合成偏好数据集和RankDPO方法,显著提升了模型的文本跟随能力(如T2I-Compbench、GenEval和DPG-Bench基准测试)和视觉质量(通过用户研究验证),为开发更优质的偏好数据集、提升文本生成图像模型的性能和安全性提供了一个实用且可扩展的解决方案。

方法

本节首先概述用于文本生成图像的扩散模型及其直接偏好优化方法。接下来,讨论策划和标注可扩展的偏好优化数据集的过程。最后,描述一种基于排名的偏好优化方法RankDPO,以利用该排序偏好数据集。下图2中对这两个组成部分进行了说明。算法3提供了在Syn-Pic上训练RankDPO的伪代码。

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了
​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

符号表示
使用符号 表示从分布 中抽取的真实数据。在设置中, 扩散过程将真实图像 转换为高斯噪声 , 并使用预定义的信号噪声调度 。扩散模型通过学习一个去噪器 来反转该过程, 是一个由参数 表示的神经网络, 用于估计条件分布 , 其中 是指导生成的条件信号。对于文本生成图像模型,将 作为与文本提示相对应的嵌入表示。为简洁起见, 符号 可表示文本提示/嵌入。

扩散模型
去噪扩散概率模型(DDPMs)通过反转ODE流来预测真实数据 。具体来说, 使用预定义的信号噪声调度 , 采样一个噪声 , 并在时间 生成一个带噪样本 :

去噪模型 由参数 定义, 其训练目标为:

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

其中, 是总步骤数, 是条件信号。

用于扩散模型的DPO

Bradley-Terry(BT)模型通过以下公式定义成对偏好:

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

其中, 为 sigmoid函数, 为更偏好的图像, 为较不偏好的图像, 为奖励模型, 用于计算条件 与图像 之间的对齐分数。基于此, Rafailov等人证明了以下目标与使用奖励模型 的显式强化学习过程(例如, PPO/REINFORCE)等价:

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

其中, 是基础参考分布, 控制分布偏差。然而, 在扩散模型的背景下, 计算图像的似然(即 )是不可行的。因此, Wallace等人提出了一种可行的替代方法, 他们证明该方法在对原始DPO目标稍作放松后是等价的。

给定一个样本 , 去噪模型和参考模型 , 定义评分函数为:

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

其中, 是在时间 时输入 的带噪潜变量。基于此, 更新后的 DPO目标可以定义为:

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

在实际操作中, 随机采样一个时间步 , 并在该时间步对获胜样本 和失败样本 分别计算可训练模型和参考模型的去噪目标。DPO目标确保对于给定的条件信号 , 去噪效果在获胜样本上得到改善, 而在失败样本上变差。这样模型会偏向生成更符合条件 的偏好图像。

合成标注偏好数据集 (Syn-Pic)

本节介绍一种高效且可扩展的方法, 用于收集在DPO目标中使用的偏好数据集 (公式4)。给定一个包含 个文本提示的列表 包含成对的偏好数据, 表示获胜和失败的图像, 即 。Pick-a-Picv2 是早期研究中使用的偏好数据集的一个示例, 包含近58,000个提示和 0.85 百万个偏好对。传统上, 数据收集过程依赖于文本生成图像模型生成的图像的人为标注, 这种方式因人工标注成本而昂贵。此外, 随着文本生成图像模型的不断改进, 这些手工策划的数据集也很快变得过时。

通过从多个最先进的T2I模型(如SD3-Medium、StableCascade、Pixart-Σ 和SDXL)生成与Pick-a-Picv2数据集相同提示的图像,收集了一个新的偏好数据集。

此外, 使用现有的现成的人工偏好模型(例如HPSv2.1)标注这些样本, 从而免去了人工标注成本。然而, 不同的奖励模型可能具有互补的优点(例如,一些模型更关注视觉质量,另一些则更擅长文本图像对齐等)。因此,提出汇总来自五种不同模型的偏好,包括HPSv2.1、 MPS、PickScore、VQAScore 和ImageReward。对于每个提示 和图像 , 通过将获胜次数与比较总次数进行汇总, 计算该图像在所有奖励模型下被偏好的概率。此得分 用于按聚合得分的降序对生成的图像进行排序,形成目标偏好排名,并用于增益函数。因此,对于 个 T 21 模型, 得到 , 这是一个完全合成排序的偏好数据集。算法1中描述详细流程。

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

讨论
本文的数据收集方法具有以下几个优点:

  • 成本效率:可以生成任意规模的偏好数据集,因为标注环节中没有人为参与,图像生成和标注均使用现成的模型,从而降低数据集策划成本。例如,收集Pick-a-Picv2数据集大约需要200的成本收集一个类似规模的数据集。
  • 可扩展性:随着数据集收集成本的降低,可以针对新的文本生成图像模型进行迭代,从而消除了由于新模型出现而导致旧偏好数据集过时的问题。
  • 基于排名的偏好优化:由于每个提示运行多个T2I模型,因此收集到的是一个排名的偏好列表,而不仅仅是早期数据集中成对的数据。这能够在偏好优化中探索排名目标。

基于排名的偏好优化 (RankDPO)

与专注于成对偏好的DPO目标不同, 合成数据集为每个提示生成多个图像, 形成一个排名的偏好数据集。因此,希望一次性优化多种偏好,而不仅仅依赖于成对偏好。具体而言,给定一个文本提示 和按偏好排名的生成图像 , 希望确保图像 的去噪效果优于 , 对于所有 。为了实现这一点, 从学习排序(LTR)文献中获得灵感,通过为每个样本添加DCG权重重新利用LambdaLoss, 类似于Liu等所做的, 具体如下:

给定一个样本 , 去噪和参考模型 , 定义得分为:

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

其中 , 是在时间 t 对输入 的噪声潜变量。此得分用于衡量模型预测相对于参考模型在给定条件 c下的表现优劣。

计算得分 后, 图像将根据从最优(最低的 )到最劣(最高的 )进行排序。这是使用模型 的预测排名。使用真实的得分 来获得真实的偏好排名 。每个图像 的排名用 表示, 其中 表示最佳图像, 表示第二最佳,以此类推。每个样本 i的增益 是根据人类偏好奖励模型得分计算的样本 i 相对于所有其他样本 j 被偏好的平均概率。

使用 , 定义增益函数 和折扣函数 如下:

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

折扣函数随着排名 的增加而减少, 从而确保排名较高的图像(即较低 的图像)对最终损失的影响更大。折扣函数的对数形式平滑了相邻排名之间的惩罚差异, 使模型对小的排名错误更具鲁棒性, 尤其是对排名较低的图像。将两张图像对 之间的权重定义为

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

最后,将所有这些结合在一起,RankDPO损失的公式可以表示为

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

其中, 为 sigmoid 函数, 控制 KL 正则化的强度。详细的训练过程在算法 2 中。

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

该损失函数不仅鼓励模型满足成对的偏好,还要尊重为相同提示生成的图像的整体排名。通过使用从排名中获得的增益和折扣来加权传统的 DPO 目标,确保模型根据排名优先生成质量更高的图像,从而在美学和提示对齐方面实现更一致的改进。

实验

实施细节实验中使用开源的 SDXL和 SD3-Medium 模型。从 Pick-a-Picv2 收集了 58K 个提示,并使用四个模型,即 SDXL、SD3-Medium、Pixart-Σ 和 Stable Cascade,准备了 Syn-Pic 数据集。在 8 张 A100 GPU 上训练 RankDPO,持续 16 小时,批次大小为 1024,训练 400 步。

比较结果

短提示在前面表 1 中,报告了 GenEval上的结果。RankDPO 在几乎所有类别上都持续改善了性能,使得 SDXL 的平均性能从 0.55 提高到 0.61,SD3-Medium 的平均性能从 0.70 提高到 0.74。特别是观察到在“两个物体”、“计数”和“颜色归属”等方面有显著提升,增幅近 10%。在前面表 2 中的 T2I-Compbench中,也观察到了类似的趋势,其中 SDXL 在“颜色”和“纹理”上提升超过 10%,并在其他类别中也有改进。

长提示在前面表 3 中,进一步在 DPG-Bench上评估了模型的视觉质量和提示对齐性,该基准包含长且详细的提示。为了测量提示对齐性,使用了原始 DSG 指标和 VQAScore,而对于视觉质量,使用了 Q-Align 模型。

Diffusion-DPO(标记为 DPO-SDXL)在 Pick-a-Picv2 上训练,能够在提示对齐性方面提供有意义的改进,而对 SDXL 进行 MaPO和 SPO的微调(标记为 MaPO-SDXL 和 SPO-SDXL)则提高了视觉质量。然而,尽管 RankDPO 仅在合成偏好上进行训练,但所有指标都有显著提升(例如,SDXL 的 DSG 得分从 74.51 提高到 79.26,Q-Align 得分从 0.72 提高到 0.81),并达到了最新的提示对齐指标。对于 SD3-Medium,在使用RankDPO 进行微调后,继续看到模型性能的提升。

用户研究
为了进一步验证本文方法的有效性,在 DPG-Bench 上对 450 个提示进行了用户研究。要求用户根据他们的整体偏好(即结合文本与图像的对齐性和视觉质量)选择更好的图像。前面图 3 显示,RankDPO 的胜率优于 DPO-SDXL和 SDXL,这表明其在提高生成图像整体质量方面的有效性。

下图 4 中展示了来自 DPG-Bench的提示的定性示例。与基础 SDXL 和其他偏好调整模型相比,RankDPO 在文本渲染方面有显著改进,能够捕捉到所有提示中描述的对象,而其他模型则遗漏了这些对象,并且在图像中对对象之间复杂关系的建模更佳。

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了
​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

计算成本讨论
需要 10 个 A100 GPU 天来生成图像并标注偏好,这是一次性成本。在生成数据上运行 RankDPO 进行 400 步训练,SDXL 在 1024² 像素下大约需要 6 个 GPU 天。相比之下,现有的奖励优化方法在较小的 SD1.5 模型(512² 像素)上需要 64-95 个 A100 GPU 天。类似地,与 Diffusion-DPO相比,RankDPO 在三分之一的数据上训练,同时避免了手动整理的偏好。还有一些通过使用文本编码器(如 T5/LLaMA 模型)来增强文本到图像模型的方法,这些方法需要 1000 万到 3400 万个密集标注的图像,并训练 50-120 个 A100 GPU 天。

消融分析

数据和标注函数的影响由于生成偏好是 RankDPO 的关键方面,在下表 4 中评估了不同的标注选择。尝试随机标注,即随机选择偏好并应用 DPO。这仅能提供有限的性能提升(从 74.65 提升至 75.66 的 DSG 得分)。还展示了来自单一奖励模型(HPSv2.1)的成对偏好的结果,以及从 5 个模型中平均偏好的结果。虽然 HPSv2.1 对提示对齐和视觉质量都提供了良好的改进,但跨多个模型的预测集成进一步提高了结果。这些结果优于应用于 Pick-a-Picv2 的 DPO,突显了在构建偏好数据集时图像质量的重要性。最后,调查了用于构建 Syn-Pic 的不同模型的影响。这是通过仅更改种子,构建与 SDXL 图像相似的数据集来完成的。虽然在提示对齐方面几乎得到了相同的改进,但在视觉质量上只看到了小幅提升。这表明,合成偏好调优可以应用于任何模型的输出,但使用不同模型生成的图像可以进一步改善结果。

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

学习目标分析偏好优化的一个关键方面是学习目标的选择,在上表 4 中进行各种实验进行比较。除了常规的 DPO 公式外,一些研究表明,在经过精心挑选的高质量数据上进行监督微调的好处,也将其纳入比较。基线包括以下几种方法:

  • 监督微调:从每对比较中选择获胜的图像,并在此子集上对 SDXL 进行微调。
  • 加权微调:在所有样本上对 SDXL 进行微调,但根据 HPSv2.1 分数为每个样本分配权重,类似于 Lee et al.。
  • DPO + 增益函数加权:通过使用增益函数对样本加权,可以改善 DPO 目标。

可以看到,RankDPO 实现了最佳结果,突显了基于成对偏好的排名标准的整合对增强偏好优化的好处。

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了
​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

结论与讨论

本文提出了一种强大且具有成本效益的文本到图像模型偏好优化方法。具体而言,展示了如何通过合成生成偏好优化数据集来收集更优的信号(例如,排名与成对偏好、跨模型集成偏好)。还提出了一种简单的方法来利用更强的信号,从而在多个基准测试中实现了针对提示跟随和视觉质量的最新结果,无论是在扩散模型还是修正流模型中。希望本文的工作为未来在文本到图像模型上扩展有效的后期训练解决方案铺平道路。

参考文献

[1] Scalable Ranked Preference Optimization for Text-to-Image Generation

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

公众号后台回复“数据集”获取100+深度学习各方向资源整理

极市干货

技术专栏:多模态大模型超详细解读专栏搞懂Tranformer系列大视觉模型 (LVM) 解读扩散模型系列极市直播
技术综述:小目标检测那点事大模型面试八股含答案万字长文!人体姿态估计(HPE)入门教程

​无需昂贵标注!大幅提升SDXL和SD3-Medium效果!文生图偏好优化新宠来了

点击阅读原文进入CV社区

收获更多技术干货

© 版权声明

相关文章

暂无评论

暂无评论...