Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LLM-Codec: Neural Audio Codec Meets Language Model Objectives

About

Neural audio codecs are widely used as tokenizers for spoken language models, but they are optimized for waveform reconstruction rather than autoregressive prediction. This mismatch injects acoustically driven uncertainty into the discrete token space and increases language-model perplexity. We propose \ours, which augments codec training with language-model-facing objectives while keeping both codec and LLM architectures unchanged. \ours introduces (i) future token prediction with Medusa-style multi-step heads to encourage multi-step predictability, and (ii) semantic alignment that matches audio and text representations via a memory-bank contrastive loss. A differentiable Gumbel bridge enables end-to-end gradients from these objectives to the codec encoder. On SALMon speech coherence, token LMs trained on \ours reach 61.6% accuracy (+12.1 points over AUV) while reducing perplexity 35. On Codec-SUPERB-tiny, \ours improves speech Mel distance by 5.0% over AUV while simultaneously achieving the learnability gains, demonstrating that reconstruction fidelity and token predictability can be improved together.

Ho-Lam Chung, Yiming Chen, Hung-yi Lee• 2026

Related benchmarks

TaskDatasetResultRank
Acoustic ConsistencySALMon
Speaker Consistency63
66
Audio ReconstructionCodec-SUPERB tiny (Speech)
Mel1.836
14
Audio ReconstructionCodec-SUPERB tiny (Overall)
Mel1.229
7
Audio ReconstructionCodec-SUPERB Music tiny
Mel1.126
7
Token-level PredictabilityLibriSpeech 100
Eval Loss8.44
5
Showing 5 of 5 rows

Other info

Follow for update