用R1-Zero思路训练,让小模型视频空间推理能力大幅跃升。
Improved Visual-Spatial Reasoning via R1-Zero-Like Training
- 基于GRPO和精心构建的VSI-100k数据集,激活小模型视觉空间推理能力。
- 2B模型仅用120小时训练,性能比基线高12.1%,超越GPT-4o。
- 适合想提升多模态模型空间推理能力的研究者与开发者。
近年来,提升多模态大语言模型(MLLMs)的推理能力受到广泛关注。作为在物理世界中运行的AI智能体的核心能力,基于视频的视觉空间智能(VSI)成为MLLMs最关键的能力之一。本文首次深入研究通过类R1-Zero的训练方式提升MLLMs的视觉空间推理能力。技术上,我们发现小到中等规模的Qwen2-VL模型无法通过思维链(CoT)提示激活其视觉空间推理能力。随后,我们采用GRPO训练方法,基于精心构建的VSI-100k数据集,参照DeepSeek-R1-Zero范式进行优化。研究中发现,即使使用较小值,保持KL惩罚项仍是必要的。仅用120 GPU小时,从Qwen2-VL-2B微调的vsGRPO-2B模型性能比基线提升12.1%,并超过GPT-4o;而从Qwen2-VL-7B微调的vsGRPO-7B模型性能接近最佳开源模型LLaVA-NeXT-Video-72B。此外,与监督微调和直接偏好优化基线相比,vsGRPO展现出显著优势。代码与数据集即将开源。
原文摘要 · Abstract (English)
Increasing attention has been placed on improving the reasoning capacities of multi-modal large language models (MLLMs). As the cornerstone for AI agents that function in the physical realm, video-based visual-spatial intelligence (VSI) emerges as one of the most pivotal reasoning capabilities of MLLMs. This work conducts a first, in-depth study on improving the visual-spatial reasoning of MLLMs via R1-Zero-like training. Technically, we first identify that the visual-spatial reasoning capacities of small- to medium-sized Qwen2-VL models cannot be activated via Chain of Thought (CoT) prompts. We then incorporate GRPO training for improved visual-spatial reasoning, using the carefully curated VSI-100k dataset, following DeepSeek-R1-Zero. During the investigation, we identify the necessity to keep the KL penalty (even with a small value) in GRPO. With just 120 GPU hours, our vsGRPO-2B model, fine-tuned from Qwen2-VL-2B, can outperform the base model by 12.1% and surpass GPT-4o. Moreover, our vsGRPO-7B model, fine-tuned from Qwen2-VL-7B, achieves performance comparable to that of the best open-source model LLaVA-NeXT-Video-72B. Additionally, we compare vsGRPO to supervised fine-tuning and direct preference optimization baselines and observe strong performance superiority. The code and dataset will be available soon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。