OmniJigsaw: Enhancing Omni-Modal Reasoning via Modality-Orchestrated Reordering
Organizations: Zhejiang University · Xiaomi Inc.
Abstract
To extend the reinforcement learning post-training paradigm to omni-modal models for concurrently bolstering video-audio understanding and collaborative reasoning, we propose OmniJigsaw, a generic self-supervised framework built upon a temporal reordering proxy task. Centered on the chronological reconstruction of shuffled audio-visual clips, this paradigm strategically orchestrates visual and auditory signals to compel cross-modal integration through three distinct strategies: Joint Modality Integration, Sample-level Modality Selection, and Clip-level Modality Masking. Recognizing that the efficacy of such proxy tasks is fundamentally tied to puzzle quality, we design a two-stage coarse-to-fine data filtering pipeline, which facilitates the efficient adaptation of OmniJigsaw to massive unannotated omni-modal data. Our analysis reveals a ``bi-modal shortcut phenomenon'' in joint modality integration and demonstrates that fine-grained clip-level modality masking mitigates this issue while outperforming sample-level modality selection. Extensive evaluations on 15 benchmarks show substantial gains in video, audio, and collaborative reasoning, validating OmniJigsaw as a scalable paradigm for self-supervised omni-modal learning.
Figures & tables
| Method | AoT Bench | TUNA -Bench | Temp Compass | Video -TT | Video -Holmes | MLVU -Test | Video -MME | MLVU |
| Infer w/ Audio | ||||||||
| Omni-R1 | 52.09 | 53.84 | 63.10 | 36.80 | 40.72 | 52.59 | 63.20 | 65.69 |
| HumanOmniV2 | 48.58 | 49.16 | 63.86 | 40.30 | 42.90 | 49.00 | 65.60 | 66.70 |
| Qwen3-Omni-30B | 64.88 | 62.57 | 70.63 | 44.30 | 50.84 | 57.97 | 72.10 | 70.01 |
| VideoJigsaw | 67.45 2.57 | 63.41 0.84 | 72.28 1.65 | 44.90 0.60 | 51.99 1.15 | 60.16 2.19 | 72.90 0.80 | 71.90 1.89 |
| OmniJigsaw (JMI) | 66.83 1.95 | 62.78 0.21 | 71.08 0.45 | 44.70 0.40 | 50.24 0.60 | 58.76 0.79 | 72.90 0.80 | 71.39 1.38 |
| Method | MMAU -Pro | MMAU -test-mini | MMSU | MMAR |
| Omni-R1 | 52.36 | 77.70 | 61.87 | 59.70 |
| HumanOmniV2 | 53.49 | 75.90 | 60.83 | 61.80 |
| Qwen3-Omni-30B | 56.61 | 74.40 | 70.16 | 68.50 |
| AudioJigsaw | 57.67 1.06 | 75.40 1.00 | 70.30 0.14 | 70.70 2.20 |
| OmniJigsaw (JMI) | 58.33 1.72 | 74.50 0.10 | 69.80 0.36 | 69.10 0.60 |
| OmniJigsaw (SMS) | 58.46 1.85 | 75.80 1.40 | 70.48 0.32 | 69.50 1.00 |
| Method | Daily -Omni | Intent Bench | Omni Video Bench |
| Omni-R1 | 54.14 | 64.18 | 32.40 |
| HumanOmniV2 | 58.48 | 68.21 | 33.50 |
| Qwen3-Omni-30B | 69.92 | 67.40 | 38.80 |
| OmniJigsaw (JMI) | 70.26 0.34 | 67.95 0.55 | 40.10 1.30 |
| OmniJigsaw (SMS) | 70.26 0.34 | 68.21 0.81 | 40.20 1.40 |
| OmniJigsaw (CMM) | 71.09 1.17 | 68.89 1.49 | 40.50 1.70 |
| Video | Audio | Omni-Modal | |||||||||||||
| Method | AoT | TUNA | TempC | V-TT | V-Holmes | MLVU-T | V-MME | MLVU | MMAU-P | MMAU-TM | MMSU | MMAR | D-Omni | Intent | OVB |
| OmniJigsaw | 66.83 | 66.20 | 72.34 | 46.50 | 48.29 | 62.75 | 69.30 | 73.46 | 58.59 | 76.30 | 70.70 | 71.00 | 71.09 | 68.89 | 40.50 |
| w/o Filtering | 64.94 1.89 | 65.29 0.91 | 71.14 1.20 | 45.40 1.10 | 47.41 0.88 | 58.76 3.99 | 68.60 0.70 | 72.59 0.87 | 57.67 0.92 | 76.00 0.30 | 70.40 0.30 | 68.90 2.10 | 70.01 1.08 | 66.77 2.12 | 40.10 0.40 |
| w/o DF | 66.00 0.83 | 64.11 2.09 | 71.27 1.07 | 44.70 1.80 | 47.96 0.33 | 61.95 0.80 | 68.60 0.70 | 72.36 1.10 | 57.81 0.78 | 75.90 0.40 | 70.48 0.22 | 69.30 1.70 | 69.76 1.33 | 67.66 1.23 | 39.50 1.00 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Category | Parameter | Value |
| Visual Input | Maximum Sampled Frames | 200 |
| Maximum Resolution (pixels) | 100,352 | |
| Sampling Params | Temperature ( ) | 0 (Greedy Decoding) |
| Top- | 1.0 | |
| Top- | -1 (Disabled) | |
| Repetition Penalty | 1.05 |
| Source Dataset | Raw Samples | After Stage 1 | After Stage 2 |
| YouCook2 [ 59 ] | 2,000 | 327 | 327 |
| FineVideo [ 5 ] | 43,751 | 7,737 | 6,986 |
| LLaVA-Video-178K [ 54 ] (NextQA) | 3,868 | 982 | 907 |
| Total | 49,619 | 9,046 | 8,220 |
| Hyperparameter | Value |
| Model & Infrastructure | |
| Base Policy Model | Qwen3-Omni-30B-A3B-Instruct [ 42 ] |
| Hardware Infrastructure | NVIDIA H200 GPUs |
| Frozen Components | Vision Tower, Audio Tower, Router |
| Training Dynamics (VeRL) | |
| Total Training Steps | 1,000 |
| Modality | Benchmark | Evaluation File |
| Video | AoTBench [ 43 ] | data_files/AoTBench_QA.json |
| TUNA-Bench [ 12 ] | TUNA-MCQ.json | |
| TempCompass [ 19 ] | multi-choice/test-00000-of-00001.parquet | |
| Video-TT [ 53 ] | data/test-00000-of-00001.parquet | |
| Video-Holmes [ 3 ] | test_Video-Holmes.json | |
| MLVU-Test [ 58 ] | test_multi_choice_tasks.json |
| Parameter | Value |
| Temperature ( ) | 0 (Greedy Decoding) |
| Top- | 1.0 |
| Top- | -1 (Disabled) |
| Max New Tokens | 2,048 |
| Repetition Penalty | 1.05 |
| Method | Temporal Perception | Spatial Perception | Attribute Perception | Action Recognition | Object Recognition | OCR Problems | Counting Problem | Temporal Reasoning | Spatial Reasoning | Action Reasoning | Object Reasoning | Information Synopsis |
| Qwen3-Omni-30B | 80.0 | 70.4 | 77.0 | 69.0 | 76.6 | 74.1 | 46.3 | 57.1 | 78.6 | 57.2 | 65.9 | 80.5 |
| OmniJigsaw (JMI) | 78.2 | 74.1 | 78.8 | 71.2 | 74.6 | 75.5 | 47.4 | 62.7 | 76.8 | 58.6 | 65.9 | 80.8 |
| OmniJigsaw (SMS) | 80.0 | 72.2 | 80.2 | 71.2 | 74.6 | 74.1 | 48.1 | 61.6 | 73.2 | 59.3 | 67.0 | 80.5 |
| OmniJigsaw (CMM) | 76.4 | 75.9 | 78.8 | 73.8 | 76.3 | 77.0 | 44.4 | 62.1 | 73.2 | 61.8 | 65.9 | 80.8 |
| Task-wise | Difficulty-wise | |||||
| Method | Sound | Music | Speech | Easy | Hard | Medium |
| Qwen3-Omni-30B | 76.88 | 72.75 | 73.57 | 70.09 | 70.76 | 77.78 |
| OmniJigsaw (JMI) | 77.18 | 73.65 | 72.67 | 70.54 | 68.22 | 78.89 |
| OmniJigsaw (SMS) | 78.98 | 73.65 | 74.77 | 70.98 | 68.64 | 80.93 |
| OmniJigsaw (CMM) | 79.28 | 75.15 | 74.47 | 73.66 | 72.03 | 79.26 |
| Method | Attribute Comparison | Bkg. & Music Understanding | Causal Reasoning | Counting | Ego Reasoning | Fine-grained Perception | Hypothetical Reasoning | Reference Reasoning | Relationship Reasoning | Sentiment Analysis | Spatial Understanding | Summarization | Temporal Understanding |
| Qwen3-Omni-30B | 28.57 | 34.04 | 47.14 | 25.16 | 45.07 | 38.46 | 58.33 | 41.84 | 70.83 | 38.27 | 22.58 | 63.27 | 33.58 |
| OmniJigsaw (JMI) | 61.90 | 31.91 | 49.29 | 21.29 | 54.93 | 47.25 | 45.83 | 45.92 | 66.67 | 38.27 | 22.58 | 55.10 | 32.85 |
| OmniJigsaw (SMS) | 38.10 | 27.66 | 50.00 | 26.45 | 47.89 | 41.76 | 41.67 | 37.76 | 79.17 | 44.44 | 30.65 | 61.22 | 34.31 |
| OmniJigsaw (CMM) | 38.10 | 29.79 | 46.43 | 24.52 | 43.66 | 41.76 | 58.33 | 44.90 | 70.83 | 41.98 | 32.26 | 55.10 | 40.15 |
Audio-Visual Understanding
Audio-visual understanding studies how AI combines sound and visual signals to recognize events, speech, people, and scenes over time.
See all 155 papers