Robust Speech Recognition via Large-Scale Weak Supervision

Alec Radford, Jong Wook Kim, Tao Xu, Greg Brockman, Christine McLeavey, Ilya Sutskever

2022 · ICML

Robust Speech Recognition via Large-Scale Weak Supervision

Problem

Framing

Supervised ASR had strong in-domain accuracy but weak transfer, brittle multilingual coverage, and fragmented task pipelines. Whisper closes this with a single encoder-decoder Transformer trained on 680k hours of weakly supervised web audio, using task-specifying text tokens for transcription, translation, language ID, and VAD.

Currently Used Methods

Foundational

Proposed Method

Architecture

Whisper uses a sequence-to-sequence Transformer over 30-second 80-channel log-Mel spectrograms. The encoder applies two Conv1D+GELU front-end layers, sinusoidal positions, and Transformer blocks; the decoder uses learned positions, causal self-attention, and cross-attention into the encoder. Model sizes range from Tiny to Large, with 4 to 32 layers and 39M to 1550M parameters.

Overview: multitask training data, encoder-decoder Transformer, and tokenized prompting format for transcription, translation, language ID, VAD, and timestamps.

Loss / Objective

Training is standard next-token prediction over the multitask token stream.

L(θ)=t=1Tlogpθ ⁣(yty<t,x)\mathcal{L}(\theta) = - \sum_{t=1}^{T} \log p_\theta\!\left(y_t \mid y_{<t}, \mathbf{x}\right)

Algorithm

Task control is done by prefix tokens that specify language, task, timestamp behavior, and optional prior context.

pθ(yx,s)=t=1Tpθ ⁣(yty<t,x,s)p_\theta(\mathbf{y}\mid \mathbf{x}, \mathbf{s}) = \prod_{t=1}^{T} p_\theta\!\left(y_t \mid y_{<t}, \mathbf{x}, \mathbf{s}\right)

Training Procedure

Evaluation

Datasets

Metrics

Headline results

Results plots: more speech-recognition supervision strongly lowers Fleurs WER, while translation supervision only moderately raises Fleurs BLEU across languages.

Ablations

Method Strengths and Weaknesses

Strengths

Weaknesses

Suggestions from the authors

Links

Prior Papers

Further Papers