andrewbawitlung's picture
Add CER, hyperparameters, and training logs to README
9bd5867 verified
|
Raw
History Blame
5.44 kB
metadata
language:
  - lus
license: apache-2.0
base_model: Qwen/Qwen3-ASR-0.6B
tags:
  - generated_from_trainer
datasets:
  - andrewbawitlung/mizonal-v3
metrics:
  - wer
  - cer
model-index:
  - name: qwen3-asr-0.6b-mizonal3-E4-lus-v2026.06
    results:
      - task:
          name: Automatic Speech Recognition
          type: automatic-speech-recognition
        dataset:
          name: MiZonal v3
          type: andrewbawitlung/mizonal-v3
          config: default
          split: test
        metrics:
          - name: Wer
            type: wer
            value: 22.2146
          - name: Cer
            type: cer
            value: 5.2331
          - name: Real Time Factor
            type: rtf
            value: 0.0685

qwen3-asr-0.6b-mizonal3-E4-lus-v2026.06

This model is a fine-tuned version of Qwen/Qwen3-ASR-0.6B on the MiZonal v3 dataset. Note: ~1 hour of conversational speech was added to this dataset version.

It achieves the following results on the evaluation set:

  • Wer: 22.2146
  • Cer: 5.2331
  • Real Time Factor: 0.0685

Quick Inference

import torch
import librosa
from transformers import AutoProcessor, Qwen2AudioForConditionalGeneration

device = "cuda" if torch.cuda.is_available() else "cpu"

processor = AutoProcessor.from_pretrained("andrewbawitlung/qwen3-asr-0.6b-mizonal3-E4-lus-v2026.06")
model = Qwen2AudioForConditionalGeneration.from_pretrained("andrewbawitlung/qwen3-asr-0.6b-mizonal3-E4-lus-v2026.06").to(device)

audio, sr = librosa.load("your_audio.wav", sr=16000)

conversation = [
    {"role": "user", "content": [
        {"type": "audio", "audio_url": "your_audio.wav"},
        {"type": "text", "text": "Transcribe the audio:"}
    ]}
]
text = processor.apply_chat_template(conversation, add_generation_prompt=True, tokenize=False)
inputs = processor(text=text, audios=[audio], return_tensors="pt", padding=True)
inputs.input_ids = inputs.input_ids.to(device)

with torch.no_grad():
    generate_ids = model.generate(**inputs, max_length=256)

generate_ids = generate_ids[:, inputs.input_ids.size(1):]
transcription = processor.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[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

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 16
  • 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
200 0.36 0.34 0.59 0.43 0.11 1.95e-05 27.88
400 0.73 0.22 0.40 0.33 0.08 1.86e-05 9.94
600 1.09 0.14 0.33 0.27 0.06 1.76e-05 9.44
800 1.46 0.10 0.31 0.25 0.06 1.67e-05 4.19
1000 1.82 0.10 0.29 0.23 0.05 1.58e-05 4.47
1200 2.19 0.08 0.28 0.22 0.05 1.48e-05 6.12
1400 2.55 0.06 0.28 0.21 0.05 1.39e-05 4.47
1600 2.91 0.06 0.27 0.21 0.05 1.30e-05 4.25
1800 3.28 0.04 0.28 0.21 0.05 1.20e-05 3.98
2000 3.64 0.03 0.29 0.20 0.05 1.11e-05 6.34
2200 4.01 0.04 0.29 0.21 0.05 1.02e-05 4.41
2400 4.37 0.02 0.30 0.21 0.05 9.26e-06 4.47
2600 4.74 0.02 0.30 0.21 0.05 8.33e-06 6.00
2800 5.10 0.01 0.32 0.21 0.05 7.40e-06 2.19
3000 5.46 0.01 0.32 0.21 0.05 6.47e-06 2.75
3200 5.83 0.01 0.33 0.21 0.05 5.54e-06 1.66
3400 6.19 0.01 0.34 0.21 0.05 4.61e-06 0.77
3600 6.56 0.01 0.34 0.21 0.05 3.68e-06 1.29
3800 6.92 0.01 0.34 0.21 0.05 2.76e-06 2.17
4000 7.29 0.01 0.34 0.21 0.05 1.83e-06 2.41
4200 7.65 0.01 0.34 0.21 0.05 8.97e-07 3.73