OmniVideo-R1:通过查询意图和模态注意力强化音视频推理 OmniVideo-R1: Reinforcing Audio-visual Reasoning with Query Intention and Modality Attention
首个基于强化学习的混合模态推理框架,显著提升音视频理解能力
前置知识
多模态大语言模型 (MLLM)
多模态大语言模型是在纯文本LLM基础上扩展的模型,能够同时处理视觉、音频、文本等多种模态的输入。典型代表包括Qwen-Omni系列、GPT-4o等。这类模型通过编码器将不同模态转换为token序列,再由统一的语言模型进行理解和推理。然而,多模态模型面临一个关键挑战:在融合多种模态时,可能出现模态偏见问题,即模型过度依赖某一模态而忽视其他模态的关键信息。
本文的研究对象就是omnimodal MLLM,理解其架构和训练范式是理解本文贡献的基础
强化学习 (RL) 用于LLM后训练
强化学习已逐渐成为提升大语言模型推理能力的主流范式,DeepSeek-R1等模型通过GRPO等算法显著提升了模型的推理能力。在多模态领域,RL被用于训练模型在推理过程中主动进行视觉证据追踪和推理。本文采用的GSPO是一种序列级别的优化策略,通过对整个响应序列计算重要性比率来避免token级别采样带来的高方差问题,特别适合MoE架构。
本文的核心方法就是基于强化学习的两阶段训练,理解RL在LLM中的应用范式是理解本文方法的关键
查询驱动的推理 (Query-intensive Reasoning)
查询驱动的推理是指模型能够根据用户问题的意图,主动定位和追踪与问题相关的关键信息片段。在视频理解场景中,这意味着模型需要从长视频中找到与问题相关的关键帧和音频片段,而不是被动处理所有输入。这种能力对于需要跨模态证据整合的复杂推理任务尤为重要。
本文的QI阶段就是训练模型进行查询驱动的音视频片段定位,这是理解方法的第一步
对比学习 (Contrastive Learning)
对比学习是一种通过比较正负样本对来学习表示的方法。在本文中,对比学习被用于模态注意力融合阶段,通过比较模型在全模态输入、纯视频输入和纯音频输入下的表现差异,强制模型从多模态融合中获得比单一模态更高的置信度。这种设计迫使模型学习音频和视觉事件之间的协同关系。
本文的MA阶段核心创新就是对比模态奖励,理解对比学习原理有助于深入理解这一设计
研究动机
当前omnimodal模型在音视频理解任务中面临一个悖论:增加音频模态不仅没有提升性能,反而可能损害模型已有的视觉推理能力。实验数据清楚地展示了这一问题:在Qwen3-Omni-30B-A3B家族中,Omni变体(音频-视觉)在MMStar基准上从VL变体(纯视觉)的72.1分下降到68.5分,在MathVista_mini上从80.1分下降到75.9分。这说明当前的训练范式存在根本性缺陷:现有的后训练方法(SFT或标准RL如GRPO)并没有显式地训练音视频混合推理行为,而是让模型可以通过数据集偏见或单模态捷径来获得正确答案,忽视了决定性的音频或视觉线索。这种模态偏见问题的根源在于:预训练阶段本质上涉及不同任务之间的权衡,自然会导致模态偏见;而后训练阶段缺乏针对混合模态推理的专门设计。
本文的目标是本文的目标是设计一个专门针对混合模态推理的后训练框架,使模型能够主动选择和融合多模态信息,而不是被动地处理所有输入。具体而言,框架需要实现两个核心能力:(1)查询密集的定位能力,即根据用户问题的意图,从视频序列中精确定位包含关键音视频线索的片段;(2)模态注意力融合能力,即充分挖掘和协同整合音频和视觉信息,确保混合模态输入的表现严格优于任何单一模态输入。最终目标是在不损害视觉推理能力的前提下,显著提升音视频联合理解的性能。
与已有工作不同的是,本文的独特切入点在于将混合模态推理建模为一个查询驱动的推理过程,而不是简单地增加数据或改进架构。现有方法主要集中在三个方向:直接增加混合模态监督数据(但高质量数据稀缺且对齐困难)、设计复杂的多分支架构(增加了模型复杂度)、或直接迁移标准RL方法(但没有针对音视频融合的特殊设计)。本文首次提出通过强化学习来显式地训练音视频推理行为:第一阶段通过自监督学习建立查询密集的定位行为,第二阶段通过对比学习强制模态注意力融合。这种方法的创新之处在于不需要过程级标注,而是通过自监督的方式让模型学习中间推理步骤。
核心方法
OmniVideo-R1的整体思路可以概括为'先学会找,再学会融'。直觉上,如果一个模型不能根据问题意图定位到关键的音视频片段,就无法进行有效的推理;而即使能够定位到正确的片段,如果不能有效地融合音频和视觉信息,推理效果也会大打折扣。因此,本文设计了两阶段的强化学习范式:第一阶段(QI,查询密集定位)训练模型'带着意图思考',即根据用户问题主动定位并推理关键的音视频片段;第二阶段(MA,模态注意力融合)则进一步提升模型对音视频线索的利用效率,确保混合输入的表现严格优于单模态输入。整个框架基于GSPO算法进行序列级别的优化,避免了token级别采样在MoE模型上带来的高方差问题。
本文的核心创新点在于两个关键设计:(1)自监督的查询密集定位:不同于现有的视觉推理方法需要过程级标注(如关键帧位置标注),本文设计了一种自监督学习方案。模型生成多个时间-字幕对(time-caption pairs),然后通过评判模型(Qwen3-VL-235B)来评估每个时间片段与其对应字幕的一致性。这种设计使得模型可以在没有显式标注的情况下学习中间推理步骤。(2)对比模态奖励:本文设计了一种新颖的奖励机制,对于每个输入,比较模型在三种配置下的表现:全模态输入(音频+视频)、纯视频输入和纯音频输入。只有当全模态输入的表现优于其他两种单模态输入时,才给予奖励。这种设计强制模型学习音频和视觉事件之间的协同关系,而不是依赖单一模态。这两个设计的结合使得模型能够'用全模态线索思考'。
方法步骤详情
方法分为数据准备和两阶段训练共三个步骤。数据准备阶段:从LLaVA-Video和Video-Vista收集原始数据,经过结构验证、质量评估(使用Gemini-2.5-Pro在视频依赖性、音频依赖性、问题逻辑、回答准确性四个维度打分)、启发式过滤(要求回答准确性=1,问题逻辑大于等于0.8,综合分大于等于0.7)和类别平衡(最大类别不超过第二大类别的3倍),最终得到88,173个样本用于QI阶段,12,887个高音频-视觉依赖样本用于MA阶段。QI阶段训练:模型生成格式为...............的输出,奖励由格式奖励、结果奖励、一致性奖励(评估时间-字幕对的一致性)和完整性奖励(评估定位片段对推理的支持程度)组成。MA阶段训练:对于每个输入,在三种配置下采样响应(全模态、纯视频、纯音频),只有当全模态输入得分最高时才给予注意力奖励,系数为0.3。总奖励由格式奖励、结果奖励和注意力奖励组成。
技术新颖性
本文的技术新颖性体现在多个层面。首先,这是第一个专门为混合模态推理设计的RL框架,填补了omnimodal模型后训练方法的空白。其次,自监督的查询密集定位是一个重要的方法论创新:通过让模型自己生成时间-字幕对并评估一致性,避免了昂贵的过程级标注,同时建立了查询意图与定位行为之间的联系。这种自监督方式的巧妙之处在于,它将定位任务转化为一个生成-验证问题,利用大模型自身的评判能力来提供训练信号。第三,对比模态奖励的设计非常精巧:通过比较三种模态配置下的表现,隐式地鼓励模型学习跨模态的协同关系,而不需要显式定义什么是'好的融合'。最后,采用GSPO进行序列级别的优化也是一个技术亮点,因为对于MoE架构的omnimodal模型,token级别的重要性采样会因为不同专家的激活而导致分布偏移,产生高方差的梯度估计。
实验结果
本文在多个音视频和纯视觉基准上进行了全面的实验,结果表明OmniVideo-R1在保持视觉性能的同时显著提升了音视频推理能力。在音视频理解方面,OmniVideo-R1在Daily-Omni上达到82.8分,超越了闭源SOTA模型Gemini-3-Pro(81.1分)2.1个百分点;在IntentBench上达到74.2分,超越Gemini-3-Pro(71.5分)3.8个百分点;在OmniVideoBench上达到44.8分,相比基线模型Qwen3-Omni-30B-A3B-Instruct(37.0分)提升了7.8个百分点,也超越了更大规模的开源模型Video-SALMONN 2+-72B(79.4分)。在纯视觉基准上,OmniVideo-R1不仅没有性能下降,反而有所提升:在Video-MME上达到73.6分(基线70.5分,+3.1)、在LVBench上达到51.9分(基线50.2分,+1.7)。消融实验进一步验证了各组件的有效性:移除过程级奖励导致OmniVideoBench从44.8下降到38.4;单独使用QI阶段就能在OmniVideoBench上达到43.6分(基线37.0分);单独使用MA阶段能达到41.6分。与直接SFT或标准RL相比,OmniVideo-R1在Daily-Omni上超越第二好的Vanilla RL 12.0个百分点(82.8 vs 73.9),在WorldSense上超越第二好的CoT SFT 11.1个百分点(65.8 vs 59.2)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| Daily-Omni(音视频问答) | Accuracy | 82.8 | Gemini-3-Pro 81.1, Qwen3-Omni-Thinking 75.8 | +1.7 vs Gemini-3-Pro, +7.0 vs Qwen3-Omni-Thinking |
| WorldSense(omnimodal协作) | Accuracy | 65.8 | Gemini-3-Pro 66.4, Qwen3-Omni-Thinking 48.0 | 第二名(仅次于Gemini-3-Pro),+17.8 vs Qwen3-Omni-Thinking |
| IntentBench(复杂意图理解) | Accuracy | 74.2 | Gemini-3-Pro 71.5, HumanOmniV2-7B 69.3 | +2.7 vs Gemini-3-Pro |
| VideoHolmes(复杂视频推理) | Accuracy | 62.9 | Gemini-3-Pro 67.0, Qwen3-Omni-Thinking 57.3 | 第二名,+5.6 vs Qwen3-Omni-Thinking |
| OmniVideoBench(音视频协同推理) | Accuracy | 44.8 | Qwen3-Omni-Instruct 37.0 | +7.8pp |
| Video-MME(纯视觉理解) | Accuracy | 73.6 | Qwen3-Omni-Instruct 70.5 | +3.1 |
| MLVU(Dev)(长视频理解) | Accuracy | 74.1 | Qwen3-Omni-Instruct 75.2 | -1.1(基本持平) |
| LVBench(超长视频理解) | Accuracy | 51.9 | Qwen3-Omni-Instruct 50.2 | +1.7 |
局限与改进
本文的局限性主要体现在以下几个方面。首先,当前方法仍然依赖于基于结果的ground-truth进行训练,即需要知道最终答案才能计算奖励。在缺乏ground-truth的场景下(如开放式的音视频理解任务),如何有效地训练模型仍是一个挑战。作者自己也承认这是一个重要的未来研究方向。其次,虽然消融实验表明QI和MA阶段都有贡献,但两阶段之间的最优数据配比(88,173 vs 12,887)可能不是全局最优的,且缺乏对不同配比的系统性分析。第三,QI阶段的自监督依赖于Qwen3-VL-235B作为评判模型,这引入了一个外部依赖:评判模型的偏见可能影响训练信号的质量。第四,在某些基准上(如VideoHolmes,62.9 vs Gemini-3-Pro的67.0),OmniVideo-R1仍然落后于闭源模型,说明在复杂推理任务上还有提升空间。此外,训练资源需求较高(128倍H20 GPU配置,全局batch size 256),可能限制了方法的可复现性。
独立分析的弱点
独立分析本文存在以下弱点:(1)评判模型依赖问题:QI阶段使用Qwen3-VL-235B作为一致性评判模型,这可能引入循环依赖——用一个可能有偏见的模型来训练另一个模型。改进方向是设计更鲁棒的自监督信号,例如利用多评判模型的集成或基于规则的评估。(2)超参数敏感性:MA阶段的注意力奖励系数是固定的0.3,缺乏自适应机制。不同任务对模态融合的依赖程度不同,固定的系数可能不是最优的。改进方向是根据任务难度或模态相关性动态调整系数。(3)数据效率:MA阶段仅使用了12,887个样本(从88,173个中筛选),说明大部分数据的音频-视觉依赖性不够高。改进方向是设计更精细的数据筛选策略或数据增强方法来扩大有效训练集。(4)缺乏过程级奖励:虽然自监督避免了标注成本,但QI阶段的奖励信号仍然相对稀疏(只在序列结束时评估)。如果能引入更细粒度的过程级奖励(如关键帧级别的评估),可能会进一步提升定位精度。
未来方向
作者提出了几个有价值的未来研究方向:(1)探索在缺乏ground-truth场景下的训练方法,这对于开放式的音视频理解任务至关重要。(2)将多模态训练范式扩展到更多模态,如3D点云、热成像等,作者认为查询意图和模态注意力的设计原则可以泛化到更广泛的多模态场景。基于本文的成果,还可以延伸以下方向:(3)将QI阶段的自监督定位能力与视觉定位任务(如Visual Grounding、Referring Expression Comprehension)结合,探索跨任务的知识迁移。(4)研究模态偏见的诊断和缓解方法,开发更系统的评估指标来量化模型的模态平衡程度。(5)探索在线学习范式,让模型在推理过程中动态调整模态注意力权重,而不是依赖固定的训练后权重。
复现评估
从复现角度来看,本文的复现难度较高。数据方面,作者使用了公开的LLaVA-Video和Video-Vista数据集,并提供了详细的数据筛选流程(三阶段pipeline),这部分可以复现。但数据质量评估依赖Gemini-2.5-Pro API,这需要相应的API访问权限和成本。算力方面,训练配置为128倍H20 GPU,全局batch size 256,最大序列长度32,768,这对于大多数研究团队来说是相当高的资源需求。代码方面,论文未明确提及是否开源代码和模型权重,但提供了详细的超参数配置(学习率、rollout数8、warmup比例0.05等),有助于复现。模型方面,基于Qwen3-Omni-30B-A3B进行训练,这是一个公开可用的基础模型。总体而言,完全复现需要大规模GPU集群和外部API访问,但核心算法流程描述清晰,理论上可以复现。
论文图表
图中展示了Qwen3-Omni-30B-A3B家族中Omni变体与VL变体在MMStar和MathVista_mini基准上的性能对比。数据显示Omni变体在MMStar上从72.1下降到68.5,在MathVista_mini上从80.1下降到75.9,直观地展示了增加音频模态反而损害视觉推理能力的悖论。
这张图是论文的核心动机图,清楚地展示了omnimodal模型面临的模态偏见问题,是理解整篇论文研究背景的关键。
对比了OmniVideo-R1与多个闭源和开源模型在Daily-Omni、WorldSense、IntentBench和VideoHolmes四个音视频基准上的表现。OmniVideo-R1在Daily-Omni上达到82.8(超越Gemini-3-Pro的81.1),在IntentBench上达到74.2(超越Gemini-3-Pro的71.5),在WorldSense上达到65.8(接近Gemini-3-Pro的66.4)。
这是论文的核心结果表,全面展示了OmniVideo-R1在音视频理解任务上的领先性能。
在更具挑战性的OmniVideoBench上,按音频类型(音乐、声音、语音)和视频时长进行细粒度分析。OmniVideo-R1整体达到44.8分,相比基线Qwen3-Omni-Instruct的37.0分提升了7.8个百分点,在所有维度上都超越了基线。
OmniVideoBench强调音视频协同理解,这张表展示了OmniVideo-R1在需要深度融合的任务上的显著优势。
在Video-MME、MLVU和LVBench三个纯视觉基准上,OmniVideo-R1不仅没有性能下降,反而在Video-MME上从70.5提升到73.6,在LVBench上从50.2提升到51.9,证明了混合模态训练不会损害视觉能力。
这张表回答了一个关键问题:音视频联合训练是否会损害视觉推理能力?答案是否定的。
对比了QA SFT、CoT SFT、Vanilla RL和完整的OmniVideo-R1在三个基准上的表现。OmniVideo-R1在Daily-Omni上超越第二好的Vanilla RL 12.0个百分点(82.8 vs 73.9),在WorldSense上超越第二好的CoT SFT 11.1个百分点(65.8 vs 59.2)。
这张表证明了OmniVideo-R1两阶段RL训练范式的优越性,而不仅仅是数据质量的贡献。
系统地移除各个组件(过程级奖励、注意力奖励、QI阶段、MA阶段)并测试性能变化。移除过程级奖励导致OmniVideoBench从44.8下降到38.4(-6.4),移除注意力奖励导致从44.8下降到43.7(-1.1)。还测试了移除音频输入和添加时间戳的影响。
消融实验是验证各组件有效性的关键,这张表提供了定量的证据支持论文的核心贡献。