Enhancing Spatial Understanding in Image Generation via Reward Modeling
Recent progress in text-to-image generation has greatly advanced visual fidelity and creativity, but it has also imposed higher demands on prompt complexity-particularly in encoding intricate spatial relationships. In such cases, achieving satisfactory results often requires multiple sampling attempts. To address this challenge, we introduce a novel method that strengthens the spatial understanding of current image generation models. We first construct the SpatialReward-Dataset with over 80k preference pairs. Building on this dataset, we build SpatialScore, a reward model designed to evaluate the accuracy of spatial relationships in text-to-image generation, achieving performance that even surpasses leading proprietary models on spatial evaluation. We further demonstrate that this reward model effectively enables online reinforcement learning for the complex spatial generation. Extensive experiments across multiple benchmarks show that our specialized reward model yields significant and consistent gains in spatial understanding for image generation.
69a4feb9bc2d66e68888f4fc
Zhenyu Tang
69a4feb9bc2d66e68888f4fd
Chaoran Feng
claimed_verified
2026-04-15T12:09:45.841Z
65250cda87ad4a39f84d482d
https://cdn-avatars.huggingface.co/v1/production/uploads/65250cda87ad4a39f84d482d/ryzc7hx6x09DbL4eiEBip.jpeg
Feng Chaoran
Falcary
user
Falcary
69a4feb9bc2d66e68888f4fe
Yufan Deng
claimed_verified
2026-04-13T07:38:24.246Z
68fce03ed1d0efce7ca87075
https://cdn-avatars.huggingface.co/v1/production/uploads/68fce03ed1d0efce7ca87075/GRKTeVIaLZD_M-KoJE8YF.png
yfdeng
yfdeng10
user
yfdeng10
69a4feb9bc2d66e68888f4ff
Jie Wu
69a4feb9bc2d66e68888f500
Xiaojie Li
69a4feb9bc2d66e68888f501
Rui Wang
69a4feb9bc2d66e68888f502
Yunpeng Chen
69a4feb9bc2d66e68888f503
Daquan Zhou
Enhancing Spatial Understanding in Image Generation via Reward Modeling
Recent progress in text-to-image generation has greatly advanced visual fidelity and creativity, but it has also imposed higher demands on prompt complexity-particularly in encoding intricate spatial relationships. In such cases, achieving satisfactory results often requires multiple sampling attempts. To address this challenge, we introduce a novel method that strengthens the spatial understanding of current image generation models. We first construct the SpatialReward-Dataset with over 80k preference pairs. Building on this dataset, we build SpatialScore, a reward model designed to evaluate the accuracy of spatial relationships in text-to-image generation, achieving performance that even surpasses leading proprietary models on spatial evaluation. We further demonstrate that this reward model effectively enables online reinforcement learning for the complex spatial generation. Extensive experiments across multiple benchmarks show that our specialized reward model yields significant and consistent gains in spatial understanding for image generation.
A new reward model called SpatialScore is introduced to improve spatial relationship understanding in text-to-image generation through reinforcement learning with a large-scale dataset of preference pairs.
SpatialReward-Dataset SpatialScore reward model text-to-image generation reinforcement learning spatial relationships preference pairs