【免费下载链接】VisualPRM-8B
📊 本文带你揭秘VisualPRM-8B背后的训练数据:VisualPRM400K多模态过程监督数据集。它由一条全自动的VisualPRM400K 自动化数据管线构建——无需人工逐条标注,就能为"图片 + 分步解答"打上每一步的对错标签,最终让这个 8B 参数的多模态过程奖励模型(Process Reward Model,简称 PRM)在 Best-of-N 场景下显著提升各类多模态大模型的推理准确率。
一、30秒入门:什么是多模态"过程奖励模型"?
先看一个类比 🏫:
| 评分方式 | 做法 | 问题 |
|---|---|---|
| 结果监督(ORM) | 只看最终答案对不对 | 过程错了但蒙对答案,模型学不到"哪一步错" |
| 过程监督(PRM) | 给每一步推理单独打分 | 监督粒度更细,模型能真正学会"挑错" |
VisualPRM-8B就是后者:一个 8B 参数的多模态过程奖励模型。它输入一张图和若干候选的分步解答,输出每一步(以及整体)的质量分数,从而在Best-of-N(BoN)策略中帮主模型"从 N 个回答里挑出最好的那一个"。
几个值得记住的亮点:
- 🎯 支持3 类 MLLM、4 个参数规模,跨模型家族通用
- 📈 用于能力很强的 InternVL2.5-78B 时,在7 个多模态推理基准上平均提升5.9 分
- 🆕 2025/04/11 发布了新版本 VisualPRM-8B-v1.1,进一步增强了 InternVL3 在 BoN 下的推理表现
- ⚖️ 配套的 VisualProcessBench 基准提供了人工标注的逐步对错标签,专门衡量 PRM"找错步"的能力
二、VisualPRM400K:VisualPRM-8B 的数据底座
模型再强,也离不开数据。VisualPRM400K正是为训练多模态 PRM 而构建的过程监督数据集,规模约40 万条。根据 README.md 中的声明:
To facilitate the training of multimodal PRMs, we construct a multimodal process supervision dataset VisualPRM400K using an automated data pipeline.
每条数据长什么样?大致是四元组:
- 问题图片(几何图、图表、截图等多模态内容)
- 问题文本
- 分步解答(Step 1、Step 2……Final answer)
- 逐步正确性标签(每一步"对/错")
训练 VisualPRM-8B 时实际用到的数据集还包括MMPR-v1.1与VisualPRM400K,二者均在 README.md 的datasets字段中列出。
上图是仓库自带的示例题 examples/image.png。官方演示正是拿这道"求 ∠3"的几何题,让 VisualPRM-8B 同时给两份分步解答打分——其中一份在第 2 步就走了弯路,PRM 通过逐步评分把它识别出来。这就是"过程监督"最直观的体现:不比最终答案,比每一步的推理质量。
三、VisualPRM400K 自动化管线:四大阶段拆解 🛠️
人工标注"每一步对错"的成本极高,40 万条几乎不可能手工完成。VisualPRM400K 的答案是自动化管线,核心思想是"用模型自己验证模型"。整个管线可分为四个阶段:
阶段一:多模态推理题目采集
- 从公开的多模态推理数据集中采集"图像 + 问题 + 参考解答"
- 覆盖几何、图表理解、数学题等需要多步推理的任务类型
- 这一步决定了数据集的领域分布与难度分布
阶段二:解答生成与步骤切分(Step Alignment)
- 由强多模态大模型对每题生成分步解答
- 自动把长解答切分成结构化的推理步骤(Step 1 / Step 2 / ……)
- 步骤切分必须对齐图像证据——例如"图中 ∠1 是直角"这类步骤要能对应到图里的直角标记
阶段三:基于蒙特卡洛展开的逐步打标签(核心 ⭐)
这是整条管线最关键的"自动化标注"环节:
- 从解答的每一个中间步骤出发,向后多次采样展开(rollout),推导出最终答案
- 统计展开结果中最终答案正确的比例
- 若某步之后正确率骤降,说明该步是"引入错误的转折点"
- 据此把"第一个错步"标为 0、其余有效步骤标为 1,得到逐步标签
✅好处:完全无需人工,且标签质量与"最终答案是否正确"这一可验证信号强绑定 ✅效果:这正是 40 万条过程监督数据得以规模化的关键
阶段四:过滤、去重与质检
- 剔除格式混乱、步骤过短或图文不匹配的样本
- 对同一问题做多解去重,避免数据分布失衡
- 抽样人工核验管线产出的逐步标签,保证数据可靠性
💡 补充:除了训练数据,团队还人工标注了小规模基准VisualProcessBench,用"人来找错步"的标准检验"模型找错步"的能力,形成"自动数据 + 人工评测"的闭环。
四、数据如何变成 VisualPRM-8B?
有了 VisualPRM400K,训练过程相对直接:
- 基座模型:基于 InternVL2.5-8B 系微调(README 元信息中标注
InternVL2_5-8B与InternVL2_5-8B-MPO) - 奖励模型头:在 config.json 中可以看到
architectures为InternVLRewardModel,并定义了二分类标签LABEL_0/LABEL_1——正好对应"该步错误 / 该步正确" - 训练目标:学习逐步对错预测,推理时综合每步得分给出整体分数
- 推理用法:通过
model.select_best_response()接口,把 N 个候选答案 + 同一张图交给 PRM,按得分排序选出最优回答
(注:此图为上文已引用的同一示例图,此处不再重复插入,仅作说明。)
五、仓库关键文件速查 📂
想知道"数据→模型"的每个环节落在哪些文件上,看这张表:
| 文件 | 作用 |
|---|---|
| README.md | 项目介绍、数据声明(VisualPRM400K 自动化管线)、推理示例 |
| config.json | 模型配置:InternVLRewardModel架构、二分类标签、视觉/语言子模块参数 |
| modeling_internvl_chat.py | 核心模型实现InternVLRewardModel(含 BoN 打分逻辑) |
| modeling_intern_vit.py | InternViT 视觉编码器,负责理解问题图片 |
| modeling_internlm2.py | InternLM2 语言模型骨干(32 层、hidden_size 4096) |
| conversation.py | 对话/提示模板处理 |
| tokenizer.model + tokenizer_config.json | 分词器与特殊 token 配置 |
| examples/image.png | 官方演示用几何推理示例图 |
六、快速上手:获取模型与数据集 🚀
克隆仓库后即可在本地体验(模型权重与数据集均已在仓库/元信息中给出):
git clone https://gitcode.com/OpenGVLab/VisualPRM-8B使用流程非常简洁(详见 README.md 的 Inference 部分):
- 用
AutoModel加载 VisualPRM-8B(trust_remote_code=True) - 准备一张问题图 + N 个分步候选解答
- 调用
select_best_response(...),返回排序后的答案与得分 - 取最高分答案作为最终输出
七、总结:为什么这套数据管线值得学习?
- 🎯问题精准:多模态推理缺的不是"答案对不对",而是"哪一步错了"——VisualPRM400K 用过程监督直击痛点
- ⚙️全自动标注:蒙特卡洛展开 + 可验证的最终答案信号,让 40 万条逐步标签摆脱人工依赖
- 🔁数据-模型-评测闭环:VisualPRM400K 训练、VisualProcessBench 评测、VisualPRM-8B 落地 BoN,三位一体
- 📦开箱即用:仓库内模型代码、配置、示例一应俱全,新手也能快速跑通"多候选选最优"的推理流程
一句话记住它:VisualPRM-8B 的能力上限,很大程度就是 VisualPRM400K 这条自动化数据管线的上限。
【免费下载链接】VisualPRM-8B
相关推荐
多模态数据集构建完全指南:从零开始为Align-Anything准备训练数据
多模态数据集构建完全指南:从零开始为Align Anything准备训练数据 在当今AI领域,多模态模型正成为突破的关键。Align Anything作为一个强
大模型多模态RLHF模型微调深度学习FormalVerse数据集揭秘:MathForm-8B训练数据完全解析
FormalVerse数据集揭秘:MathForm 8B训练数据完全解析 你是否好奇,MathForm 8B 这款自动形式化(Autoformalization
MMPR 300 万偏好数据是怎么来的?InternVL2_5-8B-MPO 多模态偏好数据集构造管线拆解
MMPR 300 万偏好数据是怎么来的?InternVL2_5 8B MPO 多模态偏好数据集构造管线拆解 InternVL2_5 8B MPO 是一个经过 混
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考