GPU-Accelerated Speech to Text with Kaldi: A Tutorial on Getting Started

Recently, NVIDIA achieved GPU-accelerated speech-to-text inference with exciting performance results. That blog post described the general process of the Kaldi ASR pipeline and indicated which of its elements the team accelerated, i.e. implementing the decoder on the GPU and taking advantage of Tensor Cores in the acoustic model.

Now with the latest Kaldi container on NGC, the team has accelerated an additional element, feature extraction, resulting in even faster transcription of audio recordings for Pascal and newer GPUs. A future blog will dive into the technical details of the feature extraction acceleration, covering how the process was retargeted from the CPU to the GPU. 

This blog is about how to get up and running. We’ll walk you through reproducing our most recent performance benchmark with the GPU-accelerated LibriSpeech and ASpIRE automatic speech recognition (ASR) models, which transcribe audio recordings of speech into text. We’ll also show you how to perform ASR on your own audio recordings using the accelerated model.

 

Quick Background

Kaldi is an open-source software framework for speech processing that originated in 2009 at Johns Hopkins University with the intent to develop techniques to reduce both the cost and time required to build speech recognition systems. Kaldi has since grown to become the de-facto speech recognition toolkit in the community, helping enable speech services used by millions of people each day.

The traditional speech-to-text workflow shown in the figure below takes place in three primary phases: feature extraction (converts a raw audio signal into spectral features suitable for machine learning tasks), acoustic modeling, and language modeling. Today’s acoustic models in Kaldi replace Gaussian Mixture Models (GMMs) and Hidden Markov Models (HMMs) with both recurrent and convolutional neural networks to predict states efficiently, resulting in state-of-the-art performance. 

 

Walk-through: Three Use Cases

We’re going to cover three use cases to help you start using GPU-accelerated Kaldi in a Linux environment with at least one NVIDIA GPU installed. First, as a baseline, we show you how to reproduce our results using the LibriSpeech model. Next, we demonstrate how to swap out the LibriSpeech model for another model, ASpIRE. Lastly, we will guide you through the steps to run your own speech audio through the LibriSpeech model to produce a text transcript.

Let’s get started.

 

Setup

For each use case, you need to have an NGC account. If you don’t yet have one, please go here to make one. It’s free and only takes a few minutes to set-up.

Accelerated Kaldi is hosted on an NGC as a container, so the first step is to pull it. If you’ve never used containers or Docker, don’t worry we’ll go step-by-step.

Launch a terminal or shell, and at the command line, enter:

 

nvidia-smi

 

The output should resemble the following, and you should see your GPUs listed. In our DGX Station, we see our four V100s.

 

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.87.00    Driver Version: 418.87.00    CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  Tesla V100-DGXS...  On   | 00000000:07:00.0 Off |                    0 |
| N/A   37C    P0    38W / 300W |      8MiB / 16128MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   1  Tesla V100-DGXS...  On   | 00000000:08:00.0 Off |                    0 |
| N/A   38C    P0    37W / 300W |      0MiB / 16128MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   2  Tesla V100-DGXS...  On   | 00000000:0E:00.0 Off |                    0 |
| N/A   38C    P0    36W / 300W |      0MiB / 16128MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   3  Tesla V100-DGXS...  On   | 00000000:0F:00.0 Off |                    0 |
| N/A   38C    P0    36W / 300W |      0MiB / 16128MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|    0      2511      G   /usr/lib/xorg/Xorg                             6MiB |
+-----------------------------------------------------------------------------+

 

Next, we’ll grab the NGC container that contains GPU accelerated Kaldi. At the command line, do:

 

docker login nvcr.io
docker pull nvcr.io/nvidia/kaldi:<xx.xx>-py3

 

Where <xx.xx>-py3 varies with release (check here for the latest), but we’ll use 19.09-py3.

After a few moments, the pull operation completes, and you’re ready to try the use cases for yourself. Let’s start by having you reproduce our results.

 

Reproducing Our Results with the Accelerated LibriSpeech Model

It’s time to run the Kaldi container in nvidia-docker to reproduce our results using the accelerated LibriSpeech model. 

 

nvidia-docker run -it --rm nvcr.io/nvidia/kaldi:<xx.xx>-py3

 

At this point, you’re presented with a command prompt inside the container. That is, any commands you issue are handled by the container, not the shell from which you launched the container. It should look like below when you run the container:

 

===========
== Kaldi ==
===========

NVIDIA Release new-run-scripts (build 7601331)
Kaldi Version 5.5

Container image Copyright (c) 2019, NVIDIA CORPORATION.  All rights reserved.
Copyright 2015-2019, Kaldi contributors.  All rights reserved.

Various files include modifications (c) NVIDIA CORPORATION.  All rights reserved.
NVIDIA modifications are covered by the license terms that apply to the underlying project or file.

root@da56dd4e3852:/workspace#

 

The first step is to run the command to download example audio recordings, pre-process them by converting them to WAV files, and download the trained LibriSpeech model. We’ve made a shell script for that. Run the commands:

 

cd /workspace/nvidia-examples/librispeech/
./prepare_data.sh

 

Depending on the speed of your internet connection, this step should take up to 15 minutes.

In this benchmark, we’re using the LibriSpeech model, trained on 1,000 hours of recordings of people reading English. Once the download and setup are complete, your next step will execute a script to run the speech-to-text pipeline on the example audio recordings, accelerated by the GPU. At the command line:

 

./run_benchmark

 

Understanding the Output

The first portion of text that pops out shows several of the parameters used in the benchmark. For example, you’ll see at the very beginning there’s the flag to use the GPU. The batch size for GPU processing and the beam width for the Viterbi search are other notable parameters.

 

USE_GPU: 1
NUM_PROCESSES: 1
GPU_FEATURE: true
CPU_THREADS: 40
cuda-control-threads: 2
cuda-worker-threads: 38
batch_size: 300
batch_drain_size: 40
iterations: 10
file-limit: -1
main-q-capacity=30000
aux-q-capacity=400000
beam=10
lattice-beam=7
max-active=10000
WORKSPACE=/workspace/
MODEL_PATH=/workspace//models/LibriSpeech/
MODEL_NAME=LibriSpeech
DATASET=/workspace/datasets/LibriSpeech/test_clean/
DECODER=/opt/kaldi//src/cudadecoderbin/batched-wav-nnet3-cuda
OUTPUT_PATH=/tmp/results/LibriSpeech
… … 
All tests PASSED
Process 0:
~Group 0 completed Aggregate Total Time: 9.08942 Audio: 19452.5 RealTimeX: 2140.12
~Group 1 completed Aggregate Total Time: 13.5072 Audio: 38905 RealTimeX: 2880.31
~Group 2 completed Aggregate Total Time: 18.0279 Audio: 58357.4 RealTimeX: 3237.06
~Group 3 completed Aggregate Total Time: 22.3097 Audio: 77809.9 RealTimeX: 3487.72
~Group 4 completed Aggregate Total Time: 26.5666 Audio: 97262.4 RealTimeX: 3661.07
~Group 5 completed Aggregate Total Time: 30.8791 Audio: 116715 RealTimeX: 3779.74
~Group 6 completed Aggregate Total Time: 34.9199 Audio: 136167 RealTimeX: 3899.42
~Group 7 completed Aggregate Total Time: 39.4025 Audio: 155620 RealTimeX: 3949.49
~Group 8 completed Aggregate Total Time: 43.6977 Audio: 175072 RealTimeX: 4006.44
~Group 9 completed Aggregate Total Time: 44.4936 Audio: 194525 RealTimeX: 4371.97
Overall:  Aggregate Total Time: 44.4937 Total Audio: 194525 RealTimeX: 4371.96
  %WER 5.52 [ 2904 / 52576, 389 ins, 231 del, 2284 sub ]
  %SER 51.37 [ 1346 / 2620 ]
  Scored 2620 sentences, 0 not present in hyp.
  Expected: 2620, Actual: 2620
  Decoding completed successfully.
Total RTF: 4371.96 Average RTF: 4371.9600
All WER and PERF tests passed.

 

You’ll notice that each Group is a separate iteration we use for timing calculations. At the end of the output, we calculate the RealTimeX by dividing the number of seconds of Total Audio by the Aggregate Total Time. 

By the way, you can see GPU 0 being exercised in this output from running the nvidia-smi command again in a different shell while the benchmark was running:

 

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.67       Driver Version: 418.67       CUDA Version: 10.1     |
|-------------------------------+----------------------+----------------------+
| GPU  Name     Persistence-M| Bus-Id       Disp.A | Volatile Uncorr. ECC |
| Fan  Temp Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0 Tesla V100-DGXS...  On | 00000000:07:00.0 Off |                   0 |
| N/A   47C P0   257W / 300W |  14229MiB / 16128MiB |     90%      Default |
+-------------------------------+----------------------+----------------------+
|   1 Tesla V100-DGXS...  On | 00000000:08:00.0 Off |                    0 |
| N/A   42C P0    38W / 300W |      0MiB / 16128MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   2 Tesla V100-DGXS...  On | 00000000:0E:00.0 Off |                    0 |
| N/A   42C P0    37W / 300W |      0MiB / 16128MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+
|   3 Tesla V100-DGXS...  On | 00000000:0F:00.0 Off |                    0 |
| N/A   42C P0    37W / 300W |      0MiB / 16128MiB |      0%      Default |
+-------------------------------+----------------------+----------------------+

 

On the V100 16GB GPU in our DGX Station, this benchmark took nearly 75 seconds to complete. The throughput vs. real-time is about 4,372x — not too shabby!

 

Benchmarking the Aspire ASR Model

With our upcoming Kaldi container 19-10, you’ll also be able to run the benchmark for the accelerated Automated Speech Recognition in Reverberant Environments (ASpIRE) model. ASpIRE was developed from recordings with far-field microphones of people talking on the phone in noisy, echoey rooms. If you think about how your own speech changes between reading a book out loud and conversing with someone interactively, you’ll get the gist of the difference between ASpIRE and LibriSpeech.

The procedure is the same as for LibriSpeech, but we’ll run from the ASpIRE directory. Here are the steps:

 

cd /workspace/nvidia-examples/aspire
./run_benchmark.sh

 

The output is in the format as with LibriSpeech benchmark, and you’ll notice the performance is about 2,000x realtime.

Transcribing Your Own Data Using the Accelerated LibriSpeech Model

Applying Kaldi’s ASR to your own audio is straightforward. Let’s run through an example using the LibriSpeech model. First, keep in mind that the LibriSpeech model was generated from a corpus of clean, echo-free, high-SNR recordings. Make sure your audio files were recorded with a headset or with a mic close to the speaking person’s mouth. Finally, ensure the files are in the format required by the LibriSpeech model, which is:

  •         .WAV format
  •         16 kHz sampling rate (8 kHz for ASpIRE)
  •         Mono
  •         32-bit float

You can use Audacity, a free audio-processing application for Windows, Mac, and Linux, or a command-line tool like flac2wav to convert your audio, if necessary. Here’s how my correctly formatted audio file named “DT2.wav” looked in Audacity. Check the format info at the bottom left.

Second, copy or move your audio files into a single folder, like /home/user/audio/ . You could’ve done this step first; the order doesn’t matter in this case.

Next, we’re going to create a symbolic link from inside the Kaldi container to your audio directory, while also starting up the container. In the shell command below, replace “/home/user/audio” with your own path to your directory of .wav files.

 

nvidia-docker run -v /home/user/audio:/root/audio  --rm -it gitlab-master.nvidia.com:5005/dl/dgx/kaldi:19.09-py3-devel

 

In the container, next change directories to the LibriSpeech directory:

 

cd /workspace/nvidia-examples/librispeech

 

Almost there! Run the data preparation step:

 

./prepare_data.sh

 

This step will take around 5 to15 minutes to finish, as there are several large files to pull down. When it finishes, you can point the ./run_benchmark.sh script to your audio folder like this:

 

DATASET=/root/audio ./run_benchmark.sh

 

If you have only a few audio files to transcribe, you’ll notice the speedup won’t reach the same scale as running 1,000s of files. However, you can simulate having 1,000s of files by changing the number of iterations to run with your few files. Depending on your audio, you may need to adjust the batch size down from 300 to avoid an out of memory error. Try:

 

DATASET=/root/audio ITERATIONS=500 MAX_BATCH_SIZE=200 ./run_benchmark.sh

 

In our specific case, with our audio files, the resulting throughput was 3,684.

For the best word error rate and speedup, it’s important to consider which speech model – LibriSpeech or ASpIRE – is appropriate for your audio. LibriSpeech was trained on low-noise recordings of reading aloud, while ASpIRE was trained on slightly echoey recordings while people conversed on the telephone.

Once the script is complete, the resulting transcripts are in the /tmp/results/LibriSpeech/ directory, starting with a directory named 0. Each time you run the benchmark, this directory name will increment by one. For the example here, let’s look at the result of my reading aloud from a technical publication, stored in /root/audio/DT1.wav:

 

more /tmp/results/LibriSpeech/0/0/trans
_root_audio_DT1 this paper is part of a larger effort to detect the manipulations of video by searching for it and combining evidences of multiple types of inconsistencies between the audio and visual channels

 

The file “trans” will contain multiple transcriptions if more files are in the audio directory.

 

What’s Next

Now that you’ve tried the GPU-accelerated Kaldi speech recognition toolkit for yourself, you will surely wish to learn more about how NVIDIA drastically increased its throughput. Check out this talk from GTC to see the thought process and considerations behind the accelerated implementation.

But what’s the NVIDIA team working on next? One item is optimizing even more stages of Kaldi’s processing flow, now that the feature extractor is complete. Another task is testing Kaldi on smaller GPUs, like Jetson Xavier and Nano development boards. Stay tuned!

No Comments