ThinkRL-Edit: Thinking in Reinforcement Learning for Reasoning-Centric Image Editing

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Li, Hengjia, Jiang, Liming, Yan, Qing, Song, Yizhi, Kang, Hao, Liu, Zichuan, Lu, Xin, Wu, Boxi, Cai, Deng
Format: Preprint
Veröffentlicht: 2026
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911469280428032
author Li, Hengjia
Jiang, Liming
Yan, Qing
Song, Yizhi
Kang, Hao
Liu, Zichuan
Lu, Xin
Wu, Boxi
Cai, Deng
author_facet Li, Hengjia
Jiang, Liming
Yan, Qing
Song, Yizhi
Kang, Hao
Liu, Zichuan
Lu, Xin
Wu, Boxi
Cai, Deng
contents Instruction-driven image editing with unified multimodal generative models has advanced rapidly, yet their underlying visual reasoning remains limited, leading to suboptimal performance on reasoning-centric edits. Reinforcement learning (RL) has been investigated for improving the quality of image editing, but it faces three key challenges: (1) limited reasoning exploration confined to denoising stochasticity, (2) biased reward fusion, and (3) unstable VLM-based instruction rewards. In this work, we propose ThinkRL-Edit, a reasoning-centric RL framework that decouples visual reasoning from image synthesis and expands reasoning exploration beyond denoising. To the end, we introduce Chain-of-Thought (CoT)-based reasoning sampling with planning and reflection stages prior to generation in online sampling, compelling the model to explore multiple semantic hypotheses and validate their plausibility before committing to a visual outcome. To avoid the failures of weighted aggregation, we propose an unbiased chain preference grouping strategy across multiple reward dimensions. Moreover, we replace interval-based VLM scores with a binary checklist, yielding more precise, lower-variance, and interpretable rewards for complex reasoning. Experiments show our method significantly outperforms prior work on reasoning-centric image editing, producing instruction-faithful, visually coherent, and semantically grounded edits.
format Preprint
id arxiv_https___arxiv_org_abs_2601_03467
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ThinkRL-Edit: Thinking in Reinforcement Learning for Reasoning-Centric Image Editing
Li, Hengjia
Jiang, Liming
Yan, Qing
Song, Yizhi
Kang, Hao
Liu, Zichuan
Lu, Xin
Wu, Boxi
Cai, Deng
Computer Vision and Pattern Recognition
Instruction-driven image editing with unified multimodal generative models has advanced rapidly, yet their underlying visual reasoning remains limited, leading to suboptimal performance on reasoning-centric edits. Reinforcement learning (RL) has been investigated for improving the quality of image editing, but it faces three key challenges: (1) limited reasoning exploration confined to denoising stochasticity, (2) biased reward fusion, and (3) unstable VLM-based instruction rewards. In this work, we propose ThinkRL-Edit, a reasoning-centric RL framework that decouples visual reasoning from image synthesis and expands reasoning exploration beyond denoising. To the end, we introduce Chain-of-Thought (CoT)-based reasoning sampling with planning and reflection stages prior to generation in online sampling, compelling the model to explore multiple semantic hypotheses and validate their plausibility before committing to a visual outcome. To avoid the failures of weighted aggregation, we propose an unbiased chain preference grouping strategy across multiple reward dimensions. Moreover, we replace interval-based VLM scores with a binary checklist, yielding more precise, lower-variance, and interpretable rewards for complex reasoning. Experiments show our method significantly outperforms prior work on reasoning-centric image editing, producing instruction-faithful, visually coherent, and semantically grounded edits.
title ThinkRL-Edit: Thinking in Reinforcement Learning for Reasoning-Centric Image Editing
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2601.03467