提出VITA-Audio模型,实现语音生成首帧零延迟,提升实时对话效率。
VITA-Audio: Fast Interleaved Cross-Modal Token Generation for Efficient Large Speech-Language Model
- 通过多模态令牌预测模块,单次前向传播生成多个音频标记。
- 7B参数模型推理速度提升3~5倍,首帧生成延迟极低。
- 适合需要实时语音交互的应用,如智能助手和在线客服。
随着人机自然交互需求的增长,基于语音的系统受到越来越多关注,因为语音是日常交流中最常见的形式之一。然而,现有语音模型在流式场景下生成首个音频标记时仍存在高延迟,成为部署的主要瓶颈。为此,我们提出VITA-Audio,一种端到端的大规模语音模型,可实现快速音视频标记生成。具体而言,引入轻量级多重跨模态令牌预测(MCTP)模块,在一次模型前向传播中高效生成多个音频标记,不仅加速推理,还显著降低流式场景中首个音频生成的延迟。此外,采用四阶段渐进式训练策略,在最小损失语音质量的前提下实现模型加速。据我们所知,VITA-Audio是首个能在首次前向传播中生成音频输出的多模态大语言模型,支持真正意义上的实时对话能力。该模型完全可复现,仅使用开源数据训练。实验表明,7B参数规模下,推理速度提升3~5倍,并在自动语音识别(ASR)、文本转语音(TTS)及语音问答(SQA)等多个基准上显著优于同规模开源模型。
原文摘要 · Abstract (English)
With the growing requirement for natural human-computer interaction, speech-based systems receive increasing attention as speech is one of the most common forms of daily communication. However, the existing speech models still experience high latency when generating the first audio token during streaming, which poses a significant bottleneck for deployment. To address this issue, we propose VITA-Audio, an end-to-end large speech model with fast audio-text token generation. Specifically, we introduce a lightweight Multiple Cross-modal Token Prediction (MCTP) module that efficiently generates multiple audio tokens within a single model forward pass, which not only accelerates the inference but also significantly reduces the latency for generating the first audio in streaming scenarios. In addition, a four-stage progressive training strategy is explored to achieve model acceleration with minimal loss of speech quality. To our knowledge, VITA-Audio is the first multi-modal large language model capable of generating audio output during the first forward pass, enabling real-time conversational capabilities with minimal latency. VITA-Audio is fully reproducible and is trained on open-source data only. Experimental results demonstrate that our model achieves an inference speedup of 3~5x at the 7B parameter scale, but also significantly outperforms open-source models of similar model size on multiple benchmarks for automatic speech recognition (ASR), text-to-speech (TTS), and spoken question answering (SQA) tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。