HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Separate And Diffuse: Using a Pretrained Diffusion Model for Improving Source Separation

Shahar Lutati; Eliya Nachmani; Lior Wolf

Separate And Diffuse: Using a Pretrained Diffusion Model for Improving Source Separation

Abstract

The problem of speech separation, also known as the cocktail party problem, refers to the task of isolating a single speech signal from a mixture of speech signals. Previous work on source separation derived an upper bound for the source separation task in the domain of human speech. This bound is derived for deterministic models. Recent advancements in generative models challenge this bound. We show how the upper bound can be generalized to the case of random generative models. Applying a diffusion model Vocoder that was pretrained to model single-speaker voices on the output of a deterministic separation model leads to state-of-the-art separation results. It is shown that this requires one to combine the output of the separation model with that of the diffusion model. In our method, a linear combination is performed, in the frequency domain, using weights that are inferred by a learned model. We show state-of-the-art results on 2, 3, 5, 10, and 20 speakers on multiple benchmarks. In particular, for two speakers, our method is able to surpass what was previously considered the upper performance bound.

Benchmarks

BenchmarkMethodologyMetrics
speech-separation-on-libri10mixSeparate And Diffuse
SI-SDRi: 9
speech-separation-on-libri20mixSeparate And Diffuse
SI-SDRi: 5.2
speech-separation-on-libri2mixSeparate And Diffuse
SI-SDRi: 21.5
speech-separation-on-libri5mixSeparate And Diffuse
SI-SDRi: 14.2
speech-separation-on-wsj0-2mixSeparate And Diffuse
SI-SDRi: 23.9
speech-separation-on-wsj0-3mixSeparate And Diffuse
SI-SDRi: 20.9

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