CoCoEdit: Content-Consistent Image Editing via Region Regularized Reinforcement Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wu, Yuhui, Xie, Chenxi, Li, Ruibin, Chen, Liyi, Yi, Qiaosi, Zhang, Lei
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910220455772160
author Wu, Yuhui
Xie, Chenxi
Li, Ruibin
Chen, Liyi
Yi, Qiaosi
Zhang, Lei
author_facet Wu, Yuhui
Xie, Chenxi
Li, Ruibin
Chen, Liyi
Yi, Qiaosi
Zhang, Lei
contents Image editing has achieved impressive results with the development of large-scale generative models. However, existing models mainly focus on the editing effects of intended objects and regions, often leading to unwanted changes in unintended regions. We present a post-training framework for Content-Consistent Editing (CoCoEdit) via region regularized reinforcement learning. We first augment existing editing datasets with refined instructions and masks, from which 40K diverse and high quality samples are curated as training set. We then introduce a pixel-level similarity reward to complement MLLM-based rewards, enabling models to ensure both editing quality and content consistency during the editing process. To overcome the spatial-agnostic nature of the rewards, we propose a region-based regularizer, aiming to preserve non-edited regions for high-reward samples while encouraging editing effects for low-reward samples. For evaluation, we annotate editing masks for GEdit-Bench and ImgEdit-Bench, introducing pixel-level similarity metrics to measure content consistency and editing quality. Applying CoCoEdit to Qwen-Image-Edit and FLUX-Kontext, we achieve not only competitive editing scores with state-of-the-art models, but also significantly better content consistency, measured by PSNR/SSIM metrics and human subjective ratings.
format Preprint
id arxiv_https___arxiv_org_abs_2602_14068
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle CoCoEdit: Content-Consistent Image Editing via Region Regularized Reinforcement Learning
Wu, Yuhui
Xie, Chenxi
Li, Ruibin
Chen, Liyi
Yi, Qiaosi
Zhang, Lei
Computer Vision and Pattern Recognition
Image editing has achieved impressive results with the development of large-scale generative models. However, existing models mainly focus on the editing effects of intended objects and regions, often leading to unwanted changes in unintended regions. We present a post-training framework for Content-Consistent Editing (CoCoEdit) via region regularized reinforcement learning. We first augment existing editing datasets with refined instructions and masks, from which 40K diverse and high quality samples are curated as training set. We then introduce a pixel-level similarity reward to complement MLLM-based rewards, enabling models to ensure both editing quality and content consistency during the editing process. To overcome the spatial-agnostic nature of the rewards, we propose a region-based regularizer, aiming to preserve non-edited regions for high-reward samples while encouraging editing effects for low-reward samples. For evaluation, we annotate editing masks for GEdit-Bench and ImgEdit-Bench, introducing pixel-level similarity metrics to measure content consistency and editing quality. Applying CoCoEdit to Qwen-Image-Edit and FLUX-Kontext, we achieve not only competitive editing scores with state-of-the-art models, but also significantly better content consistency, measured by PSNR/SSIM metrics and human subjective ratings.
title CoCoEdit: Content-Consistent Image Editing via Region Regularized Reinforcement Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2602.14068