Whisper small.en โ€” Gumbel-BEARD

This is an openai/whisper-small.en model whose encoder was domain-adapted with Gumbel-BEARD โ€” a self-supervised continued-pretraining recipe โ€” and then fine-tuned for automatic speech recognition (ASR) on the MyST children's speech corpus.

On the MyST test set it achieves 8.5% WER.

  • Base model: openai/whisper-small.en
  • Task: English automatic speech recognition
  • Adaptation: self-supervised (transcript-free) encoder adaptation via Gumbel-BEARD, followed by supervised ASR fine-tuning
  • Code:

Usage

from transformers import pipeline

asr = pipeline(
    "automatic-speech-recognition",
    model="Zilai2999/whisper-small.en-gumbel-beard",
)
print(asr("audio.wav")["text"])

Or with the model/processor directly:

import torch, librosa
from transformers import WhisperForConditionalGeneration, WhisperProcessor

model_id = "Zilai2999/whisper-small.en-gumbel-beard"
processor = WhisperProcessor.from_pretrained(model_id)
model = WhisperForConditionalGeneration.from_pretrained(model_id)

audio, _ = librosa.load("audio.wav", sr=16000)
inputs = processor(audio, sampling_rate=16000, return_tensors="pt")
ids = model.generate(inputs.input_features)
print(processor.batch_decode(ids, skip_special_tokens=True)[0])

Audio is expected at 16 kHz.

Training

  • Self-supervised Gumbel-BEARD adaptation of the Whisper encoder (BEST-RQ + self-distillation with automatic Gumbel-softmax layer selection).
  • Supervised ASR fine-tuning on the MyST children's speech corpus.

See the code repository for the full recipe and hyper-parameters.

Results

Test set WER
MyST 8.5%

Citation

@inproceedings{gumbelbeard2026,
  title     = {Gumbel-BEARD: Automatic Layer Selection for Self-Supervised
               Adaptation of Whisper in Low-Resource Domains},
  booktitle = {Proc. Interspeech},
  year      = {2026},
}
Downloads last month
3
Safetensors
Model size
0.2B params
Tensor type
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for Zilai2999/whisper-small.en-gumbel-beard

Finetuned
(72)
this model