Supercharged One-step Text-to-Image Diffusion Models with Negative Prompts

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Nguyen, Viet, Nguyen, Anh, Dao, Trung, Nguyen, Khoi, Pham, Cuong, Tran, Toan, Tran, Anh
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916967796965376
author Nguyen, Viet
Nguyen, Anh
Dao, Trung
Nguyen, Khoi
Pham, Cuong
Tran, Toan
Tran, Anh
author_facet Nguyen, Viet
Nguyen, Anh
Dao, Trung
Nguyen, Khoi
Pham, Cuong
Tran, Toan
Tran, Anh
contents The escalating demand for real-time image synthesis has driven significant advancements in one-step diffusion models, which inherently offer expedited generation speeds compared to traditional multi-step methods. However, this enhanced efficiency is frequently accompanied by a compromise in the controllability of image attributes. While negative prompting, typically implemented via classifier-free guidance (CFG), has proven effective for fine-grained control in multi-step models, its application to one-step generators remains largely unaddressed. Due to the lack of iterative refinement, as in multi-step diffusion, directly applying CFG to one-step generation leads to blending artifacts and diminished output quality. To fill this gap, we introduce \textbf{N}egative-\textbf{A}way \textbf{S}teer \textbf{A}ttention (NASA), an efficient method that integrates negative prompts into one-step diffusion models. NASA operates within the intermediate representation space by leveraging cross-attention mechanisms to suppress undesired visual attributes. This strategy avoids the blending artifacts inherent in output-space guidance and achieves high efficiency, incurring only a minimal 1.89\% increase in FLOPs compared to the computational doubling of CFG. Furthermore, NASA can be seamlessly integrated into existing timestep distillation frameworks, enhancing the student's output quality. Experimental results demonstrate that NASA substantially improves controllability and output quality, achieving an HPSv2 score of \textbf{31.21}, setting a new state-of-the-art benchmark for one-step diffusion models.
format Preprint
id arxiv_https___arxiv_org_abs_2412_02687
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Supercharged One-step Text-to-Image Diffusion Models with Negative Prompts
Nguyen, Viet
Nguyen, Anh
Dao, Trung
Nguyen, Khoi
Pham, Cuong
Tran, Toan
Tran, Anh
Computer Vision and Pattern Recognition
The escalating demand for real-time image synthesis has driven significant advancements in one-step diffusion models, which inherently offer expedited generation speeds compared to traditional multi-step methods. However, this enhanced efficiency is frequently accompanied by a compromise in the controllability of image attributes. While negative prompting, typically implemented via classifier-free guidance (CFG), has proven effective for fine-grained control in multi-step models, its application to one-step generators remains largely unaddressed. Due to the lack of iterative refinement, as in multi-step diffusion, directly applying CFG to one-step generation leads to blending artifacts and diminished output quality. To fill this gap, we introduce \textbf{N}egative-\textbf{A}way \textbf{S}teer \textbf{A}ttention (NASA), an efficient method that integrates negative prompts into one-step diffusion models. NASA operates within the intermediate representation space by leveraging cross-attention mechanisms to suppress undesired visual attributes. This strategy avoids the blending artifacts inherent in output-space guidance and achieves high efficiency, incurring only a minimal 1.89\% increase in FLOPs compared to the computational doubling of CFG. Furthermore, NASA can be seamlessly integrated into existing timestep distillation frameworks, enhancing the student's output quality. Experimental results demonstrate that NASA substantially improves controllability and output quality, achieving an HPSv2 score of \textbf{31.21}, setting a new state-of-the-art benchmark for one-step diffusion models.
title Supercharged One-step Text-to-Image Diffusion Models with Negative Prompts
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2412.02687