An Item is Worth a Prompt: Versatile Image Editing with Disentangled Control

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Feng, Aosong, Qiu, Weikang, Bai, Jinbin, Zhang, Xiao, Dong, Zhen, Zhou, Kaicheng, Ying, Rex, Tassiulas, Leandros
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866910800212393984
author Feng, Aosong
Qiu, Weikang
Bai, Jinbin
Zhang, Xiao
Dong, Zhen
Zhou, Kaicheng
Ying, Rex
Tassiulas, Leandros
author_facet Feng, Aosong
Qiu, Weikang
Bai, Jinbin
Zhang, Xiao
Dong, Zhen
Zhou, Kaicheng
Ying, Rex
Tassiulas, Leandros
contents Building on the success of text-to-image diffusion models (DPMs), image editing is an important application to enable human interaction with AI-generated content. Among various editing methods, editing within the prompt space gains more attention due to its capacity and simplicity of controlling semantics. However, since diffusion models are commonly pretrained on descriptive text captions, direct editing of words in text prompts usually leads to completely different generated images, violating the requirements for image editing. On the other hand, existing editing methods usually consider introducing spatial masks to preserve the identity of unedited regions, which are usually ignored by DPMs and therefore lead to inharmonic editing results. Targeting these two challenges, in this work, we propose to disentangle the comprehensive image-prompt interaction into several item-prompt interactions, with each item linked to a special learned prompt. The resulting framework, named D-Edit, is based on pretrained diffusion models with cross-attention layers disentangled and adopts a two-step optimization to build item-prompt associations. Versatile image editing can then be applied to specific items by manipulating the corresponding prompts. We demonstrate state-of-the-art results in four types of editing operations including image-based, text-based, mask-based editing, and item removal, covering most types of editing applications, all within a single unified framework. Notably, D-Edit is the first framework that can (1) achieve item editing through mask editing and (2) combine image and text-based editing. We demonstrate the quality and versatility of the editing results for a diverse collection of images through both qualitative and quantitative evaluations.
format Preprint
id arxiv_https___arxiv_org_abs_2403_04880
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle An Item is Worth a Prompt: Versatile Image Editing with Disentangled Control
Feng, Aosong
Qiu, Weikang
Bai, Jinbin
Zhang, Xiao
Dong, Zhen
Zhou, Kaicheng
Ying, Rex
Tassiulas, Leandros
Computer Vision and Pattern Recognition
Building on the success of text-to-image diffusion models (DPMs), image editing is an important application to enable human interaction with AI-generated content. Among various editing methods, editing within the prompt space gains more attention due to its capacity and simplicity of controlling semantics. However, since diffusion models are commonly pretrained on descriptive text captions, direct editing of words in text prompts usually leads to completely different generated images, violating the requirements for image editing. On the other hand, existing editing methods usually consider introducing spatial masks to preserve the identity of unedited regions, which are usually ignored by DPMs and therefore lead to inharmonic editing results. Targeting these two challenges, in this work, we propose to disentangle the comprehensive image-prompt interaction into several item-prompt interactions, with each item linked to a special learned prompt. The resulting framework, named D-Edit, is based on pretrained diffusion models with cross-attention layers disentangled and adopts a two-step optimization to build item-prompt associations. Versatile image editing can then be applied to specific items by manipulating the corresponding prompts. We demonstrate state-of-the-art results in four types of editing operations including image-based, text-based, mask-based editing, and item removal, covering most types of editing applications, all within a single unified framework. Notably, D-Edit is the first framework that can (1) achieve item editing through mask editing and (2) combine image and text-based editing. We demonstrate the quality and versatility of the editing results for a diverse collection of images through both qualitative and quantitative evaluations.
title An Item is Worth a Prompt: Versatile Image Editing with Disentangled Control
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.04880