Advancing Multimodal Reasoning via Reinforcement Learning with Cold Start

Lai Wei, Yuting Li, Kaipeng Zheng, Chen Wang, Yue Wang +3 more
5/28/2025
cs.CLcs.AIcs.CV

Official code: waltonfuture/RL-with-Cold-Start (50 stars, last commit 2025-06-27). 1 more implementation below.

Abstract

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.

View on arXivView PDF

Code Implementations(2)

SFT+RL boosts multimodal reasoning

501Shell, PythonMay 25, 20251 years ago
grpomllmsft
hiyouga/EasyR1Community100%
Apache-2.0

EasyR1: An Efficient, Scalable, Multi-Modality RL Training Framework based on veRL

5,037372Python, MakefileFeb 22, 20256 months agoApache-2.0
aideepseekgptllmnlp+3 more

Similar Papers

Cite this paper

@article{wei2025advancing,
  title  = {Advancing Multimodal Reasoning via Reinforcement Learning with Cold Start},
  author = {Lai Wei and Yuting Li and Kaipeng Zheng and Chen Wang and Yue Wang and Linghe Kong and Lichao Sun and Weiran Huang},
  year   = {2025},
  eprint = {2505.22334},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2505.22334v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.22334.svg)](https://codeofpaper.com/paper/2505.22334)

Discussion