用冷启动+强化学习提升多模态模型的推理能力
Advancing Multimodal Reasoning via Reinforcement Learning with Cold Start
- 先用结构化思维链监督微调,再用强化学习优化
- 7B模型在MathVista上从66.3%提升至73.4%
- 适合想构建高性能多模态推理模型的研究者
大型语言模型(LLMs)在链式思维推理方面取得显著进展,强化学习(RL)在其中起关键作用。我们首次发现,多模态大模型(MLLMs)在未经过强化学习训练前就存在“顿悟”模式——即通过反思实现自我修正,但这些模式并不必然提升推理表现。基于此,我们提出一种两阶段增强多模态推理的方法:(1) 通过监督微调(SFT)进行冷启动,引入结构化的链式思维模式;(2) 再通过基于GRPO的强化学习进一步优化。大量实验表明,该方法在多个挑战性多模态推理基准上均优于仅使用SFT或仅使用RL的方法。所获模型在3B与7B规模下均达到开源多模态模型的领先水平,7B模型在MathVista上从66.3%提升至73.4%,We-Math从62.9%提升至70.4%;3B模型性能可媲美多个7B模型。本工作为构建先进多模态推理模型提供了实用指导。代码已开源。
原文摘要 · Abstract (English)
Recent advancements in large language models (LLMs) have demonstrated impressive chain-of-thought reasoning capabilities, with reinforcement learning (RL) playing a crucial role in this progress. While "aha moment" patterns--where models exhibit self-correction through reflection--are often attributed to emergent properties from RL, we first demonstrate that these patterns exist in multimodal LLMs (MLLMs) prior to RL training but may not necessarily correlate with improved reasoning performance. Building on these insights, we present a comprehensive study on enhancing multimodal reasoning through a two-stage approach: (1) supervised fine-tuning (SFT) as a cold start with structured chain-of-thought reasoning patterns, followed by (2) reinforcement learning via GRPO to further refine these capabilities. Our extensive experiments show that this combined approach consistently outperforms both SFT-only and RL-only methods across challenging multimodal reasoning benchmarks. The resulting models achieve state-of-the-art performance among open-source MLLMs at both 3B and 7B scales, with our 7B model showing substantial improvements over base models (e.g., 66.3 %$\rightarrow$73.4 % on MathVista, 62.9 %$\rightarrow$70.4 % on We-Math) and our 3B model achieving performance competitive with several 7B models. Overall, this work provides practical guidance for building advanced multimodal reasoning models. Our code is available at https://github.com/waltonfuture/RL-with-Cold-Start.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。