How to Train a Language Model¶
This document explains how to train and package a language model for deployment.
You will usually want to deploy a language model in production. A good language model will improve transcription accuracy by correcting predictable spelling and grammatical mistakes. If you can predict what kind of speech your 🐸STT will encounter, you can make great gains in accuracy with a custom language model.
For example, if you want to transcribe university lectures on biology, you should train a language model on text related to biology. With this biology-specific language model, 🐸STT will be able to better recognize rare, hard to spell words like “cytokinesis”.
How to train a model¶
There are three steps to deploying a new language model for 🐸STT:
Identify and format text data for training
Train a KenLM language model using
Package the model for deployment with
Find Training Data¶
Language models are trained from text, and the more similar that text is to the speech your 🐸STT system encounters at run-time, the better 🐸STT will perform for you.
For example, if you would like to transcribe the nightly news, then transcripts of past nightly news programs will be your best training data. If you’d like to transcribe an audio book, then the exact text of that book will create the best possible language model. If you want to put 🐸STT on a smart speaker, your training text corpus should include all the commands you make available to the user, such as “turn off the music” or “set an alarm for 5 minutes”. If you can’t predict the kind of speech 🐸STT will hear at run-time, then you should try to gather as much text as possible in your target language (e.g. Spanish).
Once you have identified text that is appropriate for your application, you should save the text in a single file with one sentence per line. This text should not contain anything that a person wouldn’t say, such as markup language.
Our release language models for English are trained on the Multilingual LibriSpeech text available here (44G).
You can download and unpack the text with the following command:
wget https://dl.fbaipublicfiles.com/mls/mls_lm_english.tar.gz -O - | tar -xzvf
This training text is around 44GB compressed, which should give you an idea of the size of a corpus needed for general speech recognition. For more constrained use cases with smaller vocabularies, you don’t need as much data, but you should still try to gather as much as you can.
Train the Language Model¶
Assuming you found and formatted a text corpus, the next step is to use that text to train a KenLM language model with
For custom use cases, you might familiarize yourself with the KenLM toolkit. Most of the options exposed by
generate_lm.py are simply forwarded to KenLM options of the same name, so you should read the KenLM documentation in order to fully understand their behavior.
python generate_lm.py \ --input_txt mls_lm_english/data.txt \ --output_dir . \ --top_k 500000 \ --kenlm_bins path/to/kenlm/build/bin/ \ --arpa_order 5 \ --max_arpa_memory "85%" \ --arpa_prune "0|0|1" \ --binary_a_bits 255 \ --binary_q_bits 8 \ --binary_type trie
generate_lm.py will save the new language model as two files on disk:
Package the Language Model for Deployment¶
Finally, we package the trained KenLM model for deployment with
generate_scorer_package. You can find pre-built binaries for
generate_scorer_package on the official 🐸STT release page (inside
native_client.*.tar.xz). If for some (uncommon) reason you need to compile
generate_scorer_package yourself, please refer to Compile generate_scorer_package.
Package the language model for deployment with
generate_scorer_package as such:
./generate_scorer_package \ --checkpoint path/to/your/checkpoint \ --lm lm.binary \ --vocab vocab-500000.txt \ --package kenlm.scorer \ --default_alpha 0.931289039105002 \ --default_beta 1.1834137581510284
--checkpoint flag should point to the acoustic model checkpoint with which you will use the generated scorer. The alphabet will be loaded from the checkpoint. External scorers must be created with the same alphabet as the acoustic models they will be used with. The
--default_beta parameters shown above were found with the
lm_optimizer.py Python script.