Search papers, labs, and topics across Lattice.
This paper introduces SpatialFlow-GRPO, a novel training framework that enhances image editing by incorporating spatially fine-grained reward feedback, addressing the limitations of traditional whole-image reward systems. By converting region-aware rewards into optimization signals that align with latent positions, the framework enables more precise control over image quality across different spatial regions. Experimental results demonstrate that SpatialFlow-GRPO significantly outperforms existing Flow-GRPO methods on multiple evaluation benchmarks, showcasing its effectiveness in improving multi-region editing capabilities.
Fine-grained spatial feedback can dramatically enhance image editing quality, outperforming traditional methods that rely on whole-image rewards.
Recent online reinforcement learning has substantially improved image editing quality. However, existing Flow-GRPO-style methods usually rely on a single whole-image reward, which makes fine-grained editing optimization difficult. We observe that a key obstacle in image editing is this spatial uniformity assumption: a whole-image reward cannot distinguish how different spatial regions contribute to image quality. To address this issue, we propose SpatialFlow-GRPO, a training framework that introduces spatially fine-grained reward feedback. The framework converts region-aware rewards into semantic-region-level optimization signals and aligns region advantages with the corresponding latent positions during policy updates. We also train a region-aware reward model, SFReward, construct SFReward-14K with region-annotated editing samples, and introduce MultiEditBench to evaluate multi-region editing ability. On OmniGen2 and FLUX.2-klein-4B, SpatialFlow-GRPO outperforms Flow-GRPO on GEdit-Bench, ImgEdit-Bench, and MultiEditBench. The results show that SpatialFlow-GRPO converts local feedback into spatially aligned update signals and improves editing quality.
