Towards Understanding Cross and Self-Attention in Stable Diffusion for Text-Guided Image Editing

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liu, Bingyan, Wang, Chengyu, Cao, Tingfeng, Jia, Kui, Huang, Jun
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914704943742976
author Liu, Bingyan
Wang, Chengyu
Cao, Tingfeng
Jia, Kui
Huang, Jun
author_facet Liu, Bingyan
Wang, Chengyu
Cao, Tingfeng
Jia, Kui
Huang, Jun
contents Deep Text-to-Image Synthesis (TIS) models such as Stable Diffusion have recently gained significant popularity for creative Text-to-image generation. Yet, for domain-specific scenarios, tuning-free Text-guided Image Editing (TIE) is of greater importance for application developers, which modify objects or object properties in images by manipulating feature components in attention layers during the generation process. However, little is known about what semantic meanings these attention layers have learned and which parts of the attention maps contribute to the success of image editing. In this paper, we conduct an in-depth probing analysis and demonstrate that cross-attention maps in Stable Diffusion often contain object attribution information that can result in editing failures. In contrast, self-attention maps play a crucial role in preserving the geometric and shape details of the source image during the transformation to the target image. Our analysis offers valuable insights into understanding cross and self-attention maps in diffusion models. Moreover, based on our findings, we simplify popular image editing methods and propose a more straightforward yet more stable and efficient tuning-free procedure that only modifies self-attention maps of the specified attention layers during the denoising process. Experimental results show that our simplified method consistently surpasses the performance of popular approaches on multiple datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2403_03431
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Understanding Cross and Self-Attention in Stable Diffusion for Text-Guided Image Editing
Liu, Bingyan
Wang, Chengyu
Cao, Tingfeng
Jia, Kui
Huang, Jun
Computer Vision and Pattern Recognition
Deep Text-to-Image Synthesis (TIS) models such as Stable Diffusion have recently gained significant popularity for creative Text-to-image generation. Yet, for domain-specific scenarios, tuning-free Text-guided Image Editing (TIE) is of greater importance for application developers, which modify objects or object properties in images by manipulating feature components in attention layers during the generation process. However, little is known about what semantic meanings these attention layers have learned and which parts of the attention maps contribute to the success of image editing. In this paper, we conduct an in-depth probing analysis and demonstrate that cross-attention maps in Stable Diffusion often contain object attribution information that can result in editing failures. In contrast, self-attention maps play a crucial role in preserving the geometric and shape details of the source image during the transformation to the target image. Our analysis offers valuable insights into understanding cross and self-attention maps in diffusion models. Moreover, based on our findings, we simplify popular image editing methods and propose a more straightforward yet more stable and efficient tuning-free procedure that only modifies self-attention maps of the specified attention layers during the denoising process. Experimental results show that our simplified method consistently surpasses the performance of popular approaches on multiple datasets.
title Towards Understanding Cross and Self-Attention in Stable Diffusion for Text-Guided Image Editing
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.03431