arXiv:2605.30038cs.LGcs.AI2026-05

通过改进得分匹配,让扩散模型更准地理解文本描述。

Alignment-Guided Score Matching for Text-to-Image Alignment in Diffusion Models

论文配图:Alignment-Guided Score Matching for Text-to-Image Alignment in Diffusion Models
图 1 · 摘自论文原文
  • 在扩散模型得分层面引入文本对齐引导,优化软标记
  • 计数准确率提升超35%,解决重复生成等缺陷
  • 无需奖励信号,适配主流文生图模型

扩散模型生成图像虽逼真,但常存在文本-图像对齐不准问题。现有后训练方法依赖外部奖励或人类偏好,性能受限于奖励质量,且未直接改进扩散过程中的对齐机制。近期无奖励方法SoftREPA通过对比学习优化软文本标记,有效提升对齐效果,但其对比目标对负样本惩罚过重,导致过计数、重复等典型失败案例。为此,本文提出一种轻量级、无奖励的后训练方法(AGSM),将对比对齐引导直接融入扩散模型的得分匹配目标中,在得分层赋予对齐方向,缓解上述问题,生成更连贯、语义忠实的结果。实验表明,该方法在保持优于标准参数高效微调的基础上,显著改善SoftREPA的失败案例,在GenEval基准上计数准确率提升超35%。本方法可无缝应用于SD1.5、SDXL和SD3等现有扩散模型主干,并与基于强化学习的方法互补。

原文摘要 · Abstract (English)

Diffusion models generate highly realistic images but often struggle with precise text-image alignment. While recent post-training methods improve alignment using external rewards or human preference signals, their performance heavily depends on reward quality and does not directly address alignment within the diffusion process itself. Recent reward-free approaches such as SoftREPA demonstrate that optimizing soft text tokens via contrastive learning can effectively improve text-image representation alignment, outperforming standard parameter-efficient fine-tuning baselines. However, the contrastive formulation can excessively penalize negative pairs, which manifests as characteristic failure cases such as over-counting and repetition. To address this issue, we propose a lightweight, reward-free post-training method that refines soft tokens by integrating contrastive alignment guidance directly into the score-matching objective of diffusion models. By assigning alignment directions at the score level, our approach mitigates these limitations and yields more coherent and semantically faithful generations. Experiments show that our method matches SoftREPA while substantially improving its failure cases, achieving over 35% improvement in counting accuracy on the GenEval benchmark. Our method is seamlessly applicable to existing diffusion backbones (SD1.5, SDXL, and SD3), and is complementary to existing RL-based diffusion post-training methods. Project page: https://jaayeon.github.io/AGSM

文生图扩散模型对齐优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。