用强化学习提升视觉推理的每一步决策,让模型更准更鲁棒。
PROPA: Toward Process-level Optimization in Visual Reasoning via Reinforcement Learning
- 融合MCTS与GRPO生成每步反馈,无需人工标注中间步骤。
- 在7个基准上比顶尖方法提升17%(域内)和21%(域外)。
- 适合需要高精度多步推理的场景,如复杂图像理解任务。
尽管取得进展,视觉语言模型在复杂视觉推理中仍面临多步依赖导致早期错误传播的问题。现有后训练范式受限:监督微调依赖昂贵的逐步标注,而基于可验证奖励的强化学习方法(如GRPO)仅提供稀疏的结果级反馈,难以实现稳定优化。我们提出PROPA(过程级推理优化框架),将蒙特卡洛树搜索(MCTS)与GRPO结合,生成密集的过程级奖励,在无需人工标注的情况下对每一步推理进行优化。为解决冷启动问题,PROPA交错执行GRPO与监督微调,使模型从成功和失败的推理轨迹中学习。进一步训练一个过程奖励模型(PRM),以指导推理时的搜索,确保测试时搜索与训练信号对齐。在七个基准和四个视觉语言模型骨干上,PROPA持续优于基于SFT和RLVR的基线。其在域内任务上最高提升17.0%,域外任务上提升21.0%,展现出强大的推理与泛化能力。代码已开源:https://github.com/YanbeiJiang/PROPA。
原文摘要 · Abstract (English)
Despite significant progress, Vision-Language Models (VLMs) still struggle with complex visual reasoning, where multi-step dependencies cause early errors to cascade through the reasoning chain. Existing post-training paradigms are limited: Supervised Fine-Tuning (SFT) relies on costly step-level annotations, while Reinforcement Learning with Verifiable Rewards (RLVR) methods like GRPO provide only sparse, outcome-level feedback, hindering stable optimization. We introduce PROPA (Process-level Reasoning Optimization with interleaved Policy Alignment), a novel framework that integrates Monte Carlo Tree Search (MCTS) with GRPO to generate dense, process-level rewards and optimize reasoning at each intermediate step without human annotations. To overcome the cold-start problem, PROPA interleaves GRPO updates with SFT, enabling the model to learn from both successful and failed reasoning trajectories. A Process Reward Model (PRM) is further trained to guide inference-time search, aligning the test-time search with the training signal. Across seven benchmarks and four VLM backbones, PROPA consistently outperforms both SFT- and RLVR-based baselines. It achieves up to 17.0% gains on in-domain tasks and 21.0% gains on out-of-domain tasks compared to existing state-of-the-art, establishing a strong reasoning and generalization capability for visual reasoning tasks. The code isavailable at: https://github.com/YanbeiJiang/PROPA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。