用强化学习让视觉模型学会深度推理,避免答对题却不会思考的陷阱。
Visionary-R1: Mitigating Shortcuts in Visual Reasoning with Reinforcement Learning
- 先生成图像描述再推理,强制模型理解画面而非找答题捷径。
- 在27.3万组无思维链数据上训练,性能超越GPT-4o等主流模型。
- 适合想提升视觉模型深层推理能力的研究者和开发者。
让通用推理能力成为AI的核心挑战长期存在。近期大语言模型(如DeepSeek-R1)研究显示,通过强化学习(如GRPO)结合简单问答对,可使预训练模型发展出推理能力。本文旨在通过强化学习与视觉问答对,训练视觉语言模型(VLMs)在图像数据上进行推理,且无需显式的思维链(CoT)标注。研究发现,仅在视觉模型上应用强化学习(即提示模型先生成推理链再作答),会导致模型依赖简单问题中的捷径,从而降低在未见数据分布上的泛化能力。我们认为,抑制捷径学习的关键在于促使模型在推理前先理解图像。因此,我们设计了“描述-推理-作答”的输出格式:先生成图像的详细描述,再构建完整推理链。在27.3万组无思维链的视觉问答对上仅使用强化学习训练后,所提出的Visionary-R1模型,在多个视觉推理基准上表现优于GPT-4o、Claude3.5-Sonnet及Gemini-1.5-Pro等强大多模态模型。
原文摘要 · Abstract (English)
Learning general-purpose reasoning capabilities has long been a challenging problem in AI. Recent research in large language models (LLMs), such as DeepSeek-R1, has shown that reinforcement learning techniques like GRPO can enable pre-trained LLMs to develop reasoning capabilities using simple question-answer pairs. In this paper, we aim to train visual language models (VLMs) to perform reasoning on image data through reinforcement learning and visual question-answer pairs, without any explicit chain-of-thought (CoT) supervision. Our findings indicate that simply applying reinforcement learning to a VLM -- by prompting the model to produce a reasoning chain before providing an answer -- can lead the model to develop shortcuts from easy questions, thereby reducing its ability to generalize across unseen data distributions. We argue that the key to mitigating shortcut learning is to encourage the model to interpret images prior to reasoning. Therefore, we train the model to adhere to a caption-reason-answer output format: initially generating a detailed caption for an image, followed by constructing an extensive reasoning chain. When trained on 273K CoT-free visual question-answer pairs and using only reinforcement learning, our model, named Visionary-R1, outperforms strong multimodal models, such as GPT-4o, Claude3.5-Sonnet, and Gemini-1.5-Pro, on multiple visual reasoning benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。