Unlocking Multimodal Mathematical Reasoning via Process Reward Model

Yifan Wang (Tsinghua University, Tsinghua University) · Lei Wang (Nankai University) · Yujiu Yang (Tsinghua University) · Yiyao Yu (Tsinghua University) · Ruihang Chu (Wan, Alibaba Group) · Ruilin Luo (Tsinghua University, Tsinghua University) · Zhuofan Zheng (ByteDance Inc.) · Xinzhe Ni (Tsinghua University) · Zicheng Lin (Tsinghua University, Tsinghua University) · Songtao Jiang (Zhejiang University) · Chufan Shi (University of Southern California) · Jin zeng (University of the Chinese Academy of Sciences)
chain-of-thought reasoningdualmath-1.1mmmathcot-1mmultimodal large language modelsmultimodal prm-aided online rlmultimodal reasoningprocess labelingprocess reward modelsprocess supervision dataprocess-supervised group-relative-policy-optimizationreinforcement learningreward hackingtest-time scalingunfolding multimodal process-supervision aided training

Process Reward Models (PRMs) have shown promise in enhancing the mathematical reasoning capabilities of Large Language Models (LLMs) through Test-Time Scaling (TTS). However, their integration into multimodal reasoning remains largely unexplored. In this work, we take the first step toward unlocking the potential of PRMs in multimodal mathematical reasoning. We identify three key challenges: (i) the scarcity of high-quality reasoning data constrains the capabilities of foundation Multimodal Large Language Models (MLLMs), which imposes further limitations on the upper bounds of TTS and reinforcement learning (RL); (ii) a lack of automated methods for process labeling within multimodal contexts persists; (iii) the employment of process rewards in unimodal RL faces issues like reward hacking, which may extend to multimodal scenarios. To address these issues, we introduce URSA, a three-stage Unfolding multimodal pRocess-Supervision Aided training framework. We first construct MMathCoT-1M, a high-quality large-scale multimodal Chain-of-Thought (CoT) reasoning dataset, to build a stronger math reasoning foundation MLLM, URSA-8B. Subsequently, we go through an automatic process to synthesize process supervision data, which emphasizes both logical correctness and perceptual consistency. We introduce DualMath-1.1M to facilitate the training of URSA-8B-RM. Finally, we propose Process-Supervised Group-Relative-Policy-Optimization (PS-GRPO), pioneering a multimodal PRM-aided online RL method that outperforms vanilla GRPO. With PS-GRPO application, URSA-8B-PS-GRPO outperforms Gemma3-12B and GPT-4o by 8.4% and 2.7% on average across 6 benchmarks.