One-Step Voice Conversion by Learning kNN Transport in WavLM Space

Anton Selitskiy and David Millard
Stony Brook University University of Rochester
Source speech WavLM frames x ∈ ℝT×1024
Conditional flow Cross-attention + FiLM vθ(yt, t, z)
Converted speech Vocoder output 1, 5, or 50 steps
Target-speaker utterance z

The model learns a velocity field between source WavLM embeddings and kNN-derived target endpoints while conditioning on a target-speaker utterance.

Abstract

Voice conversion (VC) systems fall into two families: non-parametric embedding-space methods, which need no trained model but degrade on short target utterances, and spectrogram-based neural architectures, which achieve strong quality via multi-module pipelines with tens of millions of parameters. We propose kNN-FM-VC, a single conditional flow-matching network that learns to approximate the kNN-VC mapping between WavLM embedding distributions of source and target speakers, replacing explicit pointwise kNN matching with a neural regressor trained on kNN-generated pairs. The model is conditioned on the target speaker via cross-attention and FiLM, and trained under three Gaussian conditional paths (Schrödinger bridge, straight line, and constant-variance Gaussian tube), enabling few-step sampling. Unlike Phoneme Hallucinator, which uses an upsampling stage followed by kNN matching, our 13M-parameter model performs conversion with a single learned network and supports one-step inference. On LibriSpeech, the one-step Gaussian Bridge achieves lower WER and higher estimated speech quality than FreeVC and Phoneme Hallucinator. Relative to kNN and kDOT, it substantially reduces WER. We encourage readers to listen to the output.

Method

kNN pseudo-pairs

Source and target WavLM frames are pooled across utterances. Each source frame is paired with the mean of its four nearest target frames under cosine similarity.

Target conditioning

Cross-attention uses the source sequence as queries and a target-speaker sequence as keys and values. A masked temporal mean of the target sequence is combined with time conditioning in the FiLM blocks.

Probability paths

Three conditional paths are evaluated at the same inference budgets: Schrödinger Bridge, Gaussian Bridge, and Gaussian Tube, each sampled with 1, 5, or 50 integration steps.

Audio comparisons

Each table uses the same source and target-speaker reference across methods. Select one of the held-out conversion pairs.

Loading listening set…

Quantitative results

Select a metric to compare the probability paths and baselines. Ranks compare the reported mean.

Word error rate

Lower is better
Proposed Discrete Learned baseline

WER and SIM labels show mean and 95% confidence interval. UTMOSv2 labels show mean ± sample standard deviation.