RealCompo: Balancing Realism and Compositionality Improves Text-to-Image Diffusion Models

Xinchen Zhang, Ling Yang, Yaqi Cai, Zhaochen Yu, Kai-Ni Wang +6 more
2/20/2024
cs.CVcs.AIcs.LG

Official code: YangLing0818/RealCompo (122 stars, last commit 2024-11-14).

Abstract

Diffusion models have achieved remarkable advancements in text-to-image generation. However, existing models still have many difficulties when faced with multiple-object compositional generation. In this paper, we propose RealCompo, a new training-free and transferred-friendly text-to-image generation framework, which aims to leverage the respective advantages of text-to-image models and spatial-aware image diffusion models (e.g., layout, keypoints and segmentation maps) to enhance both realism and compositionality of the generated images. An intuitive and novel balancer is proposed to dynamically balance the strengths of the two models in denoising process, allowing plug-and-play use of any model without extra training. Extensive experiments show that our RealCompo consistently outperforms state-of-the-art text-to-image models and spatial-aware image diffusion models in multiple-object compositional generation while keeping satisfactory realism and compositionality of the generated images. Notably, our RealCompo can be seamlessly extended with a wide range of spatial-aware image diffusion models and stylized diffusion models. Our code is available at: https://github.com/YangLing0818/RealCompo

View on arXivView PDF

Code Implementations(1)

[NeurIPS 2024] RealCompo: Balancing Realism and Compositionality Improves Text-to-Image Diffusion Models

1223PythonFeb 20, 20241 years ago
diffusion-modelslayout-to-imagemixture-of-expertstext-to-image-diffusion

Similar Papers

Cite this paper

@article{zhang2024realcompo,
  title  = {RealCompo: Balancing Realism and Compositionality Improves Text-to-Image Diffusion Models},
  author = {Xinchen Zhang and Ling Yang and Yaqi Cai and Zhaochen Yu and Kai-Ni Wang and Jiake Xie and Ye Tian and Minkai Xu and Yong Tang and Yujiu Yang and Bin Cui},
  year   = {2024},
  eprint = {2402.12908},
  archivePrefix = {arXiv},
  doi    = {10.52202/079017-3074},
  url    = {http://arxiv.org/abs/2402.12908v3},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.12908.svg)](https://codeofpaper.com/paper/2402.12908)

Discussion