RefineEdit lets users fix parts of AI image edits without retraining

Refinement Is Inherently Editable: Training-Free Prompt-to-Prompt Image Editing with Generative Refinement Network

Computer Vision and Pattern Recognition

Summary

Editing AI-generated images can be tricky because some methods can only change parts of the image in a fixed order or need extra tools like masks. The authors present RefineEdit, a new way to edit images generated by AI that doesn’t need extra training or complicated controls. It works by carefully deciding which parts of the image code to change while keeping other parts the same, allowing easy revisions even after some changes have been made. This method preserves the parts of the image you don’t want to change better than current approaches.

What this means in practice

  • For graphic designers: Edit AI-generated images by refining specific parts without retraining or extra tools, improving background preservation and overall image quality.
  • For advertising content teams: Create customized images with reliable localized edits that keep original elements intact, speeding up content revision workflows.

Authors

Yulong Chen, Ziqian Zhang, Haoyu Zhang, Ao He, Senmao Li, Kai Wang

Abstract

Text-guided image editing must introduce the requested changes while preserving unrelated source content. Diffusion-based editors rely on spatial controls whose inaccuracies can leave edits incomplete or alter unrelated regions. Causal autoregressive editors face a further constraint: their fixed decoding order limits revision of earlier decisions. We introduce RefineEdit, a training-free prompt-to-prompt image editing framework built on a Generative Refinement Network. Our key idea is to couple edit localization with content generation through the global refinement of binary image codes, allowing editing evidence to be reassessed as the image evolves. RefineEdit initializes an editing branch from an intermediate source state, reusing the emerging layout. We compare the probabilities assigned by the two branches to the same source-sampled bits, using their signed differences to select editable positions and bits. Selected bits follow editing refinement, while the remaining bits copy the evolving source state. To stabilize editing across refinement steps, adaptive spatial freezing limits unnecessary mask expansion, while finite bit locking keeps recently selected bits editable. The framework requires no additional training, external masks, or attention control. Across nine editing categories of PIE-Bench, RefineEdit achieves the best background-preservation scores in PSNR, LPIPS, MSE and SSIM, together with the highest whole-image and edited-region CLIP scores among the evaluated methods.