atlas

Research map / Large language models / Agents, code and math

BAT: Learning to Reason about Spatial Sounds with Large Language Models

ICML 2024 · Large language models › Agents, code and math · approach: Long-horizon LLM agents · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Builds BAT, an LLM (LLaMA-2 7B with LLaMA-adapter v2) that takes binaural spatial audio via a new Spatial-AST encoder and answers spatial sound perception and reasoning questions, trained on a simulated SpatialSoundQA dataset.

Problem

LLMs cannot perceive or reason about in-the-wild spatial (binaural, reverberant 3-D) audio, such as sound event identity, direction, distance, and relations between multiple sources.

Approach

Fuse a binaural spatial audio encoder (Spatial-AST, using mel-spectrograms plus interaural phase difference and three [CLS] tokens for class, distance, and direction) with LLaMA-2 7B via a projection module, trained with a perception-to-reasoning curriculum on simulated SoundSpaces 2.0 QA data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use the three-stage perception-to-reasoning curriculum instead of one-stage training (ablation)Bat · SpatialSoundQA Type EAverage binary accuracy: 73.55 -> 76.89Helped
Remove question types C and D from one-stage training (ablation)Bat · SpatialSoundQA Type EAverage binary accuracy: 73.55 -> 54.68Hurt
Train on source-separation tasks before the final reasoning stage (ablation)Bat · SpatialSoundQA Type EAverage binary accuracy: 54.68 -> 71.93Helped
Use binaural rather than monaural audio (ablation)Bat · SpatialSoundQA Type EAverage binary accuracy: 54.53 -> 76.89Helped
Include both CLS and audio tokens rather than CLS tokens only (ablation)One-stage Bat · SpatialSoundQA Type EAverage binary accuracy: 66.62 -> 73.55Helped
Include both CLS and audio tokens rather than audio tokens only (ablation)One-stage Bat · SpatialSoundQA Type EAverage binary accuracy: 70.33 -> 73.55Helped
Increase learnable query size from 32 to 64 (ablation)Bat · SpatialSoundQA Type EAverage binary accuracy: 72.45 -> 76.89Helped
Increase the detection loss weight lambda_1 (ablation)Spatial-AST · Spatial-AST encodermAP and MAEMixed
Remove reverberation (ablation)Spatial-AST · Monaural encodermAP: 51.13% -> 52.64%Helped
Add audio normalization (ablation)Spatial-AST · Monaural encodermAP: 51.13% -> 51.39%Helped
Add audio normalization (ablation)Spatial-AST · Monaural encoderDER: 39.57% -> 26.87%Helped
Include IPD features (ablation)Spatial-AST · Binaural single-task detectionmAP: 51.70 -> 52.26Helped

Surprising findings

Limitations

Experimental setup

Datasets: AudioSet SoundSpaces 2.0 (simulated binaural RIRs over Matterport3D) SpatialSoundQA

Benchmarks: SpatialSoundQA

Metrics: mAP accuracy distance error rate mean angular error binary accuracy

Baselines: AudioMAE SELDnet Monaural BAT (LTU-like) Random

Builds on: LLaMA-adapter v2 AudioMAE SoundSpaces 2.0 LTU

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy