You can deploy this model and start earning money today!

khanhld/
chunkformer-rnnt-large-vie

Name: chunkformer-rnnt-large-vie
Availability: InStock

automatic-speech-recognition

Views

No views yet

ChunkFormer-RNNT-Large-Vie: Large-Scale Pretrained ChunkFormer-RNNT for Vietnamese Automatic Speech Recognition

License: CC BY 4.0

Model Description
Documentation and Implementation
Benchmark Results
Usage
Citation
Contact

Model Description

ChunkFormer-RNNT-Large-Vie is a large-scale Vietnamese Automatic Speech Recognition (ASR) model based on the ChunkFormer architecture, introduced at ICASSP 2025. The model has been fine-tuned on approximately 5000 hours of public Vietnamese speech data sourced from diverse datasets.

Documentation and Implementation

The Documentation and Implementation of ChunkFormer are publicly available.

Benchmark Results

We evaluate the models using Word Error Rate (WER). To ensure consistency and fairness in comparison, we manually apply Text Normalization, including the handling of numbers, uppercase letters, and punctuation.

Public Models:

STT	Model	#Params	Vivos	Common Voice	VLSP 2020 - Task 1	VLSP 2020 - Task 2	Avg.
1	ChunkFormer-RNNT-Large-Vie	113M	2.49	5.18	12.75	20.47	10.22
2	ChunkFormer-CTC-Large-Vie	110M	4.18	6.66	14.09	25.81	12.69
3	vinai/PhoWhisper-large	1.55B	4.67	8.14	13.75	26.68	13.31
4	nguyenvulebinh/wav2vec2-base-vietnamese-250h	95M	10.77	18.34	13.33	51.45	23.47
5	openai/whisper-large-v3	1.55B	8.81	15.45	20.41	68.61	28.32
6	khanhld/wav2vec2-base-vietnamese-160h	95M	15.05	10.78	31.62	62.01	29.87
7	homebrewltd/Ichigo-whisper-v0.1	22M	13.46	23.52	21.64	62.92	30.39

Private Models (API):

STT	Model	VLSP - Task 1
1	ChunkFormer-RNNT-Large-Vie	12.8
2	ChunkFormer-CTC-Large-Vie	14.1
3	Viettel	14.5
4	Google	19.5
5	FPT	28.8

Quick Usage

To use the ChunkFormer model for Vietnamese Automatic Speech Recognition, follow these steps:

Option 1: Install from PyPI (Recommended)

pip install chunkformer

Option 2: Install from source

bash

git clone https://github.com/khanld/chunkformer.git
cd chunkformer
pip install -e .

Python API Usage

python

from chunkformer import ChunkFormerModel

# Load the Vietnamese model from Hugging Face
model = ChunkFormerModel.from_pretrained("khanhld/chunkformer-rnnt-large-vie")

# For single long-form audio transcription
transcription = model.endless_decode(
    audio_path="path/to/long_audio.wav",
    chunk_size=64,
    left_context_size=128,
    right_context_size=128,
    total_batch_duration=14400,  # in seconds
    return_timestamps=True
)
print(transcription)

# For batch processing of multiple audio files
audio_files = ["audio1.wav", "audio2.wav", "audio3.wav"]
transcriptions = model.batch_decode(
    audio_paths=audio_files,
    chunk_size=64,
    left_context_size=128,
    right_context_size=128,
    total_batch_duration=1800  # Total batch duration in seconds
)

for i, transcription in enumerate(transcriptions):
    print(f"Audio {i+1}: {transcription}")

Command Line Usage

After installation, you can use the command line interface:

bash

chunkformer-decode \
    --model_checkpoint khanhld/chunkformer-rnnt-large-vie \
    --long_form_audio path/to/audio.wav \
    --total_batch_duration 14400 \
    --chunk_size 64 \
    --left_context_size 128 \
    --right_context_size 128

Example Output:

[00:00:01.200] - [00:00:02.400]: this is a transcription example
[00:00:02.500] - [00:00:03.700]: testing the long-form audio

Advanced Usage can be found HERE

Citation

If you use this work in your research, please cite:

bibtex

@INPROCEEDINGS{10888640,
  author={Le, Khanh and Ho, Tuan Vu and Tran, Dung and Chau, Duc Thanh},
  booktitle={ICASSP 2025 - 2025 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)}, 
  title={ChunkFormer: Masked Chunking Conformer For Long-Form Speech Transcription}, 
  year={2025},
  volume={},
  number={},
  pages={1-5},
  keywords={Scalability;Memory management;Graphics processing units;Signal processing;Performance gain;Hardware;Resource management;Speech processing;Standards;Context modeling;chunkformer;masked batch;long-form transcription},
  doi={10.1109/ICASSP49660.2025.10888640}}
}

khanhld/chunkformer-rnnt-large-vie