智源社区 09月20日 23:29
多模态推理新框架:DreamPRM-1.5优化样本权重提升模型性能
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

本文介绍了DreamPRM-1.5,一个用于多模态推理的新训练框架。该框架通过实例级重加权(instance reweighting)策略,动态调整训练样本的权重,从而有效利用高质量样本并抑制噪声样本的影响。DreamPRM-1.5采用双层优化框架,将样本权重作为可学习参数,并利用生成式奖励模型对推理过程进行评分。研究人员设计了Instance Table和Instance Net两种互补方案来实现样本级加权,并在一系列实验中取得了显著成果,尤其是在MMMU基准上超越了多个顶尖模型,为未来多模态推理研究提供了新的方向。

💡 **实例级重加权框架**:DreamPRM-1.5 引入了实例级重加权(instance reweighting)技术,将样本权重细化到单个训练样本,使模型能够动态调整不同数据样本在训练中的影响力。高质量样本获得更高权重,而低质量或噪声样本的权重则被降低,从而更有效地利用数据资源。

⚖️ **双层优化与生成式奖励模型**:该框架采用双层优化(bi-level optimization)框架,其中上层优化基于元数据集评估推理表现并动态更新样本权重。此外,它还利用生成式奖励模型(Generative Reward Model)对推理过程中的每一步进行评分,为实例重加权提供更精细化的信号,确保权重学习由推理效果驱动。

⚙️ **Instance Table与Instance Net两种实现**:为了实现样本级加权,研究人员设计了两种互补方案:Instance Table为每个样本分配独立权重参数,灵活度高,适合小规模数据集;Instance Net则使用小型MLP网络预测权重,参数量固定,更适合大规模训练并具有更强的泛化能力。

🚀 **MMMU基准上的SOTA表现**:在最具挑战性的多模态推理基准MMMU上,DreamPRM-1.5取得了显著的性能提升,其Instance Table和Instance Net版本分别达到了84.6%和83.6%的准确率,超越了包括GPT-5-mini w/ thinking、VisualPRM、Self-consistency以及GPT-5和Gemini 2.5 Pro Deep-Think等在内的多种强基线和闭源模型。

编辑:LRST

一图看透全球大模型!新智元十周年钜献,2025 ASI前沿趋势报告37页首发

近年来,大语言模型(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 的训练流程采用 双层优化框架

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

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


生成式奖励模型
面向推理过程的打分机制


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

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

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

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

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


实验设计与实现细节


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

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

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

    训练流程

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

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

    计算资源:单卡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


内容中包含的图片若涉及版权问题,请及时与我们联系删除

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

多模态推理 大模型 DreamPRM-1.5 样本加权 双层优化 生成式奖励模型 MMMU AI Multimodal Reasoning Large Language Models Instance Reweighting Bi-level Optimization Generative Reward Model
相关文章