用合成数据训练图像生成模型,提升指令跟随和画质。
Scalable Ranked Preference Optimization for Text-to-Image Generation
- 用预训练评分函数自动生成成对图像偏好数据,无需人工标注。
- 通过多模型平均与排序反馈,显著提升SDXL和SD3-Medium的性能。
- 适合追求高效、可扩展训练数据的研究者和开发者。
直接偏好优化(DPO)已成为对齐文本到图像(T2I)模型与人类反馈的有效方法。然而,将DPO成功应用于T2I模型需要大量资源来收集和标注大规模数据集,例如数百万张由人类标注偏好的生成图像对。此外,随着T2I模型快速迭代,生成图像质量提升,这些人工标注的数据集很快过时。本文提出一种可扩展的方案,用于构建大规模全合成的DPO训练数据集。具体而言,利用预训练的奖励函数生成成对图像的偏好标签,完全避免人工标注,极大提高数据收集效率。此外,我们证明了可通过多个模型预测平均并采用排序反馈而非成对偏好。为此,我们引入RankDPO,利用排序反馈增强基于DPO的方法。在SDXL和SD3-Medium模型上,使用我们生成的合成偏好数据集「Syn-Pic」进行训练,显著提升了指令跟随能力(在T2I-Compbench、GenEval和DPG-Bench等基准测试中)和视觉质量(通过用户研究验证)。该流程为开发更优偏好数据集提供了实用且可扩展的解决方案。
原文摘要 · Abstract (English)
Direct Preference Optimization (DPO) has emerged as a powerful approach to align text-to-image (T2I) models with human feedback. Unfortunately, successful application of DPO to T2I models requires a huge amount of resources to collect and label large-scale datasets, e.g., millions of generated paired images annotated with human preferences. In addition, these human preference datasets can get outdated quickly as the rapid improvements of T2I models lead to higher quality images. In this work, we investigate a scalable approach for collecting large-scale and fully synthetic datasets for DPO training. Specifically, the preferences for paired images are generated using a pre-trained reward function, eliminating the need for involving humans in the annotation process, greatly improving the dataset collection efficiency. Moreover, we demonstrate that such datasets allow averaging predictions across multiple models and collecting ranked preferences as opposed to pairwise preferences. Furthermore, we introduce RankDPO to enhance DPO-based methods using the ranking feedback. Applying RankDPO on SDXL and SD3-Medium models with our synthetically generated preference dataset "Syn-Pic" improves both prompt-following (on benchmarks like T2I-Compbench, GenEval, and DPG-Bench) and visual quality (through user studies). This pipeline presents a practical and scalable solution to develop better preference datasets to enhance the performance of text-to-image models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。