Tuning-Free Inversion-Enhanced Control for Consistent Image Editing

Authors

  • Xiaoyue Duan School of Automation Science and Electrical Engineering, Beihang University, China Meituan
  • Shuhao Cui Meituan
  • Guoliang Kang School of Automation Science and Electrical Engineering, Beihang University, China Zhongguancun Laboratory, Beijing, China
  • Baochang Zhang School of Automation Science and Electrical Engineering, Beihang University, China Zhongguancun Laboratory, Beijing, China Hangzhou Research Institute, Beihang University, China Nanchang Institute of Technology, Nanchang, China
  • Zhengcong Fei Meituan
  • Mingyuan Fan Meituan
  • Junshi Huang Meituan

DOI:

https://doi.org/10.1609/aaai.v38i2.27931

Keywords:

CV: Computational Photography, Image & Video Synthesis, CV: Applications, CV: Multi-modal Vision, ML: Deep Generative Models & Autoencoders

Abstract

Consistent editing of real images is a challenging task, as it requires performing non-rigid edits (e.g., changing postures) to the main objects in the input image without changing their identity or attributes. To guarantee consistent attributes, some existing methods fine-tune the entire model or the textual embedding for structural consistency, but they are time-consuming and fail to perform non-rigid edits. Other works are tuning-free, but their performances are weakened by the quality of Denoising Diffusion Implicit Model (DDIM) reconstruction, which often fails in real-world scenarios. In this paper, we present a novel approach called Tuning-free Inversion-enhanced Control (TIC), which directly correlates features from the inversion process with those from the sampling process to mitigate the inconsistency in DDIM reconstruction. Specifically, our method effectively obtains inversion features from the key and value features in the self-attention layers, and enhances the sampling process by these inversion features, thus achieving accurate reconstruction and content-consistent editing. To extend the applicability of our method to general editing scenarios, we also propose a mask-guided attention concatenation strategy that combines contents from both the inversion and the naive DDIM editing processes. Experiments show that the proposed method outperforms previous works in reconstruction and consistent editing, and produces impressive results in various settings.

Published

2024-03-24

How to Cite

Duan, X., Cui, S., Kang, G., Zhang, B., Fei, Z., Fan, M., & Huang, J. (2024). Tuning-Free Inversion-Enhanced Control for Consistent Image Editing. Proceedings of the AAAI Conference on Artificial Intelligence, 38(2), 1644–1652. https://doi.org/10.1609/aaai.v38i2.27931

Issue

Section

AAAI Technical Track on Computer Vision I