🤖 AI Summary
This study addresses the limitations of text-to-image models regarding continuous concept guidance and unreliable local structure generation by proposing a training-free, gradient-free method for precise concept control. By quantifying inter-layer concept mutual information to reveal localization properties in generation, the approach employs a skip-layer prediction weighted combination strategy to achieve accurate denoising guidance within the latent space. Experimental results demonstrate that this method significantly enhances multi-object generation performance and local consistency across mainstream architectures, including PixArt-alpha, SD3, and FLUX.1. Consequently, this work establishes an efficient and reliable paradigm for concept-level regulation in text-to-image synthesis, overcoming critical challenges in fine-grained semantic control without requiring additional model retraining or gradient computation.
📝 Abstract
Text-to-image diffusion models have two major drawbacks that severely limit their practical utility: (1) standard models lack an intrinsic mechanism for continuous, concept-specific guidance (e.g., for precisely controlling how aesthetically pleasing an image looks), and (2) they lack reliability for tasks requiring high local coherence (e.g., generating text or human hands). To tackle these issues, we introduce a novel notion of concept-wise mutual information and find large, concept-dependent differences between individual layers, demonstrating that the generation of specific structures is localized in distinct parts of the network. We exploit this insight by reinforcing the impact of concept-relevant layers in Concept Guidance (CoG), a precise, target-specific guidance method that works for models out-of-the-box without additional training, external models, gradients, or prompt engineering. CoG first quantifies each layer's concept-specific impact and then guides denoising using a weighted combination of predictions generated with concept-relevant layers skipped. We demonstrate performance increases across various targets and popular models like PixArt-alpha, SD3, SD3.5, and FLUX.1-dev. Code is available at https://github.com/CompVis/concept_guidance