Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Rethinking Token-Level Policy Optimization for Multimodal Chain-of-Thought

About

Multimodal Chain-of-Thought (CoT) reasoning requires large vision-language models to construct reasoning trajectories that interleave perceptual grounding with multi-step inference. However, existing Reinforcement Learning with Verifiable Rewards (RLVR) methods typically optimize reasoning at a coarse granularity, treating CoT uniformly without distinguishing their varying degrees of visual grounding. In this work, we conduct a token-level analysis of multimodal reasoning trajectories and show that successful reasoning is characterized by structured token dynamics reflecting both perceptual grounding and exploratory inference. Building upon this analysis, we propose Perception-Exploration Policy Optimization (PEPO), which derives a perception prior from hidden state similarity and integrates it with token entropy through a smooth gating mechanism to produce token-level advantages. PEPO integrates seamlessly with existing RLVR frameworks such as GRPO and DAPO, requiring neither additional supervision nor auxiliary branches. Extensive experiments across diverse multimodal benchmarks demonstrate consistent and robust improvements over strong RL baselines, spanning geometry reasoning, visual grounding, visual puzzle solving, and few-shot classification, while maintaining stable training dynamics. Code: https://github.com/xzxxntxdy/PEPO

Yunheng Li, Hangyi Kuang, Hengrui Zhang, Jiangxia Cao, Zhaojie Liu, Qibin Hou, Ming-Ming Cheng• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMathVerse--
183
Visual GroundingRefCOCO (testA)--
162
Visual GroundingRefCOCO (testB)--
159
Visual Mathematical ReasoningWeMath--
149
Logical reasoningLogicVista
Accuracy34.45
113
Multimodal Mathematical ReasoningMathVista mini
Accuracy0.5475
111
Logical reasoningLogicVista
Avg Pass@839.85
48
Multimodal Mathematical ReasoningMathVerse mini
Accuracy45.42
39
Visual ReasoningMMMU-Pro
Avg@827.47
29
Geometric ReasoningGeometry3K (test)
Accuracy31.28
22
Showing 10 of 19 rows

Other info

GitHub

Follow for update