
Disclaimer / Notice: Details for these are in Peer Review and publications of the paper will be made available soon for more details.
xlsr-300m-mizonal3-E1-lus-v2026.06
This model is a fine-tuned version of facebook/wav2vec2-xls-r-300m on the MiZonal v3.0 dataset. Note: ~1 hour of conversational speech was added to this dataset version.
It achieves the following results on the evaluation set:
- Wer: 19.1028
- Cer: 3.7453
- Real Time Factor: 0.0024
Quick Inference
import torch
import librosa
from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
device = "cuda" if torch.cuda.is_available() else "cpu"
processor = Wav2Vec2Processor.from_pretrained("andrewbawitlung/xlsr-300m-mizonal3-E1-lus-v2026.06")
model = Wav2Vec2ForCTC.from_pretrained("andrewbawitlung/xlsr-300m-mizonal3-E1-lus-v2026.06").to(device)
audio, sr = librosa.load("your_audio.wav", sr=16000)
input_values = processor(audio, sampling_rate=16000, return_tensors="pt").input_values.to(device)
with torch.no_grad():
logits = model(input_values).logits
predicted_ids = torch.argmax(logits, dim=-1)
transcription = processor.batch_decode(predicted_ids)[0]
print(transcription)
Model description
Experiment Configurations
This repository is part of a series of experiments. The different configurations are:
- E1 (Baseline): Standard training configuration.
- E2 (Noise): Training with background noise augmentation.
- E3 (Speed): Training with speed perturbation augmentation.
- E4 (SpecAug): Training with SpecAugment (time and frequency masking).
- E5 (Combined): Training with a combination of all augmentations.
All Models in this Family
| Experiment | Hugging Face Repository |
|---|---|
| E1 (Baseline) | andrewbawitlung/xlsr-300m-mizonal3-E1-lus-v2026.06 |
| E2 (Noise) | andrewbawitlung/xlsr-300m-mizonal3-E2-lus-v2026.06 |
| E3 (Speed) | andrewbawitlung/xlsr-300m-mizonal3-E3-lus-v2026.06 |
| E4 (SpecAug) | andrewbawitlung/xlsr-300m-mizonal3-E4-lus-v2026.06 |
| E5 (Combined) | andrewbawitlung/xlsr-300m-mizonal3-E5-lus-v2026.06 |
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.0003
- train_batch_size: 8
- eval_batch_size: 8
- seed: 42
- optimizer: OptimizerNames.ADAMW_TORCH_FUSED
- lr_scheduler_type: SchedulerType.LINEAR
- num_epochs: 8
Training results
| step | epoch | train_loss | eval_loss | eval_wer | eval_cer | learning_rate | grad_norm |
|---|---|---|---|---|---|---|---|
| 250 | 0.91 | 10.3415 | 2.0216 | 99.94 | 67.13 | 1.49e-04 | 2.11 |
| 500 | 1.82 | 2.0685 | 0.4081 | 43.59 | 10.73 | 2.99e-04 | 3.12 |
| 750 | 2.73 | 1.3136 | 0.2638 | 32.21 | 7.24 | 2.56e-04 | 2.03 |
| 1000 | 3.64 | 0.9428 | 0.2079 | 25.07 | 5.36 | 2.12e-04 | 1.83 |
| 1250 | 4.55 | 0.6437 | 0.1911 | 22.76 | 4.68 | 1.68e-04 | 1.46 |
| 1500 | 5.46 | 0.4784 | 0.1780 | 20.67 | 4.26 | 1.24e-04 | 1.04 |
| 1750 | 6.36 | 0.4316 | 0.1628 | 19.78 | 3.93 | 7.96e-05 | 1.37 |
| 2000 | 7.27 | 0.3200 | 0.1597 | 18.16 | 3.67 | 3.55e-05 | 1.32 |
| 2200 | 8.00 | 0.3271 | 0.1567 | 17.67 | 3.57 | 1.76e-07 | 0.92 |