无需配对图像偏好数据,用文本扰动实现扩散模型的免费对齐。
Free Lunch Alignment of Text-to-Image Diffusion Models without Preference Image Pairs
- 用大语言模型生成错位文本,让模型学会区分匹配与不匹配提示。
- 在多个基准上优于原始方法,人类评分更高,文本图像对齐更准。
- 开源可复现,适合想提升生成质量又不想依赖人工标注的研究者。
基于扩散的文本到图像(T2I)模型虽能生成高质量图像,但确保文本与图像准确对齐仍是挑战。现有方法多依赖人类反馈强化学习(RLHF),需昂贵的人工标注或训练奖励函数,难以扩展。本文提出文本偏好优化(TPO)框架,实现无需配对图像偏好数据的“免费对齐”。通过扰动原始描述生成不匹配提示,训练模型偏好正确匹配的文本。该框架兼容现有偏好算法,我们将其扩展至DPO和KTO,形成TDPO与TKTO。在多个基准上的定量与定性评估显示,新方法持续优于原版,人类偏好得分更高,文本-图像对齐更优。代码已开源:https://github.com/DSL-Lab/T2I-Free-Lunch-Alignment。
原文摘要 · Abstract (English)
Recent advances in diffusion-based text-to-image (T2I) models have led to remarkable success in generating high-quality images from textual prompts. However, ensuring accurate alignment between the text and the generated image remains a significant challenge for state-of-the-art diffusion models. To address this, existing studies employ reinforcement learning with human feedback (RLHF) to align T2I outputs with human preferences. These methods, however, either rely directly on paired image preference data or require a learned reward function, both of which depend heavily on costly, high-quality human annotations and thus face scalability limitations. In this work, we introduce Text Preference Optimization (TPO), a framework that enables "free-lunch" alignment of T2I models, achieving alignment without the need for paired image preference data. TPO works by training the model to prefer matched prompts over mismatched prompts, which are constructed by perturbing original captions using a large language model. Our framework is general and compatible with existing preference-based algorithms. We extend both DPO and KTO to our setting, resulting in TDPO and TKTO. Quantitative and qualitative evaluations across multiple benchmarks show that our methods consistently outperform their original counterparts, delivering better human preference scores and improved text-to-image alignment. Our Open-source code is available at https://github.com/DSL-Lab/T2I-Free-Lunch-Alignment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。