arXiv:2508.16790cs.SDcs.LG2025-08NeurIPS被引 17

用扩散模型实现端到端语音分词,帧率仅6.25赫兹却保持高音质。

TaDiCodec: Text-aware Diffusion Speech Tokenizer for Speech Language Modeling

  • 通过扩散自编码器实现单阶段端到端训练,无需预训练模型。
  • 在24kHz语音上达6.25Hz帧率、0.0875kbps比特率,性能优于主流方法。
  • 支持零样本文生语音,适合语音语言建模与高效生成任务。

语音分词器是语音语言模型的基础组件,但现有方法存在多重局限:依赖多层残差向量量化结构或高帧率、需借助预训练模型进行语义蒸馏、以及复杂的两阶段训练流程。本文提出文本感知扩散语音编解码器(TaDiCodec),通过扩散自编码器实现量化与重建的端到端优化,并在扩散解码器中融入文本引导,提升重建质量并实现最优压缩。该方法仅用单层码本即达成24kHz语音下6.25Hz的极低帧率和0.0875kbps的比特率,同时在词错误率(WER)、说话人相似度(SIM)和语音质量(UTMOS)等关键指标上表现优异。此外,TaDiCodec采用单阶段端到端训练,无需辅助预训练模型。我们验证了其在基于语言模型的零样本文生语音任务中,适用于自回归与掩码生成建模,展现出显著小的重建-生成差距。代码与模型权重已开源。

原文摘要 · Abstract (English)

Speech tokenizers serve as foundational components for speech language models, yet current designs exhibit several limitations, including: 1) dependence on multi-layer residual vector quantization structures or high frame rates, 2) reliance on auxiliary pre-trained models for semantic distillation, and 3) requirements for complex two-stage training processes. In this work, we introduce the Text-aware Diffusion Transformer Speech Codec (TaDiCodec), a novel approach designed to overcome these challenges. TaDiCodec employs end-to-end optimization for quantization and reconstruction through a diffusion autoencoder, while integrating text guidance into the diffusion decoder to enhance reconstruction quality and achieve optimal compression. TaDiCodec achieves an extremely low frame rate of 6.25 Hz and a corresponding bitrate of 0.0875 kbps with a single-layer codebook for 24 kHz speech, while maintaining superior performance on critical speech generation evaluation metrics such as Word Error Rate (WER), speaker similarity (SIM), and speech quality (UTMOS). Notably, TaDiCodec employs a single-stage, end-to-end training paradigm, and obviating the need for auxiliary pre-trained models. We also validate the compatibility of TaDiCodec in language model based zero-shot text-to-speech with both autoregressive modeling and masked generative modeling, demonstrating its effectiveness and efficiency for speech language modeling, as well as a significantly small reconstruction-generation gap. We will open source our code and model checkpoints. Audio samples are are available at https:/tadicodec.github.io/. We release code and model checkpoints at https:/github.com/HeCheng0625/Diffusion-Speech-Tokenizer.

语音生成扩散模型分词器零样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。