CoSTA: Compressed Speech Token Adaptation for Efficient Spoken Dialogue Models
AuthorsYingzhuo Deng, Chong Li, Bingxin Liu, Yi Shu, Tianyu Peng, Can Xie, Siyao Song, Shuo Ren, Jiajun Zhang
AffiliationsInstitute of Automation, Chinese Academy of Sciences · School of Artificial Intelligence, University of Chinese Academy of Sciences · Wuhan AI Researchdengyingzhuo2024@ia.ac.cn · jjzhang@nlpr.ia.ac.cn
CoSTA helps spoken dialogue systems generate speech faster by using compressed tokens internally and expanding them before synthesis.
Key results
Expanded speech-token generation speed relative to Step-Audio 2 mini.
Expanded speech-token generation speed relative to Step-Audio 2 mini.
CoSTA-ZH generated-token rate on the main benchmarks.
CoSTA-EN generated-token rate on the main benchmarks.
What the paper found
CoSTA tackles a bottleneck in spoken dialogue models: long autoregressive speech-token sequences. Instead of replacing the tokenizer and speech decoder, it uses byte-pair encoding to create shorter, losslessly expandable token sequences for generation, then deterministically restores the original tokens before synthesis. The method adapts an existing model through mean-initialized vocabulary extension, pronunciation training, and dialogue training; the speech decoder stays unchanged. On the 8B Step-Audio 2 mini, CoSTA reaches 1.736× expanded speech-token generation speed in Chinese and 1.659× in English, with generated-token compression rates of 66.81% and 70.17%, respectively. Tests across URO-Bench show lower response latency and higher throughput while maintaining or improving quality; comparison models include Qwen2.5-Omni, Kimi-Audio, and GLM-4-Voice. CoSTA also transfers to OpenS2S, which uses a GLM-4-Voice tokenizer, suggesting the approach can complement low-rate tokenizers rather than replace them. The results make token-space decoupling—not a new speech codec—the paper’s central contribution: models can generate compact representations while retaining their existing synthesis pipeline.
Original abstract
Spoken Dialogue Models (SDMs) typically share a speech-token space for generation and synthesis, leading to long autoregressive sequences and costly inference. Lower-rate speech tokenizers shorten sequences but may introduce compression-quality trade-offs and costly replacement without breaking this coupling. To address this limitation, we propose CoSTA, a compatibility-preserving post-hoc framework that decouples generation and synthesis token spaces while retaining the original speech tokenizer and decoder. CoSTA extends the generation token space with compressed representations via mean-initialized vocabulary extension, pronunciation adaptation, and dialogue adaptation, followed by deterministic expansion back to the original speech-token space before synthesis. On bilingual speech-dialogue benchmarks, CoSTA achieves approximately $1.7\times$ expanded speech-token generation speed, lower full-response latency, and higher throughput while preserving response quality. Analyses show more sustained interaction under limited budgets and a practical trade-off among compression, decoding efficiency, and quality. Further experiments demonstrate cross-architecture transfer and complementarity to low-rate tokenization.
Read the original paperMore in Speech AI
Browse all 39 papers →Lipi-Ghor: A Large-Scale, Diarized, Multi-Domain Bengali Speech Corpus and a Scalable Pipeline for Building It
Sanjid Hasan, A H M Fuad, Risalat Labib, Bayazid Hasan Simon
The paper introduces a large, multi-domain Bengali speech corpus with speaker diarization and a scalable recipe for building similar resources.
VoxLock: Target-Speaker Extraction for Voice Agents Without an Enrollment Step
Arbaz Siddiqui
VoxLock filters a live call to keep the caller’s speech and suppress nearby voices, without requiring a separate enrollment recording.
AnyEdit: A Unified Framework for Speech and Singing Voice Editing with Real-World Environmental Consistency
Yunjia Zhang, Junan Zhang, Jing Yang, Xueyao Zhang, Fan Fan, Zhizheng Wu
AnyEdit edits speech and singing while aiming to preserve the acoustic environment surrounding the original voice.