Can We Perform Online RL for Image Editing without Editing Rewards?
2026-08-24 • Computer Vision and Pattern Recognition
Computer Vision and Pattern Recognition
AI summaryⓘ
The authors explore how to improve image editing using reinforcement learning (RL) without relying on hard-to-get editing-specific rewards. They propose mapping image editing goals into rewards used in text-to-image (T2I) generation, like image quality and how well the image matches a description. Since editing instructions focus on changes rather than full descriptions, the authors develop Lever-Edit, a system that creates suitable captions for target images and uses only the T2I rewards to train editing. Their experiments show this method works well, matching or beating traditional approaches that need special editing rewards.
reinforcement learningimage editingtext-to-image generationreward functionsemantic alignmentcaptioningpolicy optimizationvisual preferenceslever-edit
Authors
Qichao Ma, Jikang Cheng, Ling Liang, Zhaofei Yu, Tiejun Huang, Renye Yan
Abstract
Reinforcement learning (RL) enables direct preference optimization for image editing through editing-specific rewards, which remain less developed due to costly triplet supervision and complex task-dependent calibration. In contrast, text-to-image (T2I) generation benefits from a mature and diverse reward ecosystem spanning semantic alignment, aesthetics, realism, glyph shape, and other visual preferences. Extending this ecosystem to image editing would substantially broaden the range of visual preferences accessible to RL-based optimization, prompting the central question: \emph{Can We Perform Image Editing RL without Editing Rewards?} In this paper, we argue that the standard image editing dimensions have potential to be mapped to the T2I reward space: image quality can transfer directly, prompt following can be aligned through a description of the desired visual state, and reference consistency admits a coarse semantic conversion by encoding the source content to preserve. However, editing instructions specify relative changes, whereas T2I rewards require self-contained target descriptions; moreover, semantically valid captions from generic vision-language models may be incompatible with the frozen reward. Hence, we further introduce Lever-Edit, a two-stage framework that learns a reward-aligned captioner for counterfactual target descriptions, freezes it, and optimizes the editing policy solely with the transferred T2I reward. Experiments show competitive editing alignment and source preservation against editing-reward-based fine-tuning, while outperforming intuitive transfer baselines.