EDICT: Exact Diffusion Inversion via Coupled Transformations

By Bram Wallace, Akash Gokul, Nikhil Vijay Naik. Computer Vision and Pattern Recognition, 2022

308 citations recorded by the source corpus.

Finding an initial noise vector that produces an input image when fed into the diffusion process (known as inversion) is an important problem in denoising diffusion models (DDMs), with applications for real image editing. The standard approach for real image editing with inversion uses denoising diffusion implicit models (DDIMs [29]) to deterministically noise the image to the intermediate state along the path that the denoising would follow given the original conditioning. However, DDIM inversion for real images is unstable as it relies on local linearization assumptions, which result in the propagation of errors, leading to incorrect image reconstruction and loss of content. To alleviate these problems, we propose Exact Diffusion Inversion via Coupled Transformations (EDICT), an inversion method that draws inspiration from affine coupling layers. EDICT enables mathematically exact inversion of real and model-generated images by maintaining two coupled noise vectors which are used to invert each other in an alternating fashion. Using Stable Diffusion [25], a state-of-the-art latent diffusion model, we demonstrate that EDICT successfully reconstructs real images with high fidelity. On complex image datasets like MS-COCO, EDICT reconstruction significantly outperforms DDIM, improving the mean square error of reconstruction by a factor of two. Using noise vectors inverted from real images, EDICT enables a wide range of image edits—from local and global semantic edits to image stylization—while maintaining fidelity to the original image structure. EDICT requires no model training/finetuning, prompt tuning, or extra data and can be combined with any pretrained DDM.

Subfields: Computer Vision and Pattern Recognition, Statistical and Nonlinear Physics

View Paper: https://www.semanticscholar.org/paper/7e80f79472d9b5aaa109075910d3ff9f9149f4c9

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Ming-Hsuan Yang
  3. Dacheng Tao
  4. Steven C. H. Hoi
  5. Xiaogang Wang
  6. Trevor Darrell
  7. Chen Change Loy
  8. Yu Qiao

In Statistical and Nonlinear Physics

  1. Oron Ashual
  2. Feng Yang
  3. Dimitris N. Metaxas
  4. Peyman Milanfar
  5. Steven C. H. Hoi
  6. Tomer Michaeli
  7. Yuval Kirstain
  8. Vladimir Kulikov

Citations in the corpus