用证据学习提升文本图像生成中物体一致性
ELDiff: When Evidential Learning Meets Text-to-Image Diffusion

- 引入证据正则化与冲突检测机制,缓解分割图偏差问题
- 在多个主流模型上超越现有训练与免训练方法,无需推理时调整
- 适合关注生成质量与语义一致性的图像生成研究者
在多对象文本到图像扩散模型中,确保文本提示与生成图像内容之间的语义一致性至关重要,但这一约束常被扩散过程忽视。尽管基于标记的扩散模型可通过学习图像内容与物体分割图间的对象级一致性来缓解该问题,却易受分割图偏差和语义重叠冲突影响,尤其在多对象场景下。本文提出ELDiff,一种基于证据学习监督的文本到图像扩散模型,利用不确定性度量与冲突检测增强对不可靠分割图的容错能力,并抑制语义冲突,强化对象级一致性学习。具体而言,设计像素证据损失,通过证据正则化抑制对不可靠标签的过度自信;引入标记冲突损失,通过优化测量冲突因子减弱语义矛盾。大量实验表明,ELDiff在SD v1.4、SD v2.1、SDXL、SD v3.5和Qwen-Image上均优于现有基于训练和免训练的方法,且无需额外推理阶段操作。值得注意的是,ELDiff可无缝集成至现有文本到图像扩散模型训练流程。代码已公开于https://github.com/QingtaoPan/ELDiff。
原文摘要 · Abstract (English)
In multi-object text-to-image (T2I) diffusion, ensuring semantic consistency between textual prompts and generated visual content is crucial for image synthesis. However, such consistency constraint is often underemphasized in the denoising process of diffusion models. Although token supervised diffusion models can mitigate this issue by learning object-wise consistency between the image content and object segmentation maps, it tends to suffer from the problems of segmentation map bias and semantic overlap conflict, especially when involving multiple objects. In this paper, we propose ELDiff, a new evidential learning-supervised T2I diffusion model, which leverages the advantages of uncertainty metric and conflict detection to enhance the fault tolerance of unreliable segmentation maps and suppress semantic conflicts, strengthening object-wise consistency learning. Specifically, a pixel evidence loss is proposed to restrain overconfidence in unreliable labels through evidential regularization, and a token conflict loss is designed to weaken the contradiction between semantics through optimizing a measured conflict factor. Extensive experiments show that our ELDiff outperforms existing training based and train-free based T2I diffusion models on SD v1.4, SD v2.1, SDXL, SD v3.5, and Qwen-Image, without requiring additional inference-time manipulations. Notably, ELDiff can be seamlessly extended to the existing training pipeline of T2I diffusion models. Code can be found at https://github.com/QingtaoPan/ELDiff.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。