English

EliGen: Entity-Level Controlled Image Generation with Regional Attention

Computer Vision and Pattern Recognition 2025-01-31 v3

Abstract

Recent advancements in diffusion models have significantly advanced text-to-image generation, yet global text prompts alone remain insufficient for achieving fine-grained control over individual entities within an image. To address this limitation, we present EliGen, a novel framework for Entity-level controlled image Generation. Firstly, we put forward regional attention, a mechanism for diffusion transformers that requires no additional parameters, seamlessly integrating entity prompts and arbitrary-shaped spatial masks. By contributing a high-quality dataset with fine-grained spatial and semantic entity-level annotations, we train EliGen to achieve robust and accurate entity-level manipulation, surpassing existing methods in both spatial precision and image quality. Additionally, we propose an inpainting fusion pipeline, extending its capabilities to multi-entity image inpainting tasks. We further demonstrate its flexibility by integrating it with other open-source models such as IP-Adapter, In-Context LoRA and MLLM, unlocking new creative possibilities. The source code, model, and dataset are published at https://github.com/modelscope/DiffSynth-Studio.git.

Keywords

Cite

@article{arxiv.2501.01097,
  title  = {EliGen: Entity-Level Controlled Image Generation with Regional Attention},
  author = {Hong Zhang and Zhongjie Duan and Xingjun Wang and Yingda Chen and Yu Zhang},
  journal= {arXiv preprint arXiv:2501.01097},
  year   = {2025}
}
R2 v1 2026-06-28T20:54:21.590Z