atlas

Research map / Image generation / GANs

FINALLY: fast and universal speech enhancement with studio-like quality

NeurIPS 2024 · Image generation › GANs · approach: GAN vocoders and audio synthesis · 6 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Revisits GANs for universal speech enhancement, arguing LS-GAN training seeks the mode of the clean-speech distribution, and builds FINALLY, a WavLM-perceptual-loss HiFi++ variant with MS-STFT discriminators that outputs 48 kHz speech in one forward pass.

Problem

Universal speech enhancement must restore speech from many simultaneous distortions (noise, reverberation, bandwidth limits, microphone artefacts) at high quality and fast inference, while diffusion-based methods are computationally expensive due to iterative sampling.

Approach

Theoretically shows that LS-GAN training with a deterministic generator drives the output toward the mode of the conditional clean-speech density, and proposes a WavLM convolutional-feature perceptual loss (LMOS, combined with L1 STFT loss) plus MS-STFT discriminators, with a WavLM encoder fed into HiFi++ and a WaveUNet upsampler to produce 48 kHz output.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use LMOS regression loss rather than no regression loss (ablation)FINALLY, stage 2 (small model) · VoxCeleb (real data)MOS: 3.71 \pm 0.08 -> 4.20 \pm 0.05Helped
Add WavLM encoder features as input (ablation)FINALLY · UNIVERSE validationMOS: 3.49 ± 0.08 -> 3.75 ± 0.07Helped
Add WavLM encoder features as input (ablation)FINALLY · UNIVERSE validationPhER: 0.27 ± 0.04 -> 0.21 ± 0.04Helped
Scale model by increasing the number of channels (ablation)FINALLY · VoxCeleb (real data)MOS: 4.21 ± 0.06 -> 4.30 ± 0.05Helped
Add Upsample WaveUNet and the 48 kHz third training stage (ablation)FINALLY-16 · VoxCeleb (real data)MOS: 4.30 ± 0.05 -> 4.59 ± 0.05Helped
Add human feedback loss during the third stage (ablation)FINALLY · VoxCeleb (real data)UTMOS: 3.78 ± 0.07 -> 4.05 ± 0.07Helped
Use LMOS rather than RecLoss for regression (ablation)Small speech enhancement model · VoxCeleb (real data)MOS: 4.07 \pm 0.06 -> 4.20 \pm 0.05Helped
Use LMOS rather than MS-STFT regression loss (ablation)Small speech enhancement model · Regression-loss ablationUTMOS: 2.54 ± 0.10 -> 3.43 ± 0.09Helped
Use LMOS rather than RecLoss (ablation)Small speech enhancement model · Regression-loss ablationUTMOS: 2.53 ± 0.10 -> 3.43 ± 0.09Helped
Use LMOS rather than L1Spec regression loss (ablation)Small speech enhancement model · Regression-loss ablationUTMOSHelped
Combine WavLM-conv feature loss with L1-STFT loss (ablation)HiFi-GAN vocoder · Neural vocodingMOS: 4.31 ± 0.08Helped

Surprising findings

Limitations

Experimental setup

Datasets: LibriTTS-R DAPS-clean DNS Challenge noises VoxCeleb1 UNIVERSE validation data VCTK-DEMAND LJSpeech Miipher test samples

Benchmarks: VoxCeleb1 real samples UNIVERSE validation set VCTK-DEMAND LibriTTS test-other (Miipher comparison)

Metrics: MOS UTMOS WV-MOS DNSMOS PhER WER PESQ STOI SI-SDR RTF

Baselines: UNIVERSE BBED STORM VoiceFixer DEMUCS HiFi-GAN-2 MetricGAN+ DB-AIAT Miipher

Builds on: HiFi++ HiFi-GAN MS-STFT discriminator WavLM LS-GAN

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy