arXiv:2505.20241cs.LGcs.AI2025-05NeurIPS被引 16

通过动态重加权数据域,提升多模态推理模型的泛化能力。

DreamPRM: Domain-Reweighted Process Reward Model for Multimodal Reasoning

  • 采用双层优化框架,按数据质量动态调整训练权重。
  • 在多个多模态推理基准上显著提升SOTA模型表现。
  • 适合需要高泛化能力的多模态推理系统研发者。

推理显著提升了大语言模型在复杂任务上的性能。当前研究的核心是过程奖励模型(PRM),其对中间推理步骤进行细粒度评估并引导推理过程。然而,将PRM扩展到多模态大语言模型(MLLMs)面临挑战:多模态推理涵盖的任务范围更广,导致训练与测试集间的分布偏移更严重,泛化难度更大。因此,训练可靠的多模态PRM需要大规模且多样化的数据集以保证覆盖性。但现有多模态推理数据集存在明显的质量不平衡,损害了PRM性能,凸显了有效数据选择策略的必要性。为此,我们提出DreamPRM,一种用于多模态PRM的域重加权训练框架,采用双层优化。下层优化中,DreamPRM在多个数据集上进行微调,并分配域权重,使PRM优先关注高质量推理信号,缓解数据集质量不平衡的影响。上层优化中,使用独立的元学习数据集评估PRM性能;通过聚合损失函数反馈更新域权重,从而提升训练后PRM的泛化能力。在涵盖数学与通用推理的多个多模态推理基准上的大量实验表明,结合DreamPRM的测试时缩放能持续提升先进MLLM的表现。进一步对比显示,DreamPRM的域重加权策略优于其他数据选择方法,并带来比现有测试时缩放方法更高的准确率提升。

原文摘要 · Abstract (English)

Reasoning has substantially improved the performance of large language models (LLMs) on complicated tasks. Central to the current reasoning studies, Process Reward Models (PRMs) offer a fine-grained evaluation of intermediate reasoning steps and guide the reasoning process. However, extending PRMs to multimodal large language models (MLLMs) introduces challenges. Since multimodal reasoning covers a wider range of tasks compared to text-only scenarios, the resulting distribution shift from the training to testing sets is more severe, leading to greater generalization difficulty. Training a reliable multimodal PRM, therefore, demands large and diverse datasets to ensure sufficient coverage. However, current multimodal reasoning datasets suffer from a marked quality imbalance, which degrades PRM performance and highlights the need for an effective data selection strategy. To address the issues, we introduce DreamPRM, a domain-reweighted training framework for multimodal PRMs which employs bi-level optimization. In the lower-level optimization, DreamPRM performs fine-tuning on multiple datasets with domain weights, allowing the PRM to prioritize high-quality reasoning signals and alleviating the impact of dataset quality imbalance. In the upper-level optimization, the PRM is evaluated on a separate meta-learning dataset; this feedback updates the domain weights through an aggregation loss function, thereby improving the generalization capability of trained PRM. Extensive experiments on multiple multimodal reasoning benchmarks covering both mathematical and general reasoning show that test-time scaling with DreamPRM consistently improves the performance of state-of-the-art MLLMs. Further comparisons reveal that DreamPRM's domain-reweighting strategy surpasses other data selection methods and yields higher accuracy gains than existing test-time scaling approaches.

多模态推理奖励模型数据加权泛化能力

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。