Doubly Abductive Counterfactual Inference for Text-based Image Editing

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Song, Xue, Cui, Jiequan, Zhang, Hanwang, Chen, Jingjing, Hong, Richang, Jiang, Yu-Gang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916176148299776
author Song, Xue
Cui, Jiequan
Zhang, Hanwang
Chen, Jingjing
Hong, Richang
Jiang, Yu-Gang
author_facet Song, Xue
Cui, Jiequan
Zhang, Hanwang
Chen, Jingjing
Hong, Richang
Jiang, Yu-Gang
contents We study text-based image editing (TBIE) of a single image by counterfactual inference because it is an elegant formulation to precisely address the requirement: the edited image should retain the fidelity of the original one. Through the lens of the formulation, we find that the crux of TBIE is that existing techniques hardly achieve a good trade-off between editability and fidelity, mainly due to the overfitting of the single-image fine-tuning. To this end, we propose a Doubly Abductive Counterfactual inference framework (DAC). We first parameterize an exogenous variable as a UNet LoRA, whose abduction can encode all the image details. Second, we abduct another exogenous variable parameterized by a text encoder LoRA, which recovers the lost editability caused by the overfitted first abduction. Thanks to the second abduction, which exclusively encodes the visual transition from post-edit to pre-edit, its inversion -- subtracting the LoRA -- effectively reverts pre-edit back to post-edit, thereby accomplishing the edit. Through extensive experiments, our DAC achieves a good trade-off between editability and fidelity. Thus, we can support a wide spectrum of user editing intents, including addition, removal, manipulation, replacement, style transfer, and facial change, which are extensively validated in both qualitative and quantitative evaluations. Codes are in https://github.com/xuesong39/DAC.
format Preprint
id arxiv_https___arxiv_org_abs_2403_02981
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Doubly Abductive Counterfactual Inference for Text-based Image Editing
Song, Xue
Cui, Jiequan
Zhang, Hanwang
Chen, Jingjing
Hong, Richang
Jiang, Yu-Gang
Computer Vision and Pattern Recognition
We study text-based image editing (TBIE) of a single image by counterfactual inference because it is an elegant formulation to precisely address the requirement: the edited image should retain the fidelity of the original one. Through the lens of the formulation, we find that the crux of TBIE is that existing techniques hardly achieve a good trade-off between editability and fidelity, mainly due to the overfitting of the single-image fine-tuning. To this end, we propose a Doubly Abductive Counterfactual inference framework (DAC). We first parameterize an exogenous variable as a UNet LoRA, whose abduction can encode all the image details. Second, we abduct another exogenous variable parameterized by a text encoder LoRA, which recovers the lost editability caused by the overfitted first abduction. Thanks to the second abduction, which exclusively encodes the visual transition from post-edit to pre-edit, its inversion -- subtracting the LoRA -- effectively reverts pre-edit back to post-edit, thereby accomplishing the edit. Through extensive experiments, our DAC achieves a good trade-off between editability and fidelity. Thus, we can support a wide spectrum of user editing intents, including addition, removal, manipulation, replacement, style transfer, and facial change, which are extensively validated in both qualitative and quantitative evaluations. Codes are in https://github.com/xuesong39/DAC.
title Doubly Abductive Counterfactual Inference for Text-based Image Editing
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.02981