NTH

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

November 2, 2026 2 min read
Watch on YouTube
The one-line take

CoSTA helps spoken dialogue systems generate speech faster by using compressed tokens internally and expanding them before synthesis.

Key results

1.736
Chinese generation speedup

Expanded speech-token generation speed relative to Step-Audio 2 mini.

1.659
English generation speedup

Expanded speech-token generation speed relative to Step-Audio 2 mini.

66.81%
Chinese generated-token compression rate

CoSTA-ZH generated-token rate on the main benchmarks.

70.17%
English generated-token compression rate

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 paper

More in Speech AI

Browse all 39 papers →