atlas

Research map / Large language models / Alignment and preferences

SpeechAlign: Aligning Speech Generation to Human Preferences

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Robust and calibrated DPO · 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 SpeechAlign, an iterative self-improvement method that builds a preference codec dataset (golden vs. synthetic first-layer codec tokens) and applies DPO, CoH, RLHF-PPO, or best-of-N to align a SpeechGPT-based codec language model.

Problem

Neural codec language models for speech suffer a distribution gap between golden AR codec tokens (training) and synthetic AR tokens (inference), and speech outputs are rarely aligned to human preferences.

Approach

Treat golden codec tokens from real speech as preferred and model-generated synthetic tokens as dispreferred, avoiding human annotation of codec tokens, then iteratively run preference optimization (mainly DPO) so the model converts weak codec LMs into stronger ones.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace baseline SFT with DPO through iteration 3SpeechAlign-sft · LibriSpeech test-cleanWER: 7.2 -> 6.0Helped
replace baseline SFT with DPO through iteration 3SpeechAlign-sft · VCTKWER: 8.8 -> 7.9Helped
increase DPO iterations from 1 to 3SpeechAlign · LibriSpeech test-cleanWER: 6.7 -> 6.0Helped
increase DPO iterations from 1 to 3SpeechAlign · VCTKWER: 8.5 -> 7.9Helped
condition the NAR model on golden rather than synthetic AR tokensSoundStorm NAR model · NAR reconstructionWER: 7.2 -> 5.9Helped
condition the NAR model on golden rather than synthetic AR tokensSoundStorm NAR model · NAR reconstructionSIM: 0.87 -> 0.93Helped
increase preference data from 0 to 50k (ablation)SpeechAlign DPO · zero-shot TTSWERHelped
increase preference data from 50k to 250k (ablation)SpeechAlign DPO · zero-shot TTSWERNo clear effect
continue supervised fine-tuning on ground-truth AR tokens instead of preference-aware optimization (ablation)SpeechAlign-sft · LibriSpeech test-cleanWER: 7.2 -> 8.0Hurt
continue supervised fine-tuning on ground-truth AR tokens instead of preference-aware optimization (ablation)SpeechAlign-sft · VCTKWER: 8.8 -> 9.8Hurt
apply DPO in the first iteration130M-parameter AR model · LibriSpeech test-cleanWER: 11.4 to 9.3Helped
apply DPO in the first iteration130M-parameter AR model · LibriSpeech test-cleanSIM: 0.87 to 0.88Helped

Surprising findings

Limitations

Experimental setup

Datasets: LibriSpeech VCTK Multilingual LibriSpeech

Benchmarks: LibriSpeech test-clean VCTK

Metrics: WER speaker similarity (SIM) human win rate

Baselines: SpeechAlign-sft Continue SFT SpeechAlign-CoH SpeechAlign-RLHF-PPO SpeechAlign-BoN

Builds on: SpeechGPT SoundStorm SpeechTokenizer Direct Preference Optimization Chain-of-Hindsight

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