用轻量对比学习提升扩散模型图文对齐,不需额外数据
Aligning Text to Image in Diffusion Models is Easier Than You Think
- 通过软文本标记实现图文表征对比学习
- 仅添加不到100万参数,显著提升语义一致性
- 适合需要精准图文生成的视觉生成研究者
尽管生成模型在图文对齐方面取得进展,但仍存在残余偏差。传统扩散模型基于配对图文数据训练,优化得分匹配或流匹配损失,但从表征对齐角度看并不理想。本文提出一种新方法:利用现有数据集同时作为正负样本,通过对比学习实现更优对齐。为此设计轻量级策略SoftREPA,通过引入软文本标记进行表征对齐,仅增加少于100万可训练参数。理论分析表明该方法能显式提升图文表征间的互信息。实验验证其在图文生成与文本引导图像编辑任务中均有效提升语义一致性。
原文摘要 · Abstract (English)
While recent advancements in generative modeling have significantly improved text-image alignment, some residual misalignment between text and image representations still remains. Some approaches address this issue by fine-tuning models in terms of preference optimization, etc., which require tailored datasets. Orthogonal to these methods, we revisit the challenge from the perspective of representation alignment-an approach that has gained popularity with the success of REPresentation Alignment (REPA). We first argue that conventional text-to-image (T2I) diffusion models, typically trained on paired image and text data (i.e., positive pairs) by minimizing score matching or flow matching losses, is suboptimal from the standpoint of representation alignment. Instead, a better alignment can be achieved through contrastive learning that leverages existing dataset as both positive and negative pairs. To enable efficient alignment with pretrained models, we propose SoftREPA- a lightweight contrastive fine-tuning strategy that leverages soft text tokens for representation alignment. This approach improves alignment with minimal computational overhead by adding fewer than 1M trainable parameters to the pretrained model. Our theoretical analysis demonstrates that our method explicitly increases the mutual information between text and image representations, leading to enhanced semantic consistency. Experimental results across text-to-image generation and text-guided image editing tasks validate the effectiveness of our approach in improving the semantic consistency of T2I generative models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。