manassehzw commited on
Commit
e15b122
·
verified ·
1 Parent(s): 2c811bc

Add model card and evaluation results

Browse files
Files changed (1) hide show
  1. README.md +142 -0
README.md ADDED
@@ -0,0 +1,142 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - sna
4
+ license: apache-2.0
5
+ library_name: qwen-asr
6
+ pipeline_tag: automatic-speech-recognition
7
+ tags:
8
+ - automatic-speech-recognition
9
+ - audio
10
+ - speech
11
+ - shona
12
+ - qwen3-asr
13
+ - transformers
14
+ base_model: Qwen/Qwen3-ASR-1.7B
15
+ datasets:
16
+ - manassehzw/sna-dataset-annotated
17
+ metrics:
18
+ - wer
19
+ - cer
20
+ model-index:
21
+ - name: Shona Qwen3-ASR 1.7B
22
+ results:
23
+ - task:
24
+ type: automatic-speech-recognition
25
+ name: Automatic Speech Recognition
26
+ dataset:
27
+ name: Shona Annotated Validation Probe
28
+ type: manassehzw/sna-dataset-annotated
29
+ split: validation
30
+ metrics:
31
+ - name: Normalized Word Error Rate
32
+ type: wer
33
+ value: 0.2519838056680162
34
+ - name: Normalized Character Error Rate
35
+ type: cer
36
+ value: 0.050787599076953946
37
+ ---
38
+
39
+ # manassehzw/sna-qwen-asr-1.7b
40
+
41
+ Shona Qwen3-ASR 1.7B is a Shona (`sna`) automatic speech recognition model
42
+ fine-tuned from [`Qwen/Qwen3-ASR-1.7B`](https://huggingface.co/Qwen/Qwen3-ASR-1.7B)
43
+ on [`manassehzw/sna-dataset-annotated`](https://huggingface.co/datasets/manassehzw/sna-dataset-annotated).
44
+
45
+ ## Overview
46
+
47
+ This release contains checkpoint 760 from run `20260803T140000Z`. It was selected
48
+ by generated validation WER and occurs at approximately two epochs. Qwen3-ASR
49
+ adapted rapidly to Shona and substantially outperformed its zero-shot behavior on
50
+ the project dataset. Evaluation on FLEURS also revealed a meaningful domain and
51
+ speaker-generalization gap, so the in-domain score should not be treated as a
52
+ universal Shona ASR result.
53
+
54
+ ## Model Details
55
+
56
+ - **Curated by:** [Manasseh Changachirere (Harare Institute of Technology)](https://www.manasseh.dev/)
57
+ - **Base model:** [`Qwen/Qwen3-ASR-1.7B`](https://huggingface.co/Qwen/Qwen3-ASR-1.7B)
58
+ - **Base model revision:** `7278e1e70fe206f11671096ffdd38061171dd6e5`
59
+ - **Training dataset:** [`manassehzw/sna-dataset-annotated`](https://huggingface.co/datasets/manassehzw/sna-dataset-annotated)
60
+ - **Dataset revision:** `f91b1a79cbac15520d3c808b56e5192bf903f280`
61
+ - **Training run:** `20260803T140000Z`
62
+ - **Released checkpoint:** `checkpoint-760`
63
+ - **Checkpoint epoch:** approximately `1.996`
64
+ - **Selection criterion:** lowest generated WER among retained checkpoints
65
+ - **Framework:** `qwen-asr==0.0.6`, Transformers, PyTorch
66
+
67
+ ## Evaluation
68
+
69
+ Text for the project evaluations was normalized with Unicode NFKC, case folding,
70
+ punctuation and symbol removal, and whitespace collapse.
71
+
72
+ | Evaluation | Checkpoint | Examples | WER | CER |
73
+ |---|---:|---:|---:|---:|
74
+ | In-domain validation generation probe | 760 | 256 | **25.20%** | **5.08%** |
75
+ | Full in-domain test set | 855 | 1,565 | 25.51% | 4.97% |
76
+ | PazaBench v2 / FLEURS `sn_zw` test | 855 | 925 | 52.85% | 13.57% |
77
+ | Zero-shot base model, full in-domain test set | base | 1,565 | 106.58% | 31.64% |
78
+
79
+ The full-test and FLEURS numbers are included as nearby-run context and were
80
+ measured on checkpoint 855, not the released checkpoint. Checkpoints 760 and 855
81
+ were effectively tied on the 256-example generation probe: 25.198% versus
82
+ 25.215% WER, a difference of one word error. A full external evaluation of
83
+ checkpoint 760 has not yet been recorded.
84
+
85
+ ## Training Summary
86
+
87
+ - **Training examples:** 12,170
88
+ - **Validation examples:** 1,504
89
+ - **Held-out test examples:** 1,565
90
+ - **Training audio:** approximately 62.63 hours
91
+ - **Learning rate:** `2e-5`
92
+ - **Microbatch size:** 4
93
+ - **Gradient accumulation:** 8
94
+ - **Effective batch size:** 32
95
+ - **Precision:** bfloat16
96
+ - **Scheduler:** linear with 2% warmup
97
+ - **Planned epochs:** 3
98
+ - **Released checkpoint:** approximately 2 epochs / 760 optimizer steps
99
+ - **Checkpoint evaluation loss:** `0.17747`
100
+
101
+ ## Example Usage
102
+
103
+ Install the official Qwen ASR package:
104
+
105
+ ```bash
106
+ pip install -U qwen-asr
107
+ ```
108
+
109
+ ```python
110
+ import torch
111
+ from qwen_asr import Qwen3ASRModel
112
+
113
+ model = Qwen3ASRModel.from_pretrained(
114
+ "manassehzw/sna-qwen-asr-1.7b",
115
+ dtype=torch.bfloat16,
116
+ device_map="cuda:0",
117
+ max_inference_batch_size=16,
118
+ max_new_tokens=256,
119
+ )
120
+
121
+ results = model.transcribe(audio="sample.wav", language=None)
122
+ print(results[0].text)
123
+ ```
124
+
125
+ The recorded evaluations used `language=None`. Although the fine-tuning targets
126
+ identify the output as Shona, Shona was not in the base model's original list of
127
+ supported language arguments, so automatic language handling is the tested path.
128
+
129
+ ## Limitations
130
+
131
+ - External FLEURS performance is substantially weaker than in-domain performance.
132
+ - The training corpus is relatively small and may not cover Zimbabwe's full range
133
+ of speakers, accents, recording conditions, and conversational domains.
134
+ - Code-switching was not separately quantified for this checkpoint.
135
+ - Long-form, streaming, noisy, telephone, and far-field behavior require further
136
+ evaluation.
137
+ - Orthographically close substitutions can produce a low CER while still being
138
+ penalized heavily by WER.
139
+
140
+ ## License
141
+
142
+ The model is released under Apache-2.0, matching the base Qwen3-ASR model.