登顶多模态推理榜MMMU!UCSD新方法超越GPT-5、Gemini

  新智元报道

  编辑:LRST

  DreamPRM 由加州大学圣地亚哥分校的研究团队开发,在数学推理权威测评榜 MMMU 上获得了第一名。

  近年来,大语言模型(LLM)在推理能力上的进展显著,其中过程奖励模型(Process Reward Model, PRM)的提出,使得模型能够在推理链条的中间步骤获得监督,从而更稳健地选择合理的解题路径。

  这类方法在文本推理任务中已经取得了良好效果,但在扩展至多模态场景时,仍然面临两个突出挑战:

  分布偏移:多模态输入空间巨大,训练与推理分布往往存在显著差异;

  数据质量不均:大规模训练集不可避免地包含噪声或低质量样本,降低了有效监督信号。

  因此,如何在多模态推理中有效利用高质量样本,抑制噪声样本的负面影响,成为亟需解决的问题。

  针对于此,研究人员设计了新的训练框架,通过双层优化框架,将数据样本的权重(Instance Weights)作为可学习参数,动态改变数据样本的在训练中的影响。

  论文地址:https://arxiv.org/abs/2509.05542

  代码地址:https://github.com/coder-qicao/DreamPRM-1.5

  MMMU Leaderboard

  论文第一作者为博士生 Qi Cao,通讯作者为该校副教授 Pengtao Xie。

  从 DreamPRM 到 DreamPRM-1.5

  从「领域加权」到「样本加权」

  此前,研究人员提出了DreamPRM框架,通过领域级重加权(domain reweighting)的方式,在不同数据子集之间分配权重,从而提升训练效果。

  在此基础上,DreamPRM-1.5将加权粒度进一步细化到单个训练样本

  • 高质量样本获得更大权重;

  • 低质量或噪声样本权重降低。

  这种实例级重加权(instance reweighting)策略,使模型能够充分挖掘每条数据的潜在价值。

  两种方法:Instance Table 和 Instance Net

  DreamPRM1.5 的两种模型架构

  为了实现「样本级加权」,研究人员设计了两种互补方案:

  Instance Table

  给每个训练样本一个独立的权重参数;

  灵活度高,尤其适合小规模数据集;

  缺点是参数量和样本数挂钩,数据一大就很难撑住。

  Instance Net

  不直接存表,而是用一个小型 MLP 网络来预测每条数据的权重;

  参数量固定,不受数据规模限制;

  更适合大规模训练,泛化能力更强。

  这就像两种「学习笔记」方式:Instance Table 像是给每道题都写一条批注;Instance Net 则像是总结出一套「看题给分」的规则。

  方法核心

  双层优化(Bi-level Optimization)

  DreamPRM-1.5 的训练流程采用双层优化框架

  1. 下层优化利用样本权重对 PRM 进行更新:

  1. 上层优化在元数据集上评估推理表现,并基于反馈动态更新样本权重:

  这种设计确保了权重的学习不是静态设定,而是由推理效果驱动、动态调整的,从而增强了模型在复杂任务中的适应性。

  生成式奖励模型

  面向推理过程的打分机制

  在 DreamPRM-1.5 中,研究人员采用了生成式奖励模型(Generative Reward Model)来对推理过程中的每一步进行评分。其核心思想是:

  • 评分方式:模型在每一步输出「+」或「-」,分别表示该步推理是否合理;

  • 打分机制:通过 softmax 计算 「+」 的概率,将其作为该步骤的置信度;

  • 聚合策略:对整条推理链的步骤分数进行聚合(平均),再与标准答案进行对比,用于指导样本权重的更新。

  这一设计的优点在于,它不仅能逐步评估推理链条的合理性,还能为实例重加权提供更细粒度的信号。

  实验设计与实现细节

  1. 模型基座:采用 InternVL3-1B 作为 PRM 的基础模型,并在推理阶段基于 GPT-5-mini 进行测试。设计了生成式奖励模型的

  2. 训练数据:从 VisualPRM-400k 中采样不同规模的数据(12k、100k)分别训练 Instance Table 与 Instance Net

  3. 元数据集:使用 MMMU-Pro 的标准分割(仅使用 test set 数据,以避免与 validation set 出现重合),生成候选推理链作为 meta set,用于权重更新。

  4. 训练流程

  冷启动先进行一次有监督微调(20k 样本),使模型能够稳定输出「+/-」标记;

  双层优化在此基础上进行 100k 步迭代,采用 AdamW 优化器与余弦学习率调度。

  1. 计算资源:单卡 NVIDIA A100,训练约 72 小时完成

  实验结果

  在 MMMU 基准上的表现

  研究人员在MMMU(Massive Multi-discipline Multimodal Understanding)基准上对方法进行了系统评测。

  该基准涵盖 30 个学科、183 个子领域,题型覆盖图表、地图、化学结构等多模态输入,是目前最具挑战性的推理测试之一。

  主要结果

  • GPT-5-mini w/ thinking(基线):80.0%

  • DreamPRM-1.5(Instance Table):84.6% (+4.6)

  • DreamPRM-1.5(Instance Net):83.6% (+3.6)

  对比分析

  • No Selection:使用相同数据但不做重加权,仅有 79.1%,验证了实例加权的重要性;

  • VisualPRM:尽管使用完整的 400k 数据集,但仅达到 80.5%,说明数据规模并不能完全弥补质量差异;

  • Self-consistency:经典的 test-time scaling 方法为 81.4%,依然低于 DreamPRM-1.5。

  整体来看,DreamPRM-1.5 不仅显著超越了基于 GPT-5-mini 的多种强基线,还在精度上超过了GPT-5(84.2%)和Gemini 2.5 Pro Deep-Think(84.0%)等顶级闭源模型。

  结论与展望

  DreamPRM-1.5 将实例级重加权引入多模态推理训练中,通过双层优化动态调整样本权重,使模型能够更好地识别和利用高质量数据。

  主要贡献体现在:

  • 提出实例级重加权框架,突破了仅在领域级别加权的限制;

  • 设计了Instance Table 与 Instance Net两种互补实现,兼顾小规模与大规模训练场景;

  • MMMU 基准上取得新的 SOTA 结果,超过多个闭源大模型。

  这一结果表明,在未来的推理模型研究中,数据质量的精细利用方式也是值得关注的重要方面。

  更智能的样本加权与过程评分方法,有望成为推动多模态推理进一步发展的关键方向。

  参考资料:

  https://arxiv.org/abs/2505.20241v2