← 返回 2026-04-10

OmniJigsaw:通过模态编排式重排序增强全模态推理 OmniJigsaw: Enhancing Omni-Modal Reasoning via Modality-Orchestrated Reordering

Yiduo Jia, Muzhi Zhu, Hao Zhong, Mingyu Liu, Yuling Xi, Hao Chen, Bin Qin, Yongjie Yang, Zhenbo Luo, Chunhua Shen 📅 2026-04-09 👍 27 2026-07-13 08:36
全模态学习 强化学习后训练 时序推理 自监督代理任务 音视频理解

基于音视频片段时序重排的免标注自监督RL后训练框架,通过三种模态编排策略提升全模态推理能力。

前置知识

强化学习后训练 (RL Post-Training)

在大语言或多模态基座模型完成大规模预训练之后,使用强化学习(如 GRPO、PPO、DPO)对模型进行进一步优化的范式。典型做法是先生成多条候选回答,再用可验证的奖励信号(如答案正确率、格式合规性)驱动策略更新,使模型在数学、代码等任务上涌现长链推理能力。本文将其扩展到全模态场景,用打乱音视频片段的时序还原任务作为自动可验证的奖励信号。

本文核心方法是 OmniJigsaw 框架,本质上就是一种针对全模态基模型的 RL 后训练范式。若不熟悉 GRPO、KL 约束、奖励折扣等概念,将难以理解为什么文中要精心设计组合奖励函数 $R_{tot}$ 与折扣因子 $\lambda(acc)$。

全模态大模型 (Omni-Modal LLM)

指能同时原生处理视频(V)、音频(A)和文本(T)输入并执行跨模态协同推理的大模型,典型代表为 Qwen3-Omni-30B-A3B-Instruct。这类模型通常包含冻结的视觉塔、音频塔与路由模块,后训练阶段主要对齐的是连接器或高层推理头。本文基座正是此类 30B 规模的混合专家(MoE,全文记作 A3B)模型。

所有实验的起点都是 Qwen3-Omni-30B-A3B-Instruct,理解其结构(视觉塔、音频塔、路由被冻结)才能看懂为什么 OmniJigsaw 是'轻量'的,以及为什么模态编排策略必须显式控制提示中暴露的模态。

拼图式自监督代理任务 (Jigsaw Self-Supervised Proxy Task)

把输入切成若干块再随机打乱,让模型预测原始顺序的任务。其核心思想是用数据自身的时序/空间结构作为监督,无需人工标注。在静态图像中用于学部件与空间布局,在视频中用于捕捉运动动态,本文将其首创性地扩展到音视频联合时序重排。形式上定义在大小为 $N$ 的置换群上,模型需预测与真实置换 $\pi$ 对齐的索引序列 $\hat{y}$。

OmniJigsaw 的整个问题定义都建立在这个范式上。若不理解'打乱—还原'为什么能充当 RL 奖励信号,以及 N 段裁剪与边界修剪 $T_{trim}$ 的设计用意,将无法理解为什么论文要专门讨论'双模态捷径现象'与组合奖励。

GRPO (Group Relative Policy Optimization)

一种无需额外价值网络的 RL 算法,对同一 prompt 采样一组(group)回答,根据这组内的相对奖励(即组内归一化得分)估计优势函数,从而更新策略。OmniJigsaw 使用 8 个采样(rollout group size $=8$),并使用 low$_$var$_$kl 估计器防止策略偏离太远。

OmniJigsaw 全文采用 GRPO 作为优化器,理解其'组内相对奖励'机理有助于理解为什么文中要引入 $\lambda(acc)$ 这种精确率依赖的折扣因子——它在 GRPO 的组内归一化层面人为放大'完美解'与'次优解'的价值差。

研究动机

当前 RL 后训练在纯文本推理(数学、代码)已取得显著突破,但向同时处理视频+音频的全模态模型迁移时遭遇两大瓶颈:第一,高质量、可验证的全模态标注数据极其昂贵,主流做法(如 Video-CoT、CoTasks、VIDEOP2R)依赖教师模型重标注,不仅成本高而且难以规模化;第二,现有多模态 RL 增强路线(如 Omni-R1、VideoWorld 2)需要复杂辅助目标、外部奖励模型或分阶段流程,工程开销大。这导致一个核心困境:能否借助'无需标注、可自动验证'的代理任务,将 GRPO 类训练范式扩展到全模态基模型?同时作者观察到,把视觉域的拼图重排(如 Jigsaw-R1)直接照搬进音视频时,会出现'模态捷径'——即在双模态并存的设定下,模型会被信息更丰富的优势模态吸引而忽略弱势模态,导致弱势模态的表示学习不足。

本文的目标是本文提出 OmniJigsaw 框架,目标是构建一种通用、免标注、轻量的自监督 RL 后训练范式,把'打乱音视频片段—恢复原始时序'作为可验证的代理任务,借助 GRPO 优化同时提升视频理解、音频理解以及音视频协同推理三类能力。为此必须解决三个子目标:(1)设计两种补偿捷径缺陷的模态编排策略(SMS 与 CMM);(2)设计两阶段数据过滤流水线,使 OmniJigsaw 能推广到海量无标注的全模态数据;(3)设计兼顾位置精度、连续性、格式合规与准确率折扣的组合奖励机制,引导模型追求完美恢复而非次优解。

与已有工作不同的是,本文的独特定位是把 Jigsaw 这种纯视觉代理任务首次扩展到音视频全模态,并把模态编排精细到样本级(SMS)和片段级(CMM)。与已有 Jigsaw-R1 等纯视觉工作相比,本文的切入点是'模态之间的协同与冲突':JMI 暴露并被证明存在'双模态捷径现象',而 CMM/SMS 则通过显式制造信息瓶颈或样本级仲裁,把它转化为'互惠协同'。因此本文既不是再做一个新数据集,也不是再设计一个新奖励函数,而是给出'代理任务质量—模态编排粒度—奖励函数'三者之间的耦合关系,对未来自监督全模态研究提供了可复用的范式。

核心方法

OmniJigsaw 把一段音视频切成 $N=6$ 个等长片段后随机洗牌,要求 omni-MLLM 给出还原后的索引序列 $\hat{y}=\mathcal{M}_\theta(\Phi(\tilde{S}); I_{prompt})$,其中 $\Phi(\cdot)$ 是策略相关的模态编排函数。三种策略对应三种信息暴露方式:JMI 同时保留视觉+音频;SMS 先让模型判别谁是主导模态 $d\in\{V,A\}$,再只用该模态做重排;CMM 由模型对每个片段生成模态选择向量 $m_i\in\{V,A,VA\}$,对弱势模态执行 zero masking 以形成跨模态信息瓶颈,整体优化由 GRPO 配合组合奖励驱动。

核心创新在于把'模态编排'作为可学习的代理任务超参数,并发现粗粒度暴露(JMI)会触发'双模态捷径',而细粒度屏蔽(CMM)反而把模型推向跨模态互惠。这一'由粗到细'的策略迁移不同于 Jigsaw-R1 等纯视觉版本,因为后者无模态冲突可言。本质区别是:CMM 的零替换操作 $\mathcal{M}_{cmm}$ 强制模型在每个 clip 上动态切换注意焦点,而 SMS 仅做样本级仲裁,仍可能漏掉局部高价值线索,这一差异在 Figure 7 的细粒度子能力评测中被充分暴露。

方法步骤详情

实施流程分为 6 步。第一步是数据准备:从 YouCook2、FineVideo 及 LLaVA-Video-178K 的 NextQA 子集共 49,619 条原始视频出发,经过两阶段筛选后得到 8,220 条样本。第二步是启发式信号过滤:使用 MAD 检测视觉静态帧,RMS 与 Spectral Flux 过滤静音与单调噪声,Silero VAD 把语音比例约束在 $[30\%,80\%]$,剔除静态比例超过 70% 的视频。第三步是语义筛选:使用 Qwen2.5-VL-7B-Instruct 以 CoT 形式判断是否具备不可逆的状态转移,剔除外观相似但顺序无关的拼接视频。第四步是样本裁剪:均匀切成 6 段,每段两端各剪掉 5% 时长防止边界帧被简单匹配。第五步是模态编排,按所选策略实例化 $\Phi_{jmi}, \Phi_{sms}, \Phi_{cmm}$。第六步是 GRPO 优化:8 路采样、$1\times10^{-6}$ 学习率、$1\times10^{-2}$ KL 系数,组合奖励 $R_{tot}=R_{rep}+R_{fmt}+\lambda(acc)(w_{pos}R_{pos}+w_{cont}R_{cont})$ 中 $w_{pos}=w_{cont}=0.5$,$\lambda=1.0$ 当完美匹配,否则为 $0.2$。

技术新颖性

技术新颖性体现在四个层面。第一,首次把拼图式代理任务从纯视觉扩展到音视频全模态,并把它形式化为置换预测任务。第二,发现并理论化了'双模态捷径现象':双模态并存的 JMI 反而比单模态 Jigsaw 更差,作者通过 Figure 4 的 CoT 对比直接证实模型在 JMI 下'完全依赖语言线索',跳过视觉分析。第三,提出两个细粒度编排函数 $\Phi_{sms}$(样本级最大优势选取)与 $\Phi_{cmm}$(片段级 zero masking),后者通过构造信息瓶颈反向逼迫模型进行跨模态互惠。第四,把原本依赖重标注教师模型的全模态数据管线,简化为'轻量信号过滤 + 7B MLLM 语义筛选'的两阶段流水线,使框架可扩展到百万级无标注视频。

OmniJigsaw framework for self-supervised omni-modal RL post-training.
Fig. 1: OmniJigsaw framework for self-supervised omni-modal RL post-training.
Data filtering pipeline for efficient adaptation of OmniJigsaw.
Fig. 2: Data filtering pipeline for efficient adaptation of OmniJigsaw.
Comparison of CoT reasoning between CMM and JMI at training step 800.
Fig. 4: Comparison of CoT reasoning between CMM and JMI at training step 800.

实验结果

实验在 15 个全模态基准上展开。基于 Qwen3-Omni-30B-A3B-Instruct 基座,CMM 在视频类 8 个基准上全面领先基线:MLVU-Test +4.38(62.75 vs 58.37)、AoTBench +4.02(68.90 vs 64.88)、Video-TT +2.70(46.50 vs 43.80)等,且在 Infer w/o Audio 设定下仍获得 MLVU-Test +3.99 提升,证明 CMM 习得的时序推理可被纯视觉推理吸收。音频类 4 个基准中,CMM 对 MMAR +2.50(71.00 vs 68.50)提升最大,凸显层次化音频推理增益。全模态协同类 3 个基准中,CMM 在 OmniVideoBench 取得 +1.70(40.50 vs 38.80)的提升,验证了跨模态依赖建模的价值。JMI 反常地接近基线甚至在某些基准略低,SMS 居中,这与作者主张的'CMM 通过信息瓶颈反向逼迫跨模态互惠'吻合。消融实验显示:去掉两阶段过滤后 MLVU-Test 下降 -3.99、MMAR 下降 -2.10、IntentBench 下降 -2.12;去掉折扣因子后 TUNA-Bench 下降 -2.09、MMAR 下降 -1.70、Daily-Omni 下降 -1.33,说明数据质量与折扣机制是支撑 jigsaw 代理任务可用性的两个关键支柱。

Performance comparison on video reasoning.
Table 1: Performance comparison on video reasoning.
Performance comparison on audio reasoning.
Table 2: Performance comparison on audio reasoning.
Performance comparison on omni-modal collaborative reasoning.
Table 3: Performance comparison on omni-modal collaborative reasoning.
Inference configurations for semantic screening via Qwen2.5-VL-7B-Instruct.
Table 5: Inference configurations for semantic screening via Qwen2.5-VL-7B-Instruct.
Data filtering statistics for OmniJigsaw-8K curation.
Table 6: Data filtering statistics for OmniJigsaw-8K curation.
Hyperparameter settings for OmniJigsaw GRPO post-training implemented via VeRL.
Table 7: Hyperparameter settings for OmniJigsaw GRPO post-training implemented via VeRL.
Evaluation sampling parameter configurations.
Table 9: Evaluation sampling parameter configurations.
Sub-capability evaluation results on Video-MME.
Table 10: Sub-capability evaluation results on Video-MME.
Sub-capability evaluation results on MMAU-test-mini.
Table 11: Sub-capability evaluation results on MMAU-test-mini.
Sub-capability evaluation results on OmniVideoBench.
Table 12: Sub-capability evaluation results on OmniVideoBench.
Performance comparison of JMI, CMM, and uni-modal Jigsaw across video, audio, and omni-modal benchmarks.
Fig. 3: Performance comparison of JMI, CMM, and uni-modal Jigsaw across video, audio, and omni-modal benchmarks.
Optimization dynamics w/ and w/o Discount Factor.
Fig. 5: Optimization dynamics w/ and w/o Discount Factor.
Task reward (w/o Rrep, Rfmt) dynamics of JMI, CMM, VideoJigsaw and AudioJigsaw.
Fig. 6: Task reward (w/o Rrep, Rfmt) dynamics of JMI, CMM, VideoJigsaw and AudioJigsaw.
Sub-capability performance comparison between CMM and SMS across fine-grained dimensions.
Fig. 7: Sub-capability performance comparison between CMM and SMS across fine-grained dimensions.
Qualitative example of Sub-Scene Captioning.
Fig. 10: Qualitative example of Sub-Scene Captioning.
查看结构化数据
任务指标本文基线提升
视频长时理解 (MLVU-Test,w/ Audio) top-1 准确率(%) 62.75 (OmniJigsaw-CMM) 57.97 (Qwen3-Omni-30B 基线) +4.38
时序敏感度 (AoTBench,w/ Audio) top-1 准确率(%) 68.90 64.88 +4.02
视频因果推理 (Video-TT,w/o Audio) top-1 准确率(%) 46.50 43.80 +2.70
视频综合基准 (Video-MME,w/o Audio) top-1 准确率(%) 69.30 67.90 +1.40
层次化音频推理 (MMAR) top-1 准确率(%) 71.00 68.50 +2.50
通用音频理解 (MMAU-Pro) top-1 准确率(%) 58.59 56.61 +1.98
音频细粒度感知 (MMSU) top-1 准确率(%) 70.70 70.16 +0.54
全模态协同推理 (OmniVideoBench) top-1 准确率(%) 40.50 38.80 +1.70
全模态行为意图推理 (IntentBench) top-1 准确率(%) 68.89 67.40 +1.49
全模态时序事件同步 (Daily-Omni) top-1 准确率(%) 71.09 69.92 +1.17

局限与改进

作者在 A.4 列出五点局限:(1) 仅在 30B-A3B 一种基模型上验证,模型规模、数据规模与异构全模态架构的可迁移性未知;(2) 数据筛选离线运行,无法在训练过程中根据模型能力动态调节拼图难度,缺课程化设计;(3) 代理任务局限于等长、无重叠的时序切分,缺乏变长/重叠/时空混合拼图的探索;(4) 奖励仅强调位置与邻接正确率,缺乏对推理过程结构本身(如 CoT 论证链)的显式监督;(5) 仅探索时序重排这一类自监督代理任务,更广泛的'需联合音视频线索'任务尚未铺开。我自己的额外观察包括:固定 $N=6$ 切片对于超长视频可能太短,奖励折扣 $\lambda=0.2$ 在多数样本注定不完美时可能让梯度信号过弱,且 SMS 的'先判别主导'两阶段会引入误差传播——一旦 dominance analyzer 选错模态,后续重排将完全失效。

独立分析的弱点

独立分析可见五个可改进点。第一,CMM 的零替换 $\mathcal{M}_{cmm}$ 是粗暴的 hard masking,可能破坏端到端的潜在对齐,建议改为按概率 $\sigma$ 注入可学习的 mask token 或低秩噪声向量,让梯度更平滑。第二,依赖 Qwen3-Omni-30B-A3B-Instruct 做 dominance analyzer 与 modality selector,相当于把决策外包给同一个尚未充分强化的基模型,容易出现错误传播——可改为训练一个轻量级、可独立监督的 selector head。第三,组合奖励 $R_{tot}$ 中格式奖励 $R_{fmt}=+0.2$ 与位置奖励 $w_{pos}R_{pos}$ 量级相近,在 CoT 较长时格式奖励的占比过高,会诱导模型学会'快速结束思考',建议把 $R_{fmt}$ 改为连续值,或与 CoT 长度负相关。第四,离线过滤 +8K 样本量相对 GRPO 的 1,000 step 训练仍偏小,CR 可能欠拟合,建议引入能力感知课程化采样,在训练中段提升拼图难度(如 $N=8$ 或非均匀切片)。第五,Figure 6 显示 JMI 的任务奖励显著高于 CMM,但下游反而更差,这暴露了 reward hacking 风险,可在奖励中加入对'理由文本与最终索引一致性'的语义一致性正则。

未来方向

作者在 A.4 中给出五个未来方向:扩展到不同模型规模/不同全模态架构、设计能力感知/课程化过滤、引入变长或重叠切分、加入结构感知奖励、探索更广义的自监督全模态代理任务。基于结果的延伸方向还包括:(1) 把 OmniJigsaw 与 Video-R1 的 jigsaw-only 方法做正交融合,验证其可叠加性;(2) 推广到多智能体协作场景,让 LLM agent 通过互相出题与互评提升;(3) 利用 OmniJigsaw 的自动奖励性质,把 OmniJigsaw 当作'奖励模型'辅助训练其他无标注全模态任务;(4) 把片段级 mask 决策对外开放为可控变量,做 persona 化的可控推理;(5) 与 Speech-aware Tokenizer、向量量化的离散模态编码结合,进一步把 jigsaw 推理演化为显式的符号推理。

复现评估

作者提供了相对透明的实现细节与超参(Table 7),但完整开源情况需查阅项目主页 https://aim-uofa.github.io/OmniJigsaw。复现所需的成本概览:(1) 硬件:8 块 NVIDIA H200 GPU,训练 1,000 step 约需若干天,按云端 H200 单价估算在数千到上万美元量级;(2) 数据:49,619 原始视频来自公开 YouCook2、FineVideo 与 LLaVA-Video-178K 的 NextQA 子集,过滤逻辑与阈值($\tau_v=5.0$、$-40$dB、SF 方差 $\tau_{sf}=0.5$、VAD 区间 $[0.3,0.8]$)均给出,理论上可复现 8,220 条样本;(3) 基座与算法:依赖 Qwen3-Omni-30B-A3B-Instruct、Qwen2.5-VL-7B-Instruct 与 VeRL 框架,皆为公开组件;(4) 评测:15 个基准的官方划分与 Prompt 模板在附录 A.5 全部列出,greedy 解码 + CoT 设置明确。整体复现难度中等偏高,主要瓶颈集中在大规模 GPU 算力与全模态推理栈的部署经验,而非算法细节。