Accelerating High-Fidelity Waveform Generation via Adversarial Flow Matching Optimization
About
This paper introduces PeriodWave-Turbo, a high-fidelity and high-efficient waveform generation model via adversarial flow matching optimization. Recently, conditional flow matching (CFM) generative models have been successfully adopted for waveform generation tasks, leveraging a single vector field estimation objective for training. Although these models can generate high-fidelity waveform signals, they require significantly more ODE steps compared to GAN-based models, which only need a single generation step. Additionally, the generated samples often lack high-frequency information due to noisy vector field estimation, which fails to ensure high-frequency reproduction. To address this limitation, we enhance pre-trained CFM-based generative models by incorporating a fixed-step generator modification. We utilized reconstruction losses and adversarial feedback to accelerate high-fidelity waveform generation. Through adversarial flow matching optimization, it only requires 1,000 steps of fine-tuning to achieve state-of-the-art performance across various objective metrics. Moreover, we significantly reduce inference speed from 16 steps to 2 or 4 steps. Additionally, by scaling up the backbone of PeriodWave from 29M to 70M parameters for improved generalization, PeriodWave-Turbo achieves unprecedented performance, with a perceptual evaluation of speech quality (PESQ) score of 4.454 on the LibriTTS dataset. Audio samples, source code and checkpoints will be available at https://github.com/sh-lee-prml/PeriodWave.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Audio generation from Encodec tokens | universal evaluation dataset | PESQ3.579 | 27 | |
| Waveform Generation | MUSDB18 out-of-distribution vocal samples HQ (test) | M-STFT0.8901 | 19 | |
| Speech Synthesis | LibriTTS (test) | -- | 17 | |
| Text-to-Speech | LibriSpeech clean PC (test) | -- | 17 | |
| Text-to-Speech | LibriTTS zero-shot | UTMOS4.3026 | 14 | |
| Waveform Generation | LibriTTS 24,000 Hz (test) | UTMOS3.7229 | 13 | |
| Waveform Generation | LJSpeech | UTMOS4.3894 | 12 | |
| Waveform Generation | LibriTTS (dev) | M-STFT1.1351 | 12 | |
| Zero-shot Text-to-Speech | LibriSpeech PC clean (test) | WER1.8 | 12 | |
| Speech Synthesis | LibriTTS 24,000 Hz (test) | MOS4.23 | 11 |