用扩散Transformer实现高效连续语音自回归生成
DiTAR: Diffusion Transformer Autoregressive Modeling for Speech Generation
- 分块生成+语言模型与扩散Transformer协同
- 零样本生成在自然度、说话人相似性上达顶尖水平
- 通过噪声引入时间点控制温度,平衡多样性与确定性
近期研究尝试结合扩散模型与自回归模型,直接生成连续语音表征而无需离散语音标记,但常面临计算负担过重或性能不佳的问题。本文提出扩散Transformer自回归建模(DiTAR),一种基于分块的自回归框架,融合语言模型与扩散Transformer。该方法显著提升自回归模型在连续标记上的效率并降低计算开销。DiTAR采用分而治之策略:语言模型处理聚合的分块嵌入,扩散Transformer根据其输出生成下一个分块。推理时,我们提出将温度定义为反向扩散常微分方程中引入噪声的时间点,以平衡多样性与确定性。广泛的扩展分析表明,DiTAR具有优异的可扩展性。在零样本语音生成任务中,DiTAR在鲁棒性、说话人相似性和自然度方面达到当前最优表现。
原文摘要 · Abstract (English)
Several recent studies have attempted to autoregressively generate continuous speech representations without discrete speech tokens by combining diffusion and autoregressive models, yet they often face challenges with excessive computational loads or suboptimal outcomes. In this work, we propose Diffusion Transformer Autoregressive Modeling (DiTAR), a patch-based autoregressive framework combining a language model with a diffusion transformer. This approach significantly enhances the efficacy of autoregressive models for continuous tokens and reduces computational demands. DiTAR utilizes a divide-and-conquer strategy for patch generation, where the language model processes aggregated patch embeddings and the diffusion transformer subsequently generates the next patch based on the output of the language model. For inference, we propose defining temperature as the time point of introducing noise during the reverse diffusion ODE to balance diversity and determinism. We also show in the extensive scaling analysis that DiTAR has superb scalability. In zero-shot speech generation, DiTAR achieves state-of-the-art performance in robustness, speaker similarity, and naturalness.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。