atlas

Research map / Large language models / Language, safety and interpretability

UniAudio 1.5: Large Language Model-Driven Audio Codec is A Few-Shot Audio Task Learner

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: Speech and audio language models · 3 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LLM-Codec, a semantic-guided multi-scale RVQ audio codec that maps audio into LLaMA 2's token vocabulary, enabling frozen LLaMA 2 7B to perform few-shot audio understanding and generation via in-context learning (UniAudio 1.5).

Problem

Frozen LLMs cannot directly handle audio tasks without fine-tuning; existing audio-LLMs need task-specific training data and do not support in-context learning of unseen audio tasks or audio generation.

Approach

Quantize audio with a codec whose first RVQ layer is initialized from LLaMA 2 word-level vocabulary embeddings (Oxford 5000 words) and trained with a semantic loss and a Whisper-based consistency loss, so audio becomes a "foreign language" in the LLM's token space that a frozen LLM can learn from a few demonstrations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use LLM-Codec instead of vanilla 3-layer RVQAudio codec reconstruction and sound event classification · VCTK reconstructionPESQ: 2.64 -> 2.55Hurt
use multi-scale 3-layer RVQ instead of vanilla 3-layer RVQ (ablation)LLM-Codec · ESC50 sound event classification, 2-way 1-shotAccuracy (%): 55 -> 60Helped
use multi-scale 3-layer RVQ instead of vanilla 3-layer RVQ (ablation)LLM-Codec · VCTK reconstructionPESQ: 2.64 -> 2.55Hurt
use down-sampling 320 instead of 480 (ablation)LLM-Codec multi-scale 3 RVQ · VCTK reconstructionPESQ: 2.55 -> 2.60Helped
use down-sampling 320 instead of 480 (ablation)LLM-Codec multi-scale 3 RVQ · ESC50 sound event classification, 2-way 1-shotAccuracy (%): 60 -> 57Hurt
add semantic loss (ablation)LLM-Codec · ESC50 sound event classification, 2-way 1-shotAccuracy (%): 58 -> 60Helped
add consistency loss (ablation)LLM-Codec · VCTK reconstructionPESQ: 1.19 -> 2.55Helped
add consistency loss (ablation)LLM-Codec · ESC50 sound event classification, 2-way 1-shotAccuracy (%): 48 -> 60Helped
use word-level codebook instead of sub-word codebook (ablation)LLM-Codec first VQ layer · VCTK reconstructionPESQ: 2.46 -> 2.55Helped
use word-level codebook instead of sub-word codebook (ablation)LLM-Codec first VQ layer · ESC50 sound event classification, 2-way 1-shotAccuracy (%): 59 -> 60Helped
update codebooks instead of keeping them frozen (ablation)LLM-Codec · VCTK reconstructionPESQ: 2.55 -> 2.63Helped
update codebooks instead of keeping them frozen (ablation)LLM-Codec · ESC50 sound event classification, 2-way 1-shotAccuracy (%): 60 -> 55Hurt

Surprising findings

Limitations

Experimental setup

Datasets: MLS AudioCaps ESD ESC-50 VCTK NoiseX-92 Free Spoken Digit Dataset Speech Commands Dynamic-SUPERB

Benchmarks: ESD ESC-50 Dynamic-SUPERB FSDD VCTK Speech Commands

Metrics: accuracy PESQ STOI DNSMOS

Baselines: BLSP ImageBind-LLM Whisper-LLM ASR-ChatGPT Encodec DAC SGMSE+ FastSpeech 2

Builds on: VQ-VAE EnCodec DAC SpeechTokenizer UniAudio

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy