arXiv:2505.23590cs.CVcs.AI2025-05被引 30

用拼图测试大模型视觉强化学习,发现训练后能精准解复杂谜题。

Jigsaw-R1: A Study of Rule-based Visual Reinforcement Learning with Jigsaw Puzzles

  • 用拼图任务构建规则化视觉强化学习框架,支持可调难度与精确评估。
  • 模型经微调后在简单拼图上准确率达近100%,并泛化至未见复杂配置。
  • 强化学习比监督微调更优,但初始微调可能阻碍后续强化学习优化。

将基于规则的强化学习(RL)应用于多模态大语言模型(MLLMs)面临独特挑战,尤其在感知密集型任务中。本文以拼图为结构化实验框架,系统研究规则化视觉强化学习。拼图具备天然真值、可调难度和复杂决策需求,适合作为研究载体。研究发现:首先,MLLMs在最简拼图上初始表现接近随机猜测,经微调后准确率趋近100%,并能泛化至未见过的复杂构型;其次,拼图训练可提升对其他视觉任务的泛化能力,效果取决于具体任务配置;第三,模型可在有无显式推理的情况下学习与泛化,开源模型常偏好直接作答,即使训练时要求分步推理,仍可能跳过思考过程;第四,复杂推理模式似为预存在而非涌现,其频率随训练和任务难度增加而上升;最后,结果表明强化学习的泛化效果优于监督微调(SFT),且初始SFT冷启动阶段可能抑制后续强化学习优化。尽管结论基于拼图任务,但仍为理解规则化视觉强化学习及其在多模态学习中的潜力提供了重要拼图。代码已公开于:https://github.com/zifuwanggg/Jigsaw-R1。

原文摘要 · Abstract (English)

The application of rule-based reinforcement learning (RL) to multimodal large language models (MLLMs) introduces unique challenges and potential deviations from findings in text-only domains, particularly for perception-heavy tasks. This paper provides a comprehensive study of rule-based visual RL, using jigsaw puzzles as a structured experimental framework. Jigsaw puzzles offer inherent ground truth, adjustable difficulty, and demand complex decision-making, making them ideal for this study. Our research reveals several key findings: \textit{Firstly,} we find that MLLMs, initially performing near to random guessing on the simplest jigsaw puzzles, achieve near-perfect accuracy and generalize to complex, unseen configurations through fine-tuning. \textit{Secondly,} training on jigsaw puzzles can induce generalization to other visual tasks, with effectiveness tied to specific task configurations. \textit{Thirdly,} MLLMs can learn and generalize with or without explicit reasoning, though open-source models often favor direct answering. Consequently, even when trained for step-by-step reasoning, they can ignore the thinking process in deriving the final answer. \textit{Fourthly,} we observe that complex reasoning patterns appear to be pre-existing rather than emergent, with their frequency increasing alongside training and task difficulty. \textit{Finally,} our results demonstrate that RL exhibits more effective generalization than Supervised Fine-Tuning (SFT), and an initial SFT cold start phase can hinder subsequent RL optimization. Although these observations are based on jigsaw puzzles and may vary across other visual tasks, this research contributes a valuable piece of jigsaw to the larger puzzle of collective understanding rule-based visual RL and its potential in multimodal learning. The code is available at: https://github.com/zifuwanggg/Jigsaw-R1

视觉强化学习多模态模型拼图任务泛化能力

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