Abstract Recent advancements in discrete token-based speech generation have highlighted the importance of efficient token-to-waveform synthesis in streaming and dialogue scenarios. Flow-matching acoustic decoders achieve high-quality token-to-mel generation, but their iterative sampling requires multiple neural function evaluations, limiting low-latency speech synthesis. MeanFlow reduces the sampling budget by modeling the average velocity over a temporal interval, yet maintaining high acoustic quality under extremely few-step token-to-mel generation remains challenging. To address this challenge, we propose X-Pred MeanFlow, a few-step streaming token-to-mel decoder that reparameterizes MeanFlow with mel-space prediction. The decoder predicts a generalized mel field and analytically derives the corresponding average velocity for sampling, thereby preserving the MeanFlow formulation while providing a direct acoustic prediction target. We further introduce layer-selective block-wise attention to enable continuous chunk-wise generation with bounded context. Experiments show that X-Pred MeanFlow improves few-step token-to-mel synthesis over Direct-u MeanFlow and supports stable streaming generation. Speech samples are available below.
Contents
This page is for research demonstration purposes only.
Model Overview
Figure 1. Overall framework and few-step inference process of the proposed X-Pred MeanFlow acoustic decoder. A frozen text-to-token LLM generates discrete speech tokens c, which condition the trainable token-to-mel decoder. At each sampling step, the decoder directly predicts the clean mel-spectrogram x̂θ and analytically recovers the mean velocity as ûθ = (zt − x̂θ) / t. The resulting mel-spectrogram x̂ is converted into waveform ŷ by a frozen HiFi-GAN vocoder. The snowflake symbol denotes frozen components.
Reconstruction Quality Comparison
Origin (Ground Truth) · Standard MeanFlow (Baseline) · X-Pred MeanFlow (Ours)
All reconstructions use the same DiT backbone and HiFi-GAN vocoder. X-Pred MeanFlow consistently produces higher-fidelity reconstructions with better speaker similarity and naturalness compared to the standard MeanFlow baseline. We recommend using headphones for evaluation.
| # | Speaker | Lang. & Gender | Origin (Ground Truth) |
Standard MeanFlow (Baseline) |
X-Pred MeanFlow (Ours) |
|---|---|---|---|---|---|
| 1 | AISHELL3 | CN · Female | |||
| 2 | DailyTalk | CN · Male | |||
| 3 | BC13 | CN | |||
| 4 | DB4 | CN · Female | |||
| 5 | AQY 2-Speaker | CN | |||
| 6 | HiFi-TTS | EN | |||
| 7 | VCTK | EN | |||
| 8 | LJSpeech | EN · Female |
Samples are from the Emilia corpus evaluation set, covering unseen speakers across multiple Chinese and English datasets. The same frozen HiFi-GAN vocoder is used for all systems.