Flow Matching-Based Speech Source Separation with Best-of-N Biometric Sampling

Anastasia Zorkina, Alexandr Anikin, Nikita Khmelev, Anastasiya Korenevskaya, Sergey Novoselov, Vladimir Volokhov, Maxim Korenevsky, Yuriy Matveev
ITMO University and Speech Technology Center Ltd. ICML 2026 Workshop on Machine Learning for Audio
Training pipeline of the proposed conditional flow matching demixer

Speaker embeddings define a deterministic source order during training and drive Best-of-N candidate selection and cross-chunk channel alignment at inference time.

Abstract

Single-channel speech separation remains challenging for real-world deployment due to source permutation ambiguity, sampling variability of generative models, and the difficulty of processing long recordings with chunk-wise inference. We address these issues with a conditional flow-matching-based method that produces an ordered two-source output conditioned on the mixture. A frozen speaker encoder defines the source order during training and is reused at inference for biometric best-of-N candidate selection and chunk-level channel alignment. We evaluate separation quality on Libri2Mix benchmark using SI-SDR, PESQ, and ESTOI, and measure downstream impact using cpWER for automatic speech recognition and EER for speaker verification. The results show that the proposed Transformer U-Net variant is competitive with strong baselines in objective separation metrics and achieves the lowest downstream automatic speech recognition and speaker verification error rates in all evaluated settings.

Method Overview

Results

Objective separation metrics on Libri2Mix. Values are clean / both.
Model SI-SDR PESQ ESTOI
ConvUnet (ours)14.42 / 9.282.64 / 1.650.90 / 0.75
TUnet (ours)17.30 / 11.253.11 / 1.920.93 / 0.79
DiffSep9.60 / -2.58 / -0.78 / -
SepReformer (chunk)11.30 / -2.45 / -0.88 / -
SepReformer (full)19.22 / 13.703.02 / 2.140.92 / 0.83
MeanFlow-TSE (full)17.56 / 11.683.27 / 2.180.91 / 0.80
Downstream ASR and speaker verification on Libri2Mix clean.
Model Whisper V3 cpWER Parakeet cpWER ResNet EER Wav2Vec 2.0 EER DistilWhisper EER
ConvUnet (ours)4.9925.513.170.511.28
TUnet (ours)3.8421.312.810.391.05
MeanFlow-TSE (full)9.0524.844.612.943.25
SepReformer (full)5.02-2.920.721.29
SepReformer (chunk)12.90-3.761.451.97

Audio Results

Libri2Mix

VoxConverse

Real Recording

BibTeX

@inproceedings{zorkina2026flowmatchingseparation,
  title     = {Flow Matching-Based Speech Source Separation with Best-of-N Biometric Sampling},
  author    = {Zorkina, Anastasia and Anikin, Alexandr and Khmelev, Nikita and Korenevskaya, Anastasiya and Novoselov, Sergey and Volokhov, Vladimir and Korenevsky, Maxim and Matveev, Yuriy},
  booktitle = {ICML 2026 Workshop on Machine Learning for Audio},
  year      = {2026},
  url       = {http://arxiv.org/abs/2607.06088}
}