hynt commited on
Commit
14497b0
Β·
verified Β·
1 Parent(s): 61417f1

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +84 -3
README.md CHANGED
@@ -1,3 +1,84 @@
1
- ---
2
- license: cc-by-nc-sa-4.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-nd-4.0
3
+ ---
4
+ # Vietnamese Speech-to-Text (ASR) β€” ZipFormer-30M-RNNT-6000h
5
+
6
+ ## πŸ” Overview
7
+ The **Vietnamese Speech-to-Text (ASR)** model is built on the **ZipFormer architecture** β€” an improved variant of the Conformer β€” featuring only **30 million parameters** yet achieving **exceptional performance** in both speed and accuracy.
8
+ On CPU, the model can transcribe a **12-second audio clip in just 0.3 seconds**, significantly faster than most traditional ASR systems without requiring a GPU.
9
+
10
+ ---
11
+
12
+ ## πŸš€ Online Demo
13
+
14
+ You can try the model directly here:
15
+ πŸ‘‰ https://huggingface.co/spaces/hynt/k2-automatic-speech-recognition-demo
16
+
17
+ ---
18
+
19
+ ## βš™οΈ Model Architecture and Training strategy:
20
+ - **Architecture:** ZipFormer
21
+ - **Parameters:** ~30M
22
+ - **Language:** Vietnamese
23
+ - **Loss Function:** RNN-Transducer (RNNT Loss)
24
+ - **Framework:** PyTorch + k2
25
+ - **Training strategy**: Carefully preprocess the data, apply an augmentation strategy based on the distribution of out-of-vocabulary (OOV) tokens and refine the transcriptions using Whisper.
26
+ - **Optimized for:** High-speed CPU inference
27
+
28
+ ---
29
+
30
+ ## 🧠 Training Data
31
+ The model was trained on approximately **6000 hours of high-quality Vietnamese speech** collected from various public datasets:
32
+
33
+ | Dataset | | |
34
+ |----------|----------|----------|
35
+ | VLSP2020 | VLSP2021 | VLSP2023-voting-pseudo-labeled |
36
+ | VLSP2023 | FPT | VIET_BUD500 |
37
+ | VietSpeech | FLEURS | VietMed_Labeled |
38
+ | Sub-GigaSpeech2-Vi | ViVoice | Sub-PhoAudioBook |
39
+
40
+ ---
41
+
42
+ ## πŸ§ͺ Evaluation Results
43
+
44
+ | **Dataset** | **ZipFormer-30M-6000h** | **ChunkFormer-110M-3000h** | **PhoWhisper-Large-1.5B-800h** | **VietASR-ZipFormer-68M-70.000h** |
45
+ |--------------|--------------------------|-----------------------------|--------------------------------|---------------------------------|
46
+ | **VLSP2020-Test-T1** | **12.29** | 14.09 | 13.75 | 14.45 |
47
+ | **VLSP2023-PublicTest** | **10.40** | 16.15 | 16.83 | 14.70 |
48
+ | **VLSP2023-PrivateTest** | **11.10** | 17.12 | 17.10 | 15.07 |
49
+ | **VLSP2025-PublicTest** | **7.97** | 15.55 | 16.14 | 13.55 |
50
+ | **VLSP2025-PrivateTest** | **8.10** | 16.07 | 16.31 | 13.97 |
51
+ | **GigaSpeech2-Test** | 7.56 | 10.35 | 10.00 | **6.88** |
52
+
53
+ > Lower is better (WER %)
54
+
55
+ ---
56
+
57
+ ## πŸ† Achievements
58
+ By training this model architecture on 4,000 hours of data, I **won First Place** in the **Vietnamese Language Speech Processing (VLSP)** competition **2025**.
59
+ Comprehensive details about **training data**, **optimization strategies**, **architecture improvements**, and **evaluation methodologies** are available in the paper below:
60
+
61
+ πŸ‘‰ [Read the full paper on Overleaf](https://www.overleaf.com/read/wjntrgchhbgv#48aa25)
62
+
63
+ ---
64
+
65
+ ## ⚑ Inference Speed
66
+
67
+ | **Device** | **Audio Length** | **Inference Time** |
68
+ |-------------|------------------|--------------------|
69
+ | CPU (Hugging Face Basic) | 12 seconds | **0.3 s** |
70
+ | GPU (RTX 3090) | 12 seconds | **< 0.1 s** |
71
+
72
+ ---
73
+
74
+ ## βš™οΈ How to Run This Model
75
+
76
+ Please refer to the following guides for instructions on how to run and deploy this model:
77
+ - **For Torch JIT Script:** [https://k2-fsa.github.io/sherpa/](https://k2-fsa.github.io/sherpa/)
78
+ - **For ONNX:** [https://k2-fsa.github.io/sherpa/onnx/](https://k2-fsa.github.io/sherpa/onnx/)
79
+
80
+ ## πŸ’¬ Summary
81
+ The **ZipFormer-30M-RNNT-6000h** model demonstrates that a lightweight architecture can still achieve state-of-the-art accuracy for Vietnamese ASR.
82
+ It is designed for **fast deployment on CPU-based systems**, making it ideal for **real-time speech recognition**, **callbots**, and **embedded speech interfaces**.
83
+
84
+ ---