StreamFlow让语音生成实时流畅,180毫秒首包延迟
StreamFlow: Streaming Flow Matching with Block-wise Guided Attention Mask for Speech Token Decoding
- 分块注意力掩码实现局部感知,突破长序列依赖瓶颈
- 语音质量媲美非流式方法,首包延迟仅180毫秒
- 适合低延迟语音交互场景,如语音助手、在线会议
基于离散标记的语音生成近年来强调了标记到波形生成对音质的重要性,尤其在实时交互中。传统融合语义标记与流匹配(FM)的框架因依赖全局感受野而难以实现流式处理。直接逐标记流式生成常导致音质下降。为此,我们提出 StreamFlow,一种新的神经架构,支持扩散变换器(DiT)的流式流匹配。为缓解长序列外推问题,设计局部分块感受野策略:将序列分块,引入分块注意力掩码,使当前块可获取前或后块信息,并在不同DiT层间分层组合,调控感受野。主观与客观实验表明,该方法性能接近非流式方法,优于其他流式方法,且在长序列生成中有效控制推理时间。此外,本方法实现仅180毫秒的显著首包延迟。
原文摘要 · Abstract (English)
Recent advancements in discrete token-based speech generation have highlighted the importance of token-to-waveform generation for audio quality, particularly in real-time interactions. Traditional frameworks integrating semantic tokens with flow matching (FM) struggle with streaming capabilities due to their reliance on a global receptive field. Additionally, directly implementing token-by-token streaming speech generation often results in degraded audio quality. To address these challenges, we propose StreamFlow, a novel neural architecture that facilitates streaming flow matching with diffusion transformers (DiT). To mitigate the long-sequence extrapolation issues arising from lengthy historical dependencies, we design a local block-wise receptive field strategy. Specifically, the sequence is first segmented into blocks, and we introduce block-wise attention masks that enable the current block to receive information from the previous or subsequent block. These attention masks are combined hierarchically across different DiT-blocks to regulate the receptive field of DiTs. Both subjective and objective experimental results demonstrate that our approach achieves performance comparable to non-streaming methods while surpassing other streaming methods in terms of speech quality, all the while effectively managing inference time during long-sequence generation. Furthermore, our method achieves a notable first-packet latency of only 180 ms.\footnote{Speech samples: https://dukguo.github.io/StreamFlow/}
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。