A Cat Is a Cat (Not a Dog!): Unraveling Information Mix-Ups in Text-to-Image Encoders Through Causal Analysis and Embedding Optimization
Abstract
This paper analyzes the impact of causal manner in the text encoder of text-to-image (T2I) diffusion models, which can lead to information bias and loss. Previous works have focused on addressing the issues through the denoising process. However, there is no research discussing how text embedding contributes to T2I models, especially when generating more than one object. In this paper, we share a comprehensive analysis of text embedding: i) how text embedding contributes to the generated images and ii) why information gets lost and biases towards the first-mentioned object. Accordingly, we propose a simple but effective text embedding balance optimization method, which is training-free, with an improvement of 125.42\% on information balance in stable diffusion. Furthermore, we propose a new automatic evaluation metric that quantifies information loss more accurately than existing methods, achieving 81\% concordance with human assessments. This metric effectively measures the presence and accuracy of objects, addressing the limitations of current distribution scores like CLIP's text-image similarities.
Cite
Text
Chen et al. "A Cat Is a Cat (Not a Dog!): Unraveling Information Mix-Ups in Text-to-Image Encoders Through Causal Analysis and Embedding Optimization." Neural Information Processing Systems, 2024. doi:10.52202/079017-1847Markdown
[Chen et al. "A Cat Is a Cat (Not a Dog!): Unraveling Information Mix-Ups in Text-to-Image Encoders Through Causal Analysis and Embedding Optimization." Neural Information Processing Systems, 2024.](https://mlanthology.org/neurips/2024/chen2024neurips-cat/) doi:10.52202/079017-1847BibTeX
@inproceedings{chen2024neurips-cat,
title = {{A Cat Is a Cat (Not a Dog!): Unraveling Information Mix-Ups in Text-to-Image Encoders Through Causal Analysis and Embedding Optimization}},
author = {Chen, Chieh-Yun and Tseng, Chiang and Tsao, Li-Wu and Shuai, Hong-Han},
booktitle = {Neural Information Processing Systems},
year = {2024},
doi = {10.52202/079017-1847},
url = {https://mlanthology.org/neurips/2024/chen2024neurips-cat/}
}