Adapting Diffusion Models for Improved Prompt Compliance and Controllable Image Synthesis

Recent advances in generative modeling with diffusion processes (DPs) enabled breakthroughs in image synthesis. Despite impressive image quality, these models have various prompt compliance problems, including low recall in generating multiple objects, difficulty in generating text in images, and meeting constraints like object locations and pose. For fine-grained editing and manipulation, they also require fine-grained semantic or instance maps that are tedious to produce manually. While prompt compliance can be enhanced by addition of loss functions at inference, this is time consuming and does not scale to complex scenes. To overcome these limitations, this work introduces a new family of \textit{Factor Graph Diffusion Models} (FG-DMs) that models the joint distribution of images and conditioning variables, such as semantic, sketch, depth or normal maps via a factor graph decomposition. This joint structure has several advantages, including support for efficient sampling based prompt compliance schemes, which produce images of high object recall, semi-automated fine-grained editing, text-based editing of conditions with noise inversion, explainability at intermediate levels, ability to produce labeled datasets for the training of downstream models such as segmentation or depth, training with missing data, and continual learning where new conditioning variables can be added with minimal or no modifications to the existing structure. We propose an implementation of FG-DMs by adapting a pre-trained Stable Diffusion (SD) model to implement all FG-DM factors, using only COCO dataset, and show that it is effective in generating images with 15\% higher recall than SD while retaining its generalization ability. We introduce an attention distillation loss that encourages consistency among the attention maps of all factors, improving the fidelity of the generated conditions and image.

Paper

Similar papers

Peer review

Reviewer t2kD5/10 · confidence 3/52024-07-09

Summary

This paper propose a new family of diffusion model: FG-DM for better prompt compliance. Unlike traditional diffusion model, FG-DM models the joint distribution of images and conditioning variables like semantic, sketch, depth or normal maps via a factor graph. These extra factors largely boost prompt compliance compared to use text only. Users can also easily edit target image by modifying intermediate conditions. This work shows better results compared to vanilla diffusion.

Strengths

By decomposing the text into different factors such as seg maps, depth maps, etc., and using these factors as conditions for generation, this method achieves better prompt compliance. Additionally, the introduction of intermediate factors makes this method easier to edit.

Weaknesses

Since intermediate factors are generated sequentially, these method need more time for generation. The novelty of this method is limited.

Questions

1) How much more time do we need for the conditional generation chains? (for example, how much more time do we need when we use one or two conditions compared to original SD?) 2) If we use an off-the-shelf condition generator e.g. a text-to-depth map generator, and we fed the generated depth map to controlnet for generation, what is the advantage of our method?

Rating

5

Confidence

3

Soundness

4

Presentation

3

Contribution

2

Limitations

Yes

Reviewer Xxm85/10 · confidence 4/52024-07-11

Summary

This paper proposes a method for sequentially generating images using a frozen SD. Starting from a text prompt, the process iteratively generates several visual conditions and images, with each step depending on the previous ones. The model uses the VAE in SD to encode and decode visual conditions and employs an condition-specific adapter to introduce information into the SD for generating these conditions. This approach makes the image generation process more controllable, flexible, and easier to edit.

Strengths

1. The iterative generation method using a stable diffusion model is impressive. It provides strong control over the output while maintaining flexibility, aided by classifier-free guidance training. 2. The new method allows for flexible editing by modifying visual conditions and letting the model generate realistic content based on these changes.

Weaknesses

1. The newly introduced iterative process adds extra loop to each generation step. 2. Some components and experimental details are not well-explained.

Questions

1. The assumption behind attention distillation suggests that word tokens and visual regions remain constant, even if the visual regions are non-semantic, like poses. Why does this happen? It seems odd that the attention score of a general word like "person" can help generate the poses of a person. 2. Why do you use independent noise for different factors? 3. The experimental details are unclear. Does the model run with a single visual condition for each result, or does it generate all conditions sequentially? It would help to clearly present the generation process for each experiment, including Tables 1-5 and Figures 5-7. 4. How do you implement null conditions for different variables? 5. You claim that after choosing the best segmentation factor, the image synthesis only needs to run once. Why is this? Does the segmentation factor determine the quality of the generated images, or is your goal just to ensure the object appears in the generated image?

Rating

5

Confidence

4

Soundness

4

Presentation

2

Contribution

3

Limitations

The authors have adequately addressed the limitations.

Reviewer cyg35/10 · confidence 4/52024-07-11

Summary

The paper proposes unified generation framework of simultaneously generate image, segmap. depthmap, etc.

Strengths

The idea of simultaneous generation of images with various conditioning maps is very interesting.

Weaknesses

1. Although the method shows good performance in image editing for generated images, it does not include any discussion of whether the framework can be applied to real image editing case. Does it can be applied to inverted real images? Does the framework can still predict the segmentation map and keypoints of given real images? If that is possible, there would be great impact and future usage. If it is still applicable to real images, then please show some results which are similar to ControlNet. 2. The manuscript does not contain any qualitative comparison of ablation study. Please show some visual results on ablation study. 3. Please show the generated image results using handcrafted conditions, not the edited condition from generated conditions. For example, give the hand-drawn sketches (user-identified one) to the model and show the generated outputs.

Questions

See weakness

Rating

5

Confidence

4

Soundness

3

Presentation

3

Contribution

3

Limitations

Yes

Reviewer F3xm6/10 · confidence 4/52024-07-12

Summary

This paper introduces a Factor Graph Diffusion Models (FG-DMs) to address limitations in current generative models. FG-DMs model the joint distribution of images and conditioning variables using a factor graph decomposition, offering advantages like efficient sampling for better prompt compliance, fine-grained editing, and explainability. The method's effectiveness is validated on several datasets.

Strengths

This paper introduces a method to model the joint distribution of images and conditioning variables, enhancing prompt compliance and control over image synthesis. The Attention Distillation Loss is proposed to enhance the consistency and quality of generated images and conditions.

Weaknesses

The proposed method seems promising, however, it's very important to compare with current SOTA works to justify its effectiveness. The author only put Stable Diffusion in Table 5, which is not fair. Please include more recent works especially sine 2023 such as StyleGAN-T: Unlocking the Power of GANs for Fast Large-Scale Text-to-Image Synthesis, Make a Scene.

Questions

What's the training cost of your model? Is there a comparison with other methods?

Rating

6

Confidence

4

Soundness

3

Presentation

3

Contribution

3

Limitations

Yes.

Reviewer Xxm82024-08-10

Response to the rebuttal

Thank you for your response. Most of my concerns have been addressed.

Reviewer F3xm2024-08-11

Thanks for the detailed rebuttal. Most of my concerns are properly addressed.

Reviewer cyg32024-08-12

Thank you for your detailed rebuttal. Most of my concerns have been addressed. Just one little concern remaining is that there are so many concurrent and previous methods which try similar tasks. I leave the final decision to mete-reviewer whether this manuscript can go beyond the borderline of acceptance in this extremely competitive field.

Authorsrebuttal2024-08-13

We thank all the reviewers for their time and useful comments in improving the submission. We also thank all the reviewers for positively acknowledging the rebuttal. We differentiate the contributions of FG-DM from previous works below to alleviate any further concerns. - As compared to concurrent or existing works, e.g. 4M that requires training a model from scratch for generating conditions, the proposed FG-DM framework show that it is possible to **efficiently adapt existing models like Stable Diffusion (SD) to improve prompt compliance and controllability** (Table 4, rebuttal Table 1). We show that SD can faithfully reconstruct segmentation, depth, normal or sketch maps (Figure 17, 18) which allows for adaptation of the pretrained model. This has significant practical relevance. The fact that SD can simply be adapted to generate all the conditions represented as three channels and **generalize to concepts not covered by the finetuning dataset** (Figure 3, 5), despite not being trained on them, is a novel and somewhat surprising result. - A main **novelty of the paper is to propose a paradigm shift for prompt compliance research**. While prior works focus on inference based optimizations to improve prompt compliance, we show that sampling based prompt compliance is substantially faster (4 times) and better (Table 4). Furthermore, any future improvements in the samplers for diffusion models will automatically improve the performance of our sampling based method. While we only used the standard DDIM sampler in all our experiments, fast sampling is an active area of research, and fast samplers such as PNDM [1] or DDSS [2] can further boost our speed results. - Upon prompting by reviewer cyg3, we now show that **image inversion techniques such as LEDITS++ can be applied to the FG-DM to edit the conditions and the images directly based on text**. Figure 2 of the rebuttal shows that the segmentation masks can be directly edited by text prompts using our proposed FG-DM segmentation factor. **To the best of our knowledge, this method is the first to be able to directly edit the semantic masks from text prompts**. This adds a new editing dimension, which helps to automate the process and avoid the laborious task of manually editing the masks. We believe that this aspect alone can inspire various future papers in the topic of image editing. - We introduce a **novel attention distillation loss that improves the quality of the generated conditions and images** as shown in the paper and rebuttal (Table 2, rebuttal Figure 1) - We show that FG-DM also **supports training from scratch** on datasets such as CelebA, ADE20K or CityScapes (Figure 14, 15), and provides **synthetic data for data augmentation** in downstream tasks (Table 9, 10). Further, FG-DM can support continual learning where new factors such as pose can be added with minimal training. Overall, we believe that all these contributions make **FG-DM novel and distinct from existing works, offering valuable insights for further research within the text-to-image synthesis community**. [1] Luping Liu, Yi Ren, Zhijie Lin, Zhou Zhao. Pseudo Numerical Methods for Diffusion Models on Manifolds ICLR 2022. [2] Daniel Watson, William Chan, Jonathan Ho, Mohammad Norouzi. Learning Fast Samplers for Diffusion Models by Differentiating Through Sample Quality, ICLR 2022

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC