SonicWorld: Source-Centric Latent Dynamics for Spatial Audio Reasoning
Organizations: Xi’an Jiaotong-Liverpool University · University of California, San Diego · Imperial College London · Lyuliang University · Southeast University
Abstract
Dynamic spatial audio reasoning requires a model to associate changing acoustic observations with the correct source and recover its spatial state at a queried time or interval, a challenge intensified by overlapping speech and source motion. To study this problem, we introduce Sonic-Bench, comprising 2.8K first-order Ambisonics clips and 28.8K target-specific question-answer pairs across 480 simulated indoor scenes. The benchmark covers localization, distance estimation, and motion reasoning in single- and dual-source settings. We further propose SonicWorld, which integrates physical, learned acoustic, and semantic features into persistent source states through learned transitions and recurrent updates, and projects them into an Omni LLM for spatial reasoning. Compared with Spatial-Omni on the Sonic-Bench test split, SonicWorld improves direction accuracy from 31.98% to 34.69% and reduces azimuth error from 57.22° to 48.01°. In zero-shot transfer to Spatial LibriSpeech, SonicWorld retains strong localization performance, achieving 73.98% direction accuracy and 19.29° azimuth error, compared with 27.14% and 70.74° for Spatial-Omni.
Figures & tables
| Single source | Dual source | |||||||||
| Model | Dir. | Az. | El. | Dist. | Mot. | Dir. | Az. | El. | Dist. | Mot. |
| Closed-source models | ||||||||||
| Gemini 3 Flash | 12.81 | 88.55 | 4.40 | 1.11 | 7.81 | 15.47 | 88.23 | 4.49 | 2.71 | 8.28 |
| Gemini 3.1 Pro | 9.06 | 90.87 | 15.78 | 1.86 | 12.19 | 14.53 | 90.32 | 28.88 | 2.03 | 14.38 |
| Open-source models | ||||||||||
| Qwen2.5-Omni-7B | 12.50 | 91.18 | 84.91 | 98.83 | 51.88 | 12.66 | 88.33 | 83.58 | 34.30 | 48.59 |
| SpatialSceneQA | SLS | |||||
| Model | Az. | El. | Dir. | Az. | El. | Dist. |
| Open-source models | ||||||
| Qwen2.5-Omni-7B | 60.47 | 80.14 | 12.69 | 90.26 | 88.47 | 55.04 |
| InternOmni | 80.54 | 78.05 | 12.38 | 89.69 | 57.59 | 98.83 |
| Phi-4-MM-5.6B | 61.86 | 18.31 | 12.61 | 89.43 | 7.08 | 1.99 |
| Audio Flamingo 3 | 132.66 | 24.92 | 11.87 | 89.92 | 28.08 | 33.40 |
| Physical | Neural | BEATs | BiGRU | Dir. | Mot. | Az20 | Az. | El. | Dist. |
| 4 | 8 | 8 | 4 | 28.33 | 52.40 | 25.73 | 55.68 | 0.97 | 0.62 |
| 4 | 4 | 8 | 4 | 28.65 | 52.19 | 26.88 | 55.59 | 0.99 | 0.62 |
| 4 | 4 | 4 | 8 | 32.19 | 51.98 | 33.13 | 48.11 | 0.96 | 0.63 |
| 4 | 4 | 4 | 4 | 34.69 | 53.44 | 34.90 | 48.01 | 0.92 | 0.59 |
| Variant | Dir. | Mot. | Az20 | Az. | El. | Dist. |
| w/o transition MLP | 33.44 | 51.46 | 32.71 | 49.94 | 0.96 | 0.64 |
| w/o prior-state supervision | 24.48 | 51.67 | 22.19 | 61.25 | 1.05 | 0.65 |
| w/o masked-state loss | 33.02 | 52.81 | 30.21 | 50.66 | 0.98 | 0.64 |
| w/o persistent source slots | 33.65 | 52.81 | 25.94 | 54.13 | 1.09 | 0.68 |
| Full model | 34.69 | 53.44 | 34.90 | 48.01 | 0.92 | 0.59 |
Appendix figures & tables28 assets
Supplementary material from the paper’s appendix.
Appendix
| Unit | Train | Test | Total |
| Scenes | 400 | 80 | 480 |
| Single-source clips | 800 | 160 | 960 |
| Dual-source clips | 1,600 | 320 | 1,920 |
| Source trajectories | 4,000 | 800 | 4,800 |
| Question–answer pairs | 24,000 | 4,800 | 28,800 |
| Audio duration | 6 h 40 min | 1 h 20 min | 8 h |
| Resource | Format | Scale | Motion | Spatial supervision | Queries |
| SpatialSoundQA [ 7 ] | Binaural | 872K QA | Static | Static source geometry | Source attributes / relations |
| SpatialSceneQA [ 9 ] | FOA, RGB-D | 61K samples | Static | Static 3D poses | Source identity / visual grounding |
| SAVVY-Bench [ 12 ] | 7-ch audio, video | 1,522 QA | Static / Moving | Camera-derived speaker tracks | Sound event-anchored |
| STAR-Bench [ 5 ] | Binaural | 2,353 QA | Static / Moving | Motion / trajectory QA labels | Motion-pattern reasoning |
| SO-Bench [ 10 ] | FOA | 7K QA | Static / Moving | Linked SO-Dataset tracks | Task-dependent source / time |
| Sonic-Bench (ours) | FOA | 28.8K QA | Static / Moving | Full-clip source tracks | Target- and time-specific |
| Parameter | Setting |
| Duration / sample rate | 10 s / 16 kHz |
| Geometric sampling | 12 Hz; 121 frames (including endpoints) |
| Receiver height | 1.50 m above NavMesh |
| Source height | 1.55 m above NavMesh |
| Source–receiver range | 1.5–4 m along the path |
| Moving-path length | 2–5 m |
| Task | Example question | Answer |
| Position direction | At 3.00 s, which direction is the female voice coming from? | Front |
| Azimuth | At 6.00 s, what is the azimuth of the female voice? | |
| Elevation | At 4.50 s, what is the vertical angle of the male voice? | |
| Distance | At 7.00 s, how far away is the male voice? | m |
| Movement direction | Between 1.50 and 2.50 s, which direction did the female source move relative to the listener’s fixed orientation? | Front-Right |
| Quantity | Statistic | Train | Test |
| Source range (m) | Minimum | 1.50 | 1.50 |
| Maximum | 4.00 | 4.00 | |
| Mean | 2.85 | 2.82 | |
| Median | 2.90 | 2.87 | |
| Elevation ( ∘ ) | Minimum | -16.76 | -13.45 |
| Maximum | 19.94 | 20.30 |
| Subset | S | F | FL | L | BL | B | BR | R | FR | Total |
| Single | 160 | 15 | 30 | 14 | 23 | 23 | 21 | 13 | 21 | 320 |
| Dual | 320 | 55 | 34 | 42 | 37 | 46 | 38 | 40 | 28 | 640 |
| Total | 480 | 70 | 64 | 56 | 60 | 69 | 59 | 53 | 49 | 960 |
| Layer | Channels | Kernel | Stride |
| 1 | 10 | 5 | |
| 2 | 7 | 4 | |
| 3 | 5 | 4 | |
| 4 | 5 | 4 |
| Input | Dimensions |
| Prior, observation, recurrent state, refined state, observation minus prior | |
| Direction anchor; predicted direction and distance | |
| Source-presence probability; positional log-variance | |
| Auxiliary features | 6 |
| Total | 1,295 |
| Module | Linear widths |
| Frame encoder | |
| Local temporal compressor | |
| Language-space projector |
| Objective | Weights |
| Trajectory supervision | Direction 1; log-distance 1; presence 1; uncertainty 0.05; prior 0.20 |
| Masked center prediction | Direction 1; log-distance 0.3; presence 0.2 |
| Projector training | Answer cross-entropy 1; geometric alignment 0.5 |
| Projector + LoRA | Answer cross-entropy 1; geometric alignment 0.25 |
| Representation | Shape |
| FOA waveform, | |
| Physical input features | |
| Physical tokens | |
| Neural tokens / semantic tokens | |
| Combined acoustic tokens | |
| Slot–token weights |
| Stage | Epochs | Accum. | Updates | Trainable modules and peak learning rate |
| Source states | 25 | 2 | 30,000 | Physical and neural branches, semantic pooling, recurrent filter, bidirectional GRU, state heads: |
| Projector | 5 | 8 | 15,000 | Projector and alignment heads: |
| Projector + LoRA | 3 | 8 | 9,000 | Projector : ; LoRA: |
| GPUs | NVIDIA RTX 4090; NVIDIA RTX 3090 | |||
| Backbone | Dir. | Az. | El. | Dist. | Mot. |
| Single source | |||||
| Qwen2.5-Omni-7B | 50.00 ( ) | 30.85 ( ) | 0.91 ( ) | 0.56 ( ) | 52.50 ( ) |
| InternOmni | 51.88 ( ) | 36.49 ( ) | 0.92 ( ) | 0.61 ( ) | 55.94 ( ) |
| Phi-4-MM-5.6B | 48.75 ( ) | 32.03 ( ) | 0.92 ( ) | 0.53 ( ) | 53.44 ( ) |
| Dual source | |||||
| Qwen2.5-Omni-7B | 27.03 ( ) | 56.59 ( ) | 0.93 ( ) | 0.60 ( ) | 53.91 ( ) |
| SpatialSceneQA | SLS | |||||
| Backbone | Az. | El. | Dir. | Az. | El. | Dist. |
| Qwen2.5-Omni-7B | 21.35 ( ) | 20.51 ( ) | 73.98 ( ) | 19.29 ( ) | 7.09 ( ) | 0.89 ( ) |
| InternOmni | 22.57 ( ) | 20.58 ( ) | 66.80 ( ) | 22.20 ( ) | 7.08 ( ) | 1.24 ( ) |
| Phi-4-MM-5.6B | 19.79 ( ) | 20.34 ( ) | 73.44 ( ) | 18.22 ( ) | 7.95 ( ) | 0.81 ( ) |
| Single source | Dual source | |||||
| Model | Az20 | El1 | D0.5 | Az20 | El1 | D0.5 |
| Closed-source models | ||||||
| Gemini 3 Flash | 10.00 | 30.31 | 25.94 | 11.72 | 27.34 | 26.41 |
| Gemini 3.1 Pro | 12.19 | 31.88 | 1.88 | 12.03 | 18.59 | 6.25 |
| Open-source models | ||||||
| Qwen2.5-Omni-7B | 14.06 | 1.88 | 0.00 | 11.72 | 1.09 | 0.00 |
| Sonic- Bench | Spatial SceneQA | SLS | |
| Evaluated clips | 480 | 622 | 8,512 |
| Single-source clips | 160 | 247 | 8,512 |
| Dual-source clips | 320 | 375 | 0 |
| Motion state | Stationary / Moving | Stationary | Stationary |
| Questions per evaluated clip | 10 | 2 | 4 |
| Single-source questions | 1,600 | 494 | 34,048 |
| Single source | Dual source | |||||||
| Model | Az. | El. | Az20 | El1 | Az. | El. | Az20 | El1 |
| Open-source audio models | ||||||||
| Qwen2.5-Omni-7B | 20.54 | 100.49 | 53.85 | 0.00 | 86.77 | 66.73 | 23.20 | 0.00 |
| InternOmni | 75.35 | 83.90 | 5.26 | 0.00 | 83.89 | 74.26 | 4.53 | 0.00 |
| Phi-4-MM-5.6B | 78.49 | 16.42 | 10.93 | 0.40 | 47.69 | 21.14 | 30.13 | 0.00 |
| Audio Flamingo 3 | 117.98 | 27.04 | 14.17 | 0.40 | 145.38 | 22.64 | 3.47 | 0.00 |
| Model | Dir. | Az. | El. | Dist. |
| Open-source models | ||||
| Qwen2.5-Omni-7B | 12.01 | 90.68 | 14.40 | 19.61 |
| InternOmni | 12.76 | 89.19 | 54.55 | 7.12 |
| Phi-4-MM-5.6B | 12.48 | 90.42 | 7.04 | 0.96 |
| Audio Flamingo 3 | 12.31 | 89.46 | 16.39 | 3.11 |
| Kimi-Audio-7B-Instruct | 11.83 | 90.81 | 8.59 | 2.81 |
| Variant | ADE | ID retention |
| Stationary interferer | ||
| w/o transition MLP | 0.82 | 0.00 |
| w/o masked-state loss | 0.66 | 9.09 |
| Full SonicWorld | 0.52 | 30.30 |
| Moving interferer | ||
| w/o transition MLP | 0.33 | 14.81 |