AIAny
Icon for item

Learning Where to Split: Dynamic Band Splitting for Audio Source Separation

Adapts frequency-band boundaries to each audio mixture under a fixed band-token budget to improve source separation. Predicts content-conditioned boundary scores, selects bands with hard Top-K, pools variable-width bands into tokens, and decodes full-resolution complex masks—showing large SI-SDR gains on DnR v2, MUSDB18-HQ and EchoSet.

Introduction

Why this matters

Most band-split audio separators fix spectral partitions in advance, which wastes representational capacity when discriminative spectral structure varies across mixtures. The core insight of this work is that letting band boundaries move with the input—while keeping the overall token budget constant—lets the model allocate resolution where it matters and improves separation without increasing separator cost.

Key Findings
  • Content-adaptive partitions: The method predicts boundary scores inside coarse frequency regions and uses hard Top-K selection to produce exactly the prescribed number of contiguous bands per region, so token count stays fixed while locations and widths change with the mixture.
  • Efficiency-preserving pooling and decoding: Each variable-width band is pooled into a fixed-dimensional token (content-weighted pooling), passed to an existing separator backbone, then smoothed, expanded back to the STFT grid, frequency-refined, and decoded into complex masks by a frequency-shared head.
  • Strong empirical gains: DBS improved separation across multiple backbones and datasets; e.g., DBS-BSRNN achieved a relative 29.0% gain in average SI-SDR on DnR v2 (from 7.03 dB to 9.07 dB under a 128-band budget).
  • Practical design choices: Using several coarse regions (R=4) avoids concentrating boundaries at low frequencies; a straight-through estimator with a local soft assignment and ratio regularizer enables end-to-end training despite hard Top-K selection.
Who it helps and trade-offs

Great fit if you: work on single-channel or stereo audio source separation and need higher effective frequency resolution without increasing separator complexity; want a drop-in encoder-decoder style modification that can be paired with band-split backbones (e.g., BSRNN, TIGER).

Look elsewhere if: your pipeline cannot accommodate STFT-based mask decoding or you require strictly deterministic, fixed-band behavior for downstream interpretability. DBS also increases some runtime/compute (extra boundary prediction, smoothing, expansion) compared with the simplest static band-split encoder.

Where it fits

DBS sits between fixed band-split encoders (cheap but inflexible) and full-resolution, compute-heavy separators (high fidelity but costly). It preserves the token-budget advantages of band-split designs while dynamically reallocating spectral resolution per mixture, making it attractive for research and applied systems constrained by separator throughput.

Methodology highlights
  • Coarse-region scaffold: frequency axis is partitioned into R coarse regions; each region gets a prescribed band budget J_r and predicts boundary probabilities for candidate transitions inside the region.
  • Hard Top-K + straight-through estimator: top-scoring transitions form contiguous bands in the forward pass; a local soft assignment and regularizers are used in the backward pass so the boundary predictor receives learning signal.
  • Full-resolution reconstruction: probability-guided smoothing softens band transitions before expanding tokens to the STFT grid; a frequency-refinement stage and a shared-frequency head produce complex masks for waveform reconstruction.

DBS releases code and audio demos alongside the paper, enabling reproduction and interactive exploration of predicted partitions on real mixtures.

Information

  • Websitearxiv.org
  • OrganizationsDepartment of Computer Science and Technology, Institute for AI, BNRist, Tsinghua University, Beijing 100084, China, IDG/McGovern Institute for Brain Research, Tsinghua University, Beijing 100084, China, Chinese Institute for Brain Research (CIBR), Beijing 100010, China
  • AuthorsBingzhe Zhang, Kai Li, Xiaolin Hu
  • Published date2026/10/10

More Items

Evaluates multimodal memory in large audio language models by testing four acoustic evidence types and four memory operations across multi-session spoken histories, stratified across 8K–64K token context budgets.

Provides a unified benchmark and a spatially aware omni-modal model for spatial audio-visual reasoning in embodied agents — OmniEchoBench (197 scenes, 2,972 QA pairs; 900 FOA-based navigation tasks) plus OmniEcho, which adds an FOA spatial encoder to a pretrained semantic audio pathway to enable spatial QA and sound-guided navigation.

Generates long-form, text-controlled music with explicit arrangement and planning. Uses a 50 Hz single-codebook tokenizer, a flow-matching diffusion Transformer to predict VAE latents, and an MoE autoregressor with ABC‑CoT planning to produce 48 kHz audio up to 5m30s.