Continuous Audio Language Models

Your Image

Abstract

Audio Language Models (ALM) have emerged as the dominant paradigm for speech and music generation by representing audio as sequences of discrete tokens. Yet, unlike text tokens, which are invertible, audio tokens are extracted from lossy codecs with a limited bitrate. As a consequence, increasing audio quality requires generating more tokens, which imposes a trade-off between fidelity and computational cost. We address this issue by studying Continuous Audio Language Models (CALM). These models instantiate a large Transformer backbone that produces a contextual embedding at every timestep. This sequential information then conditions an MLP that generates the next continuous frame of an audio VAE through consistency modeling. By avoiding lossy compression, CALM achieves higher quality at lower computational cost than their discrete counterpart.

On this webpage, we show some results of our speech model as well as our music models. We illustrate as well the ablation study of the paper with some music samples.

Speech Language Model

This section presents speech samples generated using a 3-second prompt. Key details of the setup and results include:

  • CALM setting: Audio stream is composed of continuous latents predicted via 1-step consistency modeling.
  • RQ-Transformer setting: Audio stream is produced using an 8-RVQ Mimi Codec and predicted in parallel by an RQ-Transformer.
  • Performance: CALM outperforms RQ-Transformer on meaningfulness. We believe this may be due to the backbone allocating less capacity to audio manipulation, leaving more for text prediction in the CALM setting. As well, we can see that temperature has a huge impact for both models, validating our heuristic for temperature sampling for CALM.
  • Efficiency:
    • Sampling each latent from the consistency head is 12.3× faster than with the RQ-Transformer.
    • Generating 30 seconds of audio is overall 1.3× faster with CALM than with the baseline.
  • Prompt RQ-Transformer 8 RVQ
    temp=0.8 (baseline)
    CALM Consistency 1 Step
    temp=0.8
    CALM Consistency 1 Step
    temp=1.0
    RQ-Transformer 8 RVQ
    temp=1.0

    Music Generation

    We compare our music generation models, all of which use a backbone with 1.35B parameters (from MusicGen Medium):

    Prompt RQ-Transformer 32 RVQ (baseline) FAD: 1.06 CALM TrigFlow 100 steps FAD: 0.64 CALM Consistency 4 steps FAD: 0.71 CALM Consistency 1 step FAD: 0.83 Retrained MusicGen FAD: 1.72

    Ablation Study

    We illustrate here the ablation study of our paper in order to show the importance of each component of our model. We showcase it on Music Generation with CALM Consistency 4 steps. All the models have been trained 300k steps instead of 500k steps.

    Prompt Our Model Without Noise Aug., Short Context Transformer, Head Batch Mult. Without Short Context Transformer Without Noise Aug. Without Head Batch Mult.

    References