Rethinking Token-Level Policy Optimization for Multimodal Chain-of-Thought
On the map: LLM reasoning → Multimodal reasoning
Abstract (arXiv)
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
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Placed in Multimodal reasoning, one of 17 directions in LLM reasoning. Builds on 76 of its 85 references that are in the map. First posted on arXiv 24 Mar 2026.
Builds on
76 of 85 references.
Closest ancestors · 42
Same direction, newest first
- 2025Spotlight on Token Perception for Multimodal Reinforcement Learning10 Oct
- 2025Look Again, Think Slowly: Enhancing Visual Reflection in Vision-Language Models15 Sep
- 2025InternVL3.5: Advancing Open-Source Multimodal Models in Versatility, Reasoning, and Efficiency25 Aug
- 2025We-Math 2.0: A Versatile MathBook System for Incentivizing Visual Mathematical Reasoning14 Aug
- 2025DocThinker: Explainable Multimodal Large Language Models with Rule-based Reinforcement Learning for Document Understanding12 Aug
- 2025Shuffle-R1: Efficient RL framework for Multimodal Large Language Models via Data-centric Dynamic Shuffle7 Aug
- 2025VL-Cogito: Progressive Curriculum Reinforcement Learning for Advanced Multimodal Reasoning30 Jul
- 2025Perception-Aware Policy Optimization for Multimodal Reasoning8 Jul
- 2025GRPO-CARE: Consistency-Aware Reinforcement Learning for Multimodal Reasoning19 Jun
- 2025ViCrit: A Verifiable Reinforcement Learning Proxy Task for Visual Perception in VLMs11 Jun
- 2025Rex-Thinker: Grounded Object Referring via Chain-of-Thought Reasoning4 Jun
- 2025SRPO: Enhancing Multimodal LLM Reasoning via Reflection-Aware Reinforcement Learning2 Jun
30 more
- 2025Grounded Reinforcement Learning for Visual Reasoning29 May
- 2025Point-RFT: Improving Multimodal Reasoning with Visually Grounded Reinforcement Finetuning26 May
- 2025VLM-R3: Region Recognition, Reasoning, and Refinement for Enhanced Multimodal Chain-of-Thought22 May
- 2025Pixel Reasoner: Incentivizing Pixel-Space Reasoning with Curiosity-Driven Reinforcement Learning21 May
- 2025GRIT: Teaching MLLMs to Think with Images21 May
- 2025VisualQuality-R1: Reasoning-Induced Image Quality Assessment via Reinforcement Learning to Rank20 May
- 2025DeepEyes: Incentivizing "Thinking with Images" via Reinforcement Learning20 May
- 2025UniVG-R1: Reasoning Guided Universal Visual Grounding with Reinforcement Learning20 May
- 2025G1: Bootstrapping Perception and Reasoning Abilities of Vision-Language Model via Reinforcement Learning19 May
- 2025OpenThinkIMG: Learning to Think with Images via Visual Tool Reinforcement Learning13 May
- 2025Relation-R1: Progressively Cognitive Chain-of-Thought Guided Reinforcement Learning for Unified Relation Comprehension20 Apr
- 2025NoisyRollout: Reinforcing Visual Reasoning with Data Augmentation17 Apr
- 2025VL-Rethinker: Incentivizing Self-Reflection of Vision-Language Models with Reinforcement Learning10 Apr
- 2025Perception-R1: Pioneering Perception Policy with Reinforcement Learning10 Apr
- 2025SoTA with Less: MCTS-Guided Sample Selection for Data-Efficient Visual Reasoning Self-Improvement10 Apr
- 2025VLM-R1: A Stable and Generalizable R1-style Large Vision-Language Model10 Apr
- 2025Think or Not Think: A Study of Explicit Thinking in Rule-Based Visual Reinforcement Fine-Tuning20 Mar
- 2025R1-Onevision: Advancing Generalized Multimodal Reasoning through Cross-Modal Formalization13 Mar
- 2025MM-Eureka: Exploring the Frontiers of Multimodal Reasoning with Rule-based Reinforcement Learning10 Mar
- 2025Visual-RFT: Visual Reinforcement Fine-Tuning3 Mar
- 2024LLaVA-CoT: Let Vision Language Models Reason Step-by-Step15 Nov
- 2024Improve Vision Language Model Chain-of-thought Reasoning21 Oct
- 2024LogicVista: Multimodal LLM Logical Reasoning Benchmark in Visual Contexts6 Jul
- 2024We-Math: Does Your Large Multimodal Model Achieve Human-like Mathematical Reasoning?1 Jul
- 2024PuzzleVQA: Diagnosing Multimodal Reasoning Challenges of Language Models with Abstract Visual Patterns20 Mar
- 2023Compositional Chain-of-Thought Prompting for Large Multimodal Models27 Nov
- 2023MathVista: Evaluating Mathematical Reasoning of Foundation Models in Visual Contexts3 Oct
- 2023LISA: Reasoning Segmentation via Large Language Model1 Aug
- 2023ViperGPT: Visual Inference via Python Execution for Reasoning14 Mar
- 2022Super-CLEVR: A Virtual Benchmark to Diagnose Domain Robustness in Visual Reasoning1 Dec
Other references · 34
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2025Qwen2.5-VL Technical Reportcited by 5,418
- 2024GPT-4o System Cardcited by 4,467
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
- 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,175
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2016Making the V in VQA Matter: Elevating the Role of Image Understanding in Visual Question Answeringcited by 2,498
22 more
- 2015VQA: Visual Question Answeringcited by 2,489
- 2025InternVL3: Exploring Advanced Training and Test-Time Recipes for Open-Source Multimodal Modelscited by 1,769
- 2013Fine-Grained Visual Classification of Aircraftcited by 1,701
- 2023Kosmos-2: Grounding Multimodal Large Language Models to the Worldcited by 1,075
- 2016Modeling Context in Referring Expressionscited by 1,054
- 2023Shikra: Unleashing Multimodal LLM's Referential Dialogue Magiccited by 891
- 2024MathVerse: Does Your Multi-modal LLM Truly See the Diagrams in Visual Math Problems?cited by 573
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
- 2024MMMU-Pro: A More Robust Multi-discipline Multimodal Understanding Benchmarkcited by 494
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 463
- 2021Inter-GPS: Interpretable Geometry Problem Solving with Formal Language and Symbolic Reasoningcited by 376
- 2024SWIFT:A Scalable lightWeight Infrastructure for Fine-Tuningcited by 345
- 2025Reinforcing Spatial Reasoning in Vision-Language Models with Interwoven Thinking and Visual Drawingcited by 130
- 2025SEED-GRPO: Semantic Entropy Enhanced GRPO for Uncertainty-Aware Policy Optimizationcited by 87
- 2025Visual Planning: Let's Think Only with Imagescited by 85
- 2025SophiaVL-R1: Reinforcing MLLMs Reasoning with Thinking Rewardcited by 45
- 2025Harnessing Uncertainty: Entropy-Modulated Policy Gradients for Long-Horizon LLM Agentscited by 43
- 2025ACTIVE-o3: Empowering MLLMs with Active Perception via Pure Reinforcement Learningcited by 41
- 2025One RL to See Them All: Visual Triple Unified Reinforcement Learningcited by 29
- 2025MoDoMoDo: Multi-Domain Data Mixtures for Multimodal LLM Reinforcement Learningcited by 29
- 2025M2-Reasoning: Empowering MLLMs with Unified General and Spatial Reasoningcited by 16
- 2025Entropy-guided sequence weighting for efficient exploration in RL-based LLM fine-tuningcited by 13
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026Visual-Advantage On-Policy Distillation for Vision-Language Modelscited by 22
- 2026PRPO: Perception-Reinforced Policy Optimization via Token-Level Dynamic Advantage Reshapingcited by 4
- 2026Reflection Anchors for Propagation-Aware Visual Retention in Long-Chain Multimodal Reasoningcited by 0
- 2026Entropy Is Not Enough: Unlocking Effective Reinforcement Learning for Visual Reasoning via Vision-Anchored Token Selectioncited by 0
- 2026Balancing Efficiency and Efficacy: Training-Free Attention-Guided Switching Between Explicit and Latent Thoughts for MLLMscited by 0
- 2026Anchoring What Matters: A Dual-Level Learning Framework for Visually-Grounded Multimodal Reasoningcited by 0
- 2026TTRSD: Test-Time Reinforcement Learning with Self-Distillation for Vision-Language Modelscited by 0
- 2026TTRSD: Test-Time Reinforcement Learning with Self-Distillation for Vision-Language Models27 Sep 2026
- 2026Anchoring What Matters: A Dual-Level Learning Framework for Visually-Grounded Multimodal Reasoning16 Sep 2026
- 2026Balancing Efficiency and Efficacy: Training-Free Attention-Guided Switching Between Explicit and Latent Thoughts for MLLMs4 Aug 2026
- 2026PRPO: Perception-Reinforced Policy Optimization via Token-Level Dynamic Advantage Reshaping7 Jun 2026
- 2026Entropy Is Not Enough: Unlocking Effective Reinforcement Learning for Visual Reasoning via Vision-Anchored Token Selection2 Jun 2026
- 2026Visual-Advantage On-Policy Distillation for Vision-Language Models21 May 2026
- 2026Reflection Anchors for Propagation-Aware Visual Retention in Long-Chain Multimodal Reasoning10 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Reinforcing Multimodal Reasoning via Token-Level Perception-Grounded Advantage Estimation
- 2026Bridging Perception and Reasoning: Token Reweighting for RLVR in Multimodal LLMs
- 2026Not All Tokens See Equally: Perception-Grounded Policy Optimization for Large Vision-Language Models
- 2026Visually-Guided Policy Optimization for Multimodal Reasoning
- 2025Unveiling Chain of Step Reasoning for Vision-Language Models with Fine-grained Rewards
- 2025VTPerception-R1: Enhancing Multimodal Reasoning via Explicit Visual and Textual Perceptual Grounding
- 2026See It, Say It, Sorted: An Iterative Training-Free Framework for Visually-Grounded Multimodal Reasoning in LVLMs
- 2026Multi-Branch Policy Optimization for Multimodal Large Language Models