让AI自己纠错:用生成式模型提升多模态数学推理能力
GM-PRM: A Generative Multimodal Process Reward Model for Multimodal Mathematical Reasoning
- 将奖励模型变为能主动修正错误的推理伙伴
- 在20K样本上训练,显著提升解题正确率和多样性
- 适合需要高精度数学推理的多模态AI研究者
多模态大语言模型在复杂多步数学推理中常因视觉感知或逻辑推导的小错误导致完全失败。现有过程奖励模型(PRM)仅能二元判断对错,无法纠正错误且缺乏解释力。为此,我们提出生成式多模态过程奖励模型(GM-PRM),将PRM从被动评判者转变为主动协作的推理伙伴。不同于传统标量评分,GM-PRM对每一步推理进行细粒度分析,评估其意图、视觉对齐性和逻辑合理性,并可生成首个错误步骤的修正版本。基于此,我们设计了改进的测试时推理策略——精炼版Best-of-N(Refined-BoN),利用生成修正引导策略模型走向更优推理路径,从而提升解题池的质量与多样性。实验表明,GM-PRM在多个多模态数学基准上达到领先水平,仅需20,000样本训练即显著提升策略模型性能,展现出卓越的数据效率。代码将在论文接受后公开。
原文摘要 · Abstract (English)
Multimodal Large Language Models (MLLMs) demonstrate remarkable capabilities but often struggle with complex, multi-step mathematical reasoning, where minor errors in visual perception or logical deduction can lead to complete failure. While Process Reward Models (PRMs) offer step-by-step supervision, existing multimodal PRMs are limited to being binary verifiers that can identify but not correct errors, offering little explanatory power. To address these deficiencies, we introduce the Generative Multimodal Process Reward Model (GM-PRM), a novel paradigm that transforms the PRM from a passive judge into an active reasoning collaborator. Instead of a simple scalar score, GM-PRM provides a fine-grained, interpretable analysis of each reasoning step, evaluating its step intent, visual alignment, and logical soundness. More critically, GM-PRM is trained to generate a corrected version of the first erroneous step it identifies. This unique corrective capability enables our new test-time inference strategy, Refined Best-of-N (Refined-BoN). This framework actively enhances solution quality by using the PRM's generated correction to guide the policy model toward a more promising reasoning trajectory, thereby improving the diversity and correctness of the solution pool. We demonstrate that GM-PRM achieves state-of-the-art results on multiple multimodal math benchmarks, significantly boosting policy model performance with remarkable data efficiency, requiring only a 20K-sample training dataset. Our code will be released upon acceptance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。