SuperAPI AI 工具集

R1-Omni

R1-Omni 是行业首个将可验证奖励强化学习(RLVR)应用于全模态大语言模型的项目,聚焦于情绪识别任务,旨在验证 RLVR 与全模态模型结合的潜力。项目基于开源 Omni 模型 HumanOmni-0.5B 构建,并已开源基础模型、冷启动模型 EMER-SFT、直接在 MAFW 和 DFEW 训练集上微调的 MAFW-DFEW-SFT 模型以及最终模型 R1-Omni。研究显示,相比监督微调(SFT),RLVR 显著提升了情绪识别任务的表现,并增强了推理能力、理解能力和泛化能力,尤其在分布外场景中表现突出。该项目曾获人民日报和彭博社报道。

访问官网

工具介绍

R1-Omni 是一个专注于可解释全模态情绪识别的开源项目,由 HumanMLLM 团队开发。其核心创新在于将可验证奖励强化学习(RLVR)首次应用于全模态大语言模型,以情绪识别这一视觉与音频模态均至关重要的任务为切入点,验证 RLVR 与全模态模型结合的潜力。

主要能力

根据项目资料,R1-Omni 展现出以下关键特性:

  • 增强的推理能力:能够更清晰地理解视觉和音频信息如何贡献于情绪识别。
  • 提升的理解能力:相比 SFT,RLVR 显著提升了情绪识别任务的性能。
  • 更强的泛化能力:RLVR 模型在分布外场景中表现尤为突出。

项目已开源多个模型,包括基础模型 HumanOmni-0.5B、冷启动模型 EMER-SFT、在 MAFW 和 DFEW 训练集上直接微调的 MAFW-DFEW-SFT,以及最终模型 R1-Omni。模型在情绪识别数据集上的表现使用符号标注数据是否为分布内(⬤,DFEW 和 MAFW)或分布外(△,RAVDESS)。

使用方式

项目提供了环境设置和推理步骤:

  • 代码基于 R1-V 框架,环境要求包括 Nvidia-driver 535.54、torch 2.5.1+cu124、torchvision 0.20.1+cu124、torchaudio 2.5.1+cu124、transformers 4.49.0、flash_attn 2.7.4。
  • 推理代码基于 HumanOmni 实现。需下载所需模型:siglip-224、whisper-large-v3、bert-base-uncased,并更新配置文件中的路径。
  • 提供 inference.py 用于单视频推理。
  • 冷启动阶段在由 232 个 Explainable Multimodal Emotion Reasoning 数据集样本和 348 个 HumanOmni 数据集样本组成的组合数据集上微调 HumanOmni-0.5B。
  • 第二阶段使用 MAFW 和 DFEW 数据集训练集,共 15,306 个视频样本。

适用场景

适用于情绪识别研究,特别是需要结合视觉和音频模态进行可解释情绪推理的场景,以及探索 RLVR 在全模态模型中的应用。

限制

项目资料中未提供收费、价格、平台、效果或具体使用步骤等细节,部分更新(如单视频和单音频模态推理、7B 版本结果、训练数据开源)尚在计划中,暂未提供。