WaveNet

The paper introduced a deep autoregressive model for raw audio using causal dilated convolutions, gated residual blocks, and sample-level likelihood training.

Metadata

FieldValue
PaperWaveNet: A Generative Model for Raw Audio
AuthorsAaron van den Oord, Sander Dieleman, Heiga Zen, Karen Simonyan, Oriol Vinyals, Alex Graves, Nal Kalchbrenner, Andrew Senior, Koray Kavukcuoglu
Year2016
VenuearXiv; ISCA Speech Synthesis Workshop demo
arXiv1609.03499
DeepMindWaveNet blog
Statusfull note started

One-Line Takeaway

WaveNet models raw audio autoregressively:

using a stack of causal dilated convolutions so each output sample has a large receptive field over previous samples.

Question

Audio is a high-rate sequence. At 16 kHz, one second has:

samples. Directly modeling raw waveform samples requires both:

  • causal generation, because future samples are unavailable;
  • a long receptive field, because speech and music depend on structure over many time scales.

The architecture question is:

WaveNet answers with dilated causal convolutions.

Main Claim

A deep causal convolutional network with exponentially increasing dilation can model raw audio waveforms and generate high-quality speech/music samples.

The core architecture:

This is important beyond audio because it shows how convolution can provide long-range sequence context without recurrence.

Architecture Contract

ItemContract
Inputdiscrete raw audio samples
Outputdistribution over next audio sample
Factorizationautoregressive sample-level likelihood
Causalityoutput at time only depends on samples before
Context mechanismdilated causal convolutions
Blockgated activation with residual and skip connections
Conditioningoptional local linguistic features or global speaker identity
Main taskraw audio generation, especially text-to-speech
Main bottlenecksequential sampling at inference

Autoregressive Audio Likelihood

For waveform:

WaveNet factorizes:

The training objective is negative log likelihood:

where is optional conditioning.

The model is trained with teacher forcing: previous ground-truth samples are available during training, so all positions can be evaluated in parallel through convolution.

Causal Convolution

A causal convolution for output time only reads current and past inputs:

It must not read:

because generation proceeds one sample at a time.

This causal contract is the sequence analogue of masking in a decoder-only Transformer:

Dilated Convolution

A dilated convolution skips input positions by a dilation factor :

For kernel size :

If dilation doubles by layer:

the receptive field grows exponentially with depth.

For a stack of layers with kernel size , the receptive field is roughly:

With doubling dilation:

This is why WaveNet can cover long audio context with a moderate number of layers.

Gated Residual Block

WaveNet uses gated activation units:

where:

SymbolMeaning
dilated causal convolution
filter convolution
gate convolution
sigmoid gate
elementwise product

The block then uses residual and skip paths:

and:

The skip sum feeds the output network. Residual paths stabilize deep stacks, while skip paths expose features from multiple receptive-field scales.

Output Distribution

The original WaveNet discretizes audio samples and predicts a categorical distribution.

For quantized sample value:

the output is:

The loss is cross-entropy over sample values:

Later neural vocoders and audio models often changed the output distribution, but the architecture idea of causal dilated waveform modeling remained influential.

Conditioning

WaveNet supports conditioning:

Global conditioning can represent speaker identity:

Local conditioning can represent time-aligned linguistic or acoustic features:

that are injected into layers.

The paper’s text-to-speech setting is therefore:

Training Vs Sampling

Training can evaluate all time positions in parallel because ground-truth history is available:

comes from data.

Sampling is sequential:

Each new sample becomes input for the next step.

This creates the major tradeoff:

PhaseProperty
trainingparallel convolution over full sequence
samplingsequential autoregressive generation
qualityhigh sample-level fidelity
latencyexpensive without caching/distillation

Relation To CNNs And Transformers

WaveNet is a sequence CNN:

Compared with a Transformer:

AxisWaveNetTransformer
mixingconvolutional filtersattention
context growthdilation stackdirect token-token attention
causalitycausal convolutioncausal mask
train parallelismyesyes
samplingsequentialsequential for autoregressive decoding
inductive biaslocal translation-in-time filterscontent-based global interaction

WaveNet remains useful when reading architectures that prefer convolutional locality and long receptive fields over attention.

Relation To State-Space And Long-Sequence Models

WaveNet, Transformer-XL, and Mamba all address long sequence context differently:

ModelContext Mechanism
WaveNetdilated causal convolution receptive field
Transformer-XLcached hidden-state memory plus relative attention
Mambaselective state-space recurrence

This gives a clean reading axis:

The question is whether the task benefits more from locality, content-based lookup, or recurrent state.

Why It Belongs In Architecture Papers

WaveNet is a canonical architecture paper because it popularized:

  • direct raw waveform modeling;
  • causal convolution for autoregressive generation;
  • dilated convolution for exponential receptive-field growth;
  • gated residual blocks for deep sequence CNNs;
  • conditioning for controllable audio generation.

The architecture lesson applies outside speech:

Evidence Pattern

The paper supports the architecture with:

EvidenceWhat It Supports
text-to-speech listening testsraw waveform generation improves naturalness
speaker conditioningone model can represent multiple speakers
music modeling samplesarchitecture is not limited to speech
phoneme recognition experimentlearned waveform features are useful discriminatively
qualitative audiosample fidelity is central for audio claims

For audio generation, subjective evaluation matters because likelihood alone may not capture perceptual quality.

Practical Reading Checks

QuestionWhy
What is the sample rate?determines sequence length and required context
What is the receptive field?limits temporal dependencies
What output distribution is used?affects audio quality and likelihood
Is generation autoregressive?determines latency
Is conditioning local or global?controls TTS and speaker behavior
Are listening tests reported?perceptual quality is central
Is inference accelerated?raw autoregressive audio is slow

Limits

  • Autoregressive sampling is slow for long audio.
  • Receptive field is finite.
  • Discretized output can introduce modeling choices.
  • Audio quality depends heavily on conditioning and data quality.
  • Likelihood and human perception may disagree.
  • Later systems often need distillation, flow, diffusion, or non-autoregressive vocoders for speed.

The concise limitation:

What To Remember

  • WaveNet models raw audio sample by sample.
  • Causal convolutions prevent future leakage.
  • Dilated convolutions grow receptive field exponentially.
  • Gated residual blocks stabilize deep waveform modeling.
  • Training is parallel with teacher forcing; generation is sequential.
  • The paper is a key reference for convolutional long-context sequence modeling.