Local Conditional Controlling for Text-to-Image Diffusion Models

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Zhao, Yibo, Peng, Liang, Yang, Yang, Luo, Zekai, Li, Hengjia, Chen, Yao, Yang, Zheng, He, Xiaofei, Zhao, Wei, lu, qinglin, Wu, Boxi, Liu, Wei
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866914919938523136
author Zhao, Yibo
Peng, Liang
Yang, Yang
Luo, Zekai
Li, Hengjia
Chen, Yao
Yang, Zheng
He, Xiaofei
Zhao, Wei
lu, qinglin
Wu, Boxi
Liu, Wei
author_facet Zhao, Yibo
Peng, Liang
Yang, Yang
Luo, Zekai
Li, Hengjia
Chen, Yao
Yang, Zheng
He, Xiaofei
Zhao, Wei
lu, qinglin
Wu, Boxi
Liu, Wei
contents Diffusion models have exhibited impressive prowess in the text-to-image task. Recent methods add image-level structure controls, e.g., edge and depth maps, to manipulate the generation process together with text prompts to obtain desired images. This controlling process is globally operated on the entire image, which limits the flexibility of control regions. In this paper, we explore a novel and practical task setting: local control. It focuses on controlling specific local region according to user-defined image conditions, while the remaining regions are only conditioned by the original text prompt. However, it is non-trivial to achieve local conditional controlling. The naive manner of directly adding local conditions may lead to the local control dominance problem, which forces the model to focus on the controlled region and neglect object generation in other regions. To mitigate this problem, we propose Regional Discriminate Loss to update the noised latents, aiming at enhanced object generation in non-control regions. Furthermore, the proposed Focused Token Response suppresses weaker attention scores which lack the strongest response to enhance object distinction and reduce duplication. Lastly, we adopt Feature Mask Constraint to reduce quality degradation in images caused by information differences across the local control region. All proposed strategies are operated at the inference stage. Extensive experiments demonstrate that our method can synthesize high-quality images aligned with the text prompt under local control conditions.
format Preprint
id arxiv_https___arxiv_org_abs_2312_08768
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Local Conditional Controlling for Text-to-Image Diffusion Models
Zhao, Yibo
Peng, Liang
Yang, Yang
Luo, Zekai
Li, Hengjia
Chen, Yao
Yang, Zheng
He, Xiaofei
Zhao, Wei
lu, qinglin
Wu, Boxi
Liu, Wei
Computer Vision and Pattern Recognition
Diffusion models have exhibited impressive prowess in the text-to-image task. Recent methods add image-level structure controls, e.g., edge and depth maps, to manipulate the generation process together with text prompts to obtain desired images. This controlling process is globally operated on the entire image, which limits the flexibility of control regions. In this paper, we explore a novel and practical task setting: local control. It focuses on controlling specific local region according to user-defined image conditions, while the remaining regions are only conditioned by the original text prompt. However, it is non-trivial to achieve local conditional controlling. The naive manner of directly adding local conditions may lead to the local control dominance problem, which forces the model to focus on the controlled region and neglect object generation in other regions. To mitigate this problem, we propose Regional Discriminate Loss to update the noised latents, aiming at enhanced object generation in non-control regions. Furthermore, the proposed Focused Token Response suppresses weaker attention scores which lack the strongest response to enhance object distinction and reduce duplication. Lastly, we adopt Feature Mask Constraint to reduce quality degradation in images caused by information differences across the local control region. All proposed strategies are operated at the inference stage. Extensive experiments demonstrate that our method can synthesize high-quality images aligned with the text prompt under local control conditions.
title Local Conditional Controlling for Text-to-Image Diffusion Models
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.08768