OmniJigsaw:通过模态编排式重排序增强全模态推理 OmniJigsaw: Enhancing Omni-Modal Reasoning via Modality-Orchestrated Reordering
基于音视频片段时序重排的免标注自监督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 语义筛选'的两阶段流水线,使框架可扩展到百万级无标注视频。
实验结果
实验在 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 代理任务可用性的两个关键支柱。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 视频长时理解 (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 算力与全模态推理栈的部署经验,而非算法细节。
论文图表
展示一段被启发式过滤保留但被语义过滤拒绝的视频:寺庙内连续演唱,背景不变。即便首阶段满足动态性,Qwen2.5-VL-7B-Instruct 仍评估为'无显著视觉状态变化、无时间标记',输出 NO。
典型案例说明仅靠低层信号过滤会漏掉'温和动态但缺因果推进'的视频,是数据管线合理性的具象化论据。
一段被拒绝的视频:场景在户外阅读 / 浴缸 / 室内艺术品之间突变,Qwen2.5-VL-7B-Instruct 指出没有统一因果链,各 clip 可任意重排。
补充说明:仅视觉多样性不等于时序可还原,进一步强化语义过滤的必要性。
对一段关于罕见血型寻人的剧情长视频,Baseline 编造'富商 / 名人后裔 / 卑微出身'等无根据细节;CMM 版忠实总结出'特定血统 / 棉农 / 契约 / 输血成功'等关键事件,长程叙事一致性显著提升。
与 Fig. 10 一起提示 OmniJigsaw(CMM) 不只在基准上得分更高,也削弱了 unsupported speculation,对实际产品落地有直接意义。
在 CMM 设定下分别移除两阶段过滤(w/o Filtering)与折扣因子(w/o DF),15 项基准均一致下降,最严重 MLVU-Test -3.99、MMAR -2.10、IntentBench -2.12(TUNA-Bench -2.09 在 DF 消融中出现)。
全论文分量最重的消融表,直接量化'数据质量'与'奖励设计'对 proxy task 可用性的决定性影响。
15 个基准的模态与官方评估文件路径:视频类 8、音频类 4、全模态 3。
给出评测脚本所需的所有原始文件指示,省去研究者重新搜索数据集的时间。