arXiv:2504.01805cs.CV2025-04被引 134

用强化学习提升大模型视频空间推理能力,效果超越GPT-4o

SpaceR: Reinforcing MLLMs in Video Spatial Reasoning

  • 基于可验证奖励的强化学习框架,引入地图想象机制引导推理
  • 构建15.1万样本数据集,涵盖9.1万个可验证空间推理问题
  • 在多个基准上超越GPT-4o,接近顶级闭源模型表现

视频空间推理需从帧中推断隐含空间结构,现有多模态大模型(MLLMs)面临两大瓶颈:缺乏高质量数据集与有效训练策略。受可验证奖励强化学习(RLVR)提升语言模型推理能力的启发,本文提出SpaceR框架。首先构建SpaceR-151k数据集,包含91,000个覆盖多样空间场景的可验证问题,以及60,000个用于保持多模态理解的样本。其次提出空间引导式强化学习(SG-RLVR),在组相对策略优化(GRPO)基础上引入地图想象机制,促使模型在思考过程中推断空间布局,增强推理能力。大量实验表明,SpaceR在空间推理基准(如VSI-Bench、STI-Bench、SPAR-Bench)上达到顶尖水平,同时在视频理解基准(Video-MME、TempCompass、LongVideoBench)上保持竞争力。尤为突出的是,其在VSI-Bench上比GPT-4o高11.6%准确率,与领先闭源模型Gemini-2.0-Flash相当,验证了数据集与方法的有效性。代码、模型与数据集已开源。

原文摘要 · Abstract (English)

Video spatial reasoning, which involves inferring the underlying spatial structure from observed video frames, poses a significant challenge for existing Multimodal Large Language Models (MLLMs). This limitation stems primarily from 1) the absence of high-quality datasets for this task, and 2) the lack of effective training strategies to develop spatial reasoning capabilities. Motivated by the success of Reinforcement Learning with Verifiable Reward (RLVR) in unlocking LLM reasoning abilities, this work aims to improve MLLMs in video spatial reasoning through the RLVR paradigm. To this end, we introduce the $\textbf{SpaceR}$ framework. First, we present $\textbf{SpaceR-151k}$, a dataset with 91k questions spanning diverse spatial reasoning scenarios with verifiable answers, and 60k samples for maintaining general multimodal understanding. Second, we propose $\textbf{Spatially-Guided RLVR (SG-RLVR)}$, a novel reinforcement learning approach that extends Group Relative Policy Optimization (GRPO) with a novel map imagination mechanism, which encourages the model to infer spatial layouts in the thinking process, thereby facilitating more effective spatial reasoning. Extensive experiments demonstrate that SpaceR achieves state-of-the-art performance on spatial reasoning benchmarks (e.g., VSI-Bench, STI-Bench, and SPAR-Bench), while maintaining competitive results on video understanding benchmarks (e.g., Video-MME, TempCompass, and LongVideoBench). Remarkably, SpaceR surpasses the advanced GPT-4o by 11.6\% accuracy on VSI-Bench and is on par with the leading proprietary model Gemini-2.0-Flash, highlighting the effectiveness of our SpaceR-151k dataset and SG-RLVR in reinforcing spatial reasoning ability of MLLMs. Code, model, and dataset are available at https://github.com/OuyangKun10/SpaceR.

视频推理强化学习多模态空间理解

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。