Model reference · open weights
wav2vec2-large-xlsr-53-th is an open-weight audio or speech model from airesearch, listed in the AxForge catalogue. AxForge can bring it up on EU-owned hardware for you on request — with the licence handled where one is required.
About
wav2vec2-large-xlsr-53-th Finetuning wav2vec2-large-xlsr-53 on Thai Common Voice 7.0 Read more on our blog We finetune wav2vec2-large-xlsr-53 based on Fine-tuning Wav2Vec2 for English ASR using Thai examples of Common Voice Corpus 7.0. The notebooks and scripts can be found in vistec-ai/wav2vec2-large-xlsr-53-th. The pretrained model and processor can be found at airesearch/wav2vec2-large-xlsr-53-th. robust-speech-event Add syllabletokenize, wordtokenize (PyThaiNLP) and deepcut tokenizers to eval.py from robust-speech-event Eval results on Common Voice 7 "test": Usage Datasets Common Voice Corpus 7.0](https://commonvoice.mozilla.org/en/datasets) contains 133 validated hours of Thai (255 total hours) at 5GB. We pre-tokenize with pythainlp.tokenize.wordtokenize. We preprocess the dataset using cleaning rules described in notebooks/cv-preprocess.ipynb by @tann9949. We then deduplicate and split as described in ekapolc/Thaicommonvoicesplit in order to 1) avoid data leakage due to random splits after cleaning in Common Voice Corpus 7.0 and 2) preserve the majority of the data for the training set. The dataset loading script is scripts/thcommonvoice70.py. You can use this scripts together with traincleand.tsv, validationcleaned.tsv and testcleaned.tsv to have the same splits as we do. The resulting dataset is as follows: Training We fintuned using the following configuration on a single V100 GPU and chose the checkpoint with the lowest validation loss. The finetuning script is scripts/wav2vec2finetune.py Evaluation We benchmark on the test set using WER with words tokenized by PyThaiNLP 2.3.1 and deepcut, and CER. We also measure performance when spell correction using TNC ngrams is applied. Evaluation codes can be found in notebooks/wav2vec2finetuningtutorial.ipynb. Benchmark is performed on test-unique split. ※ APIs are not finetuned with Common Voice 7.0 data LICENSE cc-by-sa 4.0 Ackowledgements model training and validation notebooks/scripts @cstorm125 dataset cleaning scripts @tann9949 dataset splits @ekapolc and @14mss running the training @mrpeerat spell correction @wannaphong
Summarised from the published model card. Read the full card on the HuggingFace links below.
Specifications
| Maker | airesearch |
|---|---|
| Type | Audio & music |
| Variants | 1 |
| Runs with | transformers |
| Released | 2022-03-02 |
| Popularity | 1.2M downloads / month |
| Likes | 28 |
| Licence | Open weights |
How it works
Variants
Open weights ship in several sizes and precisions. One page, all the variants — pick the one that fits your GPU. VRAM figures are estimates from model size.
| Variant | Params | Precision | VRAM | Fits 16 GB | Weights |
|---|---|---|---|---|---|
| wav2vec2-large-xlsr-53-th | — | BF16 | — | — | Weights ↗ |
Benchmarks
As published on the model card — the maker's own numbers, not measured by AxForge.
| Task | Dataset | Metric | Score |
|---|---|---|---|
| Automatic Speech Recognition | Common Voice 7 | Test WER | 0.952 |
| Automatic Speech Recognition | Common Voice 7 | Test SER | 1.235 |
| Automatic Speech Recognition | Common Voice 7 | Test CER | 0.162 |
Using it via the API
Once AxForge deploys wav2vec2-large-xlsr-53-th for you, it answers on the OpenAI-compatible API — the same base URL and keys as every other model. (wav2vec2-large-xlsr-53-th below is illustrative; you get the exact model name on deployment.)
$ curl -sS https://api.axforge.ai/v1/audio/transcriptions \ -H "Authorization: Bearer $AXFORGE_API_KEY" \ -F model="wav2vec2-large-xlsr-53-th" -F file=@audio.mp3
Details
Languages
Trained / evaluated on
Tags
Licence
Open weights under cc-by-sa-4.0 — commercial use is permitted. Deploy it on AxForge EU hardware on request. Read the licence ↗