VITA-Audio: Fast Interleaved Audio-Text Token Generation for Efficient Large Speech-Language Model

Chaoyou Fu (Nanjing University) · Yunhang Shen (Xiamen University) · Haoyu Cao (Tencent Youtu Lab) · Zuwei Long (Tencent Youtu Lab) · Heting Gao (University of Illinois at Urbana-Champaign) · Ke Li (East China Normal University) · Rongrong Ji (Xiamen University, China) · Xing Sun (Tencent YouTu Lab) · Lijiang Li · Peixian Chen (Xiamen university) · Mengdan Zhang (Tencent Youtu Lab) · Hang Shao (Tencent Youtu Lab) · Jian Li (Tencent) · Jinlong Peng (Tencent Youtu Lab)
audio token generationautomatic speech recognitionhigh latencyinference accelerationinference speedupmodel forward passmulti-modal large language modelmultiple cross-modal token predictionnatural human-computer interactionopen-source dataprogressive training strategyreal-time conversational capabilitiesreproducibilityspeech-based systemsvita-audio

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.