Machine Learning PapersLandmark papers
← All papers
RESEARCH PAPER / 2021

SoundStream: An End-to-End Neural Audio Codec

Neil Zeghidour, Alejandro Luebs, Ahmed Omran ...Full author list in the original paper.

Google Research · arxiv

Affiliations: Google Research

Abstract

We present SoundStream, a novel neural audio codec that can efficiently compress speech, music and general audio at bitrates normally targeted by speech-tailored codecs. SoundStream relies on a model architecture composed by a fully convolutional encoder/decoder network and a residual vector quantizer, which are trained jointly end-to-end. Training leverages recent advances in text-to-speech and speech enhancement, which combine adversarial and reconstruction losses to allow the generation of high-quality audio content from quantized embeddings. By training with structured dropout applied to quantizer layers, a single model can operate across variable bitrates from 3kbps to 18kbps, with a negligible quality loss when compared with models trained at fixed bitrates. In addition, the model is amenable to a low latency implementation, which supports streamable inference and runs in real time on a smartphone CPU. In subjective evaluations using audio at 24kHz sampling rate, SoundStream at 3kbps outperforms Opus at 12kbps and approaches EVS at 9.6kbps. Moreover, we are able to perform joint compression and enhancement either at the encoder or at the decoder side with no additional latency, which we demonstrate through background noise suppression for speech.

The following analysis is generated by AI and may contain errors. It is separate from the authors’ abstract and is not peer review.

Research summary

SoundStream introduces a highly efficient, bitrate-scalable neural audio codec that combines a causal convolutional autoencoder with a residual vector quantizer and quantizer dropout to achieve state-of-the-art perceptual quality at low bitrates. The paper's rigorous methodology, extensive subjective and objective evaluations, and demonstration of real-time mobile deployment establish it as a foundational contribution to neural audio compression, while its discrete tokenization framework has profoundly shaped subsequent research in generative audio modeling and multimodal representation learning.

Methodology

The paper proposes a fully convolutional, causal encoder-decoder architecture paired with a Residual Vector Quantizer (RVQ) for end-to-end neural audio compression. The core methodological innovation is "quantizer dropout," a structured dropout technique applied to RVQ stages during training that enables a single model to dynamically operate across a wide bitrate range (3–18 kbps) without architectural modifications. Training leverages a multi-scale adversarial framework (waveform and STFT discriminators) combined with feature-matching and multi-scale spectral reconstruction losses, effectively balancing perceptual quality and signal fidelity. The joint compression-enhancement variant integrates FiLM layers conditioned on a binary flag, allowing flexible denoising at either the encoder or decoder bottleneck. The architectural choices (causal convolutions, specific stride sequences, EMA codebook updates, k-means initialization) are well-motivated and systematically justified.

Results & evaluation

The evaluation is rigorous and comprehensive. Subjective quality is assessed via a crowdsourced MUSHRA-style protocol across diverse content (clean speech, noisy speech, music, real-world recordings), demonstrating that SoundStream at 3 kbps perceptually matches or exceeds Opus at 12 kbps and approaches EVS at 9.6 kbps. Objective metrics (ViSQOL) are used for ablation and hyperparameter tuning, with strong correlation to subjective results. The paper thoroughly ablates critical design choices: learnable vs. fixed mel-spectrogram encoders, encoder/decoder capacity trade-offs, quantizer depth vs. codebook size, architectural latency impacts, and denoising placement. Real-time factor profiling on a Pixel 4 CPU confirms practical deployability. The experimental design is robust, statistically sound, and directly addresses real-world deployment constraints.

Reproducibility

The paper provides highly detailed architectural specifications, loss formulations, hyperparameter settings, and training procedures. The use of standard components (ELU activations, causal 1D convolutions, RVQ with EMA updates) and publicly available datasets (LibriTTS, MagnaTagATune, Freesound) facilitates replication. While explicit code is not linked in the provided text, the methodological transparency, combined with the public demo page containing audio samples, ensures high reproducibility for practitioners.

Limitations

The codec operates at a constant bitrate (CBR) and does not implement entropy coding in the main pipeline, leaving potential rate savings unrealized. The 24 kHz sampling rate limits high-frequency fidelity compared to 48 kHz professional codecs. Adversarial training introduces inherent instability risks and requires careful loss weighting. The joint enhancement capability is only demonstrated for background noise suppression, leaving other tasks (dereverberation, bandwidth extension, source separation) unexplored. Finally, the model's performance on highly dynamic or polyphonic music at the lowest bitrates still shows a noticeable gap compared to higher-bitrate traditional codecs.

Broader impact

SoundStream establishes a practical, high-performance baseline for neural audio compression that bridges the gap between research and real-world deployment. Its discrete latent representation paradigm, particularly the RVQ + quantizer dropout framework, directly influenced the development of discrete audio tokenization used in foundational generative audio models (e.g., AudioLM, MusicLM, VALL-E). The joint compression-enhancement capability demonstrates a pathway toward unified, low-latency audio processing pipelines for mobile and edge devices, with significant implications for telecommunications, streaming, and assistive audio technologies. SoundStream introduces a highly efficient, bitrate-scalable neural audio codec that combines a causal convolutional autoencoder with a residual vector quantizer and quantizer dropout to achieve state-of-the-art perceptual quality at low bitrates. The paper's rigorous methodology, extensive subjective and objective evaluations, and demonstration of real-time mobile deployment establish it as a foundational contribution to neural audio compression, while its discrete tokenization framework has profoundly shaped subsequent research in generative audio modeling and multimodal representation learning.

Curator note

Pioneered neural audio codec architecture (encoder + RVQ + adversarial training) that became the foundation for EnCodec, DAC, and Moshi.

Related later work

Author of this paper? Send a correction.