SpatialReward: Verifiable Spatial Reward Modeling for Fine-Grained Spatial Consistency in Text-to-Image Generation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhou, Sashuai, Zhou, Qiang, Ma, Junpeng, Cao, Yue, Hu, Ruofan, Zhang, Ziang, Yang, Xiaoda, Wang, Zhibin, Song, Jun, Yu, Cheng, Zheng, Bo, Zhao, Zhou
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911538401509376
author Zhou, Sashuai
Zhou, Qiang
Ma, Junpeng
Cao, Yue
Hu, Ruofan
Zhang, Ziang
Yang, Xiaoda
Wang, Zhibin
Song, Jun
Yu, Cheng
Zheng, Bo
Zhao, Zhou
author_facet Zhou, Sashuai
Zhou, Qiang
Ma, Junpeng
Cao, Yue
Hu, Ruofan
Zhang, Ziang
Yang, Xiaoda
Wang, Zhibin
Song, Jun
Yu, Cheng
Zheng, Bo
Zhao, Zhou
contents Recent advances in text-to-image (T2I) generation via reinforcement learning (RL) have benefited from reward models that assess semantic alignment and visual quality. However, most existing reward models pay limited attention to fine-grained spatial relationships, often producing images that appear plausible overall yet contain inaccuracies in object positioning. In this work, we present \textbf{SpatialReward}, a verifiable reward model explicitly designed to evaluate spatial layouts in generated images. SpatialReward adopts a multi-stage pipeline: a \emph{Prompt Decomposer} extracts entities, attributes, and spatial metadata from free-form prompts; expert detectors provide accurate visual grounding of object positions and attributes; and a vision-language model applies chain-of-thought reasoning over grounded observations to assess complex spatial relations that are challenging for rule-based methods. To more comprehensively evaluate spatial relationships in generated images, we introduce \textbf{SpatRelBench}, a benchmark covering object attributes, orientation, inter-object relations, and rendered text placement. Experiments on Stable Diffusion and FLUX show that incorporating SpatialReward into RL training consistently improves spatial consistency and overall generation quality, with results aligned more closely to human judgments. These findings indicate that verifiable reward models hold considerable potential for enabling more accurate and controllable optimization in text-to-image generation models.
format Preprint
id arxiv_https___arxiv_org_abs_2603_22228
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle SpatialReward: Verifiable Spatial Reward Modeling for Fine-Grained Spatial Consistency in Text-to-Image Generation
Zhou, Sashuai
Zhou, Qiang
Ma, Junpeng
Cao, Yue
Hu, Ruofan
Zhang, Ziang
Yang, Xiaoda
Wang, Zhibin
Song, Jun
Yu, Cheng
Zheng, Bo
Zhao, Zhou
Computer Vision and Pattern Recognition
Artificial Intelligence
Recent advances in text-to-image (T2I) generation via reinforcement learning (RL) have benefited from reward models that assess semantic alignment and visual quality. However, most existing reward models pay limited attention to fine-grained spatial relationships, often producing images that appear plausible overall yet contain inaccuracies in object positioning. In this work, we present \textbf{SpatialReward}, a verifiable reward model explicitly designed to evaluate spatial layouts in generated images. SpatialReward adopts a multi-stage pipeline: a \emph{Prompt Decomposer} extracts entities, attributes, and spatial metadata from free-form prompts; expert detectors provide accurate visual grounding of object positions and attributes; and a vision-language model applies chain-of-thought reasoning over grounded observations to assess complex spatial relations that are challenging for rule-based methods. To more comprehensively evaluate spatial relationships in generated images, we introduce \textbf{SpatRelBench}, a benchmark covering object attributes, orientation, inter-object relations, and rendered text placement. Experiments on Stable Diffusion and FLUX show that incorporating SpatialReward into RL training consistently improves spatial consistency and overall generation quality, with results aligned more closely to human judgments. These findings indicate that verifiable reward models hold considerable potential for enabling more accurate and controllable optimization in text-to-image generation models.
title SpatialReward: Verifiable Spatial Reward Modeling for Fine-Grained Spatial Consistency in Text-to-Image Generation
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2603.22228