Score: 2

CogniEdit: Dense Gradient Flow Optimization for Fine-Grained Image Editing

Published: December 15, 2025 | arXiv ID: 2512.13276v1

By: Yan Li , Lin Liu , Xiaopeng Zhang and more

Potential Business Impact:

Makes editing pictures follow exact instructions better.

Business Areas:
Image Recognition Data and Analytics, Software

Instruction-based image editing with diffusion models has achieved impressive results, yet existing methods struggle with fine-grained instructions specifying precise attributes such as colors, positions, and quantities. While recent approaches employ Group Relative Policy Optimization (GRPO) for alignment, they optimize only at individual sampling steps, providing sparse feedback that limits trajectory-level control. We propose a unified framework CogniEdit, combining multi-modal reasoning with dense reward optimization that propagates gradients across consecutive denoising steps, enabling trajectory-level gradient flow through the sampling process. Our method comprises three components: (1) Multi-modal Large Language Models for decomposing complex instructions into actionable directives, (2) Dynamic Token Focus Relocation that adaptively emphasizes fine-grained attributes, and (3) Dense GRPO-based optimization that propagates gradients across consecutive steps for trajectory-level supervision. Extensive experiments on benchmark datasets demonstrate that our CogniEdit achieves state-of-the-art performance in balancing fine-grained instruction following with visual quality and editability preservation

Country of Origin
🇭🇰 Hong Kong

Repos / Data Links

Page Count
20 pages

Category
Computer Science:
CV and Pattern Recognition