SuperAPI AI 工具集

PR1

Perception-R1(PR1)是 NeurIPS 2025 论文的官方代码实现,提出了一种可扩展的强化学习框架,通过 Group Relative Policy Optimization(GRPO)对多模态大语言模型(MLLM)进行后训练,以提升视觉感知能力。其核心创新包括:感知困惑度分析,用于揭示感知任务中有效强化学习的关键阈值;基于规则的奖励塑造与 GRPO 优化,实现可扩展的策略学习;在多个视觉感知基准上取得显著提升,例如在 COCO2017 验证集上达到 31.9% mAP,成为首个达到该性能的 3B 规模 MLLM。项目提供了训练与评估代码,支持 grounding、counting、detection 等任务,用户需自行配置模型与数据路径,并单独下载 COCO 图像。

访问官网

工具介绍

Perception-R1 是一个面向视觉感知任务的强化学习后训练框架,由论文《Perception R1: Pioneering Perception Policy with Reinforcement Learning》提出,官方代码托管于 GitHub。本项目旨在利用强化学习提升多模态大语言模型(MLLM)在视觉感知任务上的表现。

用途方面,Perception-R1 专注于视觉感知策略的优化,通过 Group Relative Policy Optimization(GRPO)在 MLLM 后训练阶段进行可扩展的强化学习。其目标是在 grounding、counting、detection 等感知任务中提升模型的视觉理解能力。

主要能力包括:

  • 感知困惑度分析:提出一种新的分析框架,揭示感知任务中有效强化学习的关键阈值,为理解 RL 何时以及如何改善视觉理解提供洞见。
  • GRPO 优化:采用精心设计的基于规则的奖励塑造,进行可扩展的策略学习。
  • 性能提升:在多个视觉感知基准上取得显著改进,特别是在 COCO2017 验证集上达到 31.9% mAP,成为首个达到该性能的 3B 规模 MLLM。

使用方式:

  • 训练前,需修改脚本以指定模型和数据路径,然后运行实验。训练脚本包含超参数、数据加载和模型检查点的综合配置,可根据自定义场景调整学习率、批次大小等参数。
  • 评估:从 Hugging Face 下载评估数据,解压到 eval/ 文件夹。注意 COCO 图像不包含在包内,需从 COCO 官网单独下载并放置于 eval/images/coco/ 目录。评估支持 counting、grounding、detection 等任务。

适用场景:适用于研究多模态大语言模型视觉感知能力提升的学术或工业场景,特别是需要利用强化学习优化 grounding、检测、计数等任务的研究人员。

限制:COCO 图像需用户自行下载,项目未提供图像数据;训练和评估需要用户自行配置模型与数据路径;更多感知任务(如 OCR)计划发布但暂未提供。

项目持续更新:2025-04-10 首次发布模型和评估代码;2025-05-27 计划增加更多感知任务。