HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Unsupervised Source Separation via Bayesian Inference in the Latent Domain

Michele Mancusi Emilian Postolache Giorgio Mariani Marco Fumero Andrea Santilli Luca Cosmo Emanuele Rodolà

Unsupervised Source Separation via Bayesian Inference in the Latent Domain

Abstract

State of the art audio source separation models rely on supervised data-driven approaches, which can be expensive in terms of labeling resources. On the other hand, approaches for training these models without any direct supervision are typically high-demanding in terms of memory and time requirements, and remain impractical to be used at inference time. We aim to tackle these limitations by proposing a simple yet effective unsupervised separation algorithm, which operates directly on a latent representation of time-domain signals. Our algorithm relies on deep Bayesian priors in the form of pre-trained autoregressive networks to model the probability distributions of each source. We leverage the low cardinality of the discrete latent space, trained with a novel loss term imposing a precise arithmetic structure on it, to perform exact Bayesian inference without relying on an approximation strategy. We validate our approach on the Slakh dataset arXiv:1909.08494, demonstrating results in line with state of the art supervised approaches while requiring fewer resources with respect to other unsupervised methods.

Code Repositories

michelemancusi/LQVAE-separation
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
music-source-separation-on-slakh2100LQ-VAE + Scalable Transformer
SDR (bass): 7.42
SDR (drums): 5.83

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Unsupervised Source Separation via Bayesian Inference in the Latent Domain | Papers | HyperAI