X-Pred MeanFlow

for Streaming Token-to-Mel Speech Decoding

Anonymous to ISCSLP

Arxiv | GitHub Repo

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

X-Pred MeanFlow Architecture 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 θ and analytically recovers the mean velocity as ûθ = (ztθ) / t. The resulting mel-spectrogram 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.