FMLLR explained

In signal processing, Feature space Maximum Likelihood Linear Regression (fMLLR) is a global feature transform that are typically applied in a speaker adaptive way, where fMLLR transforms acoustic features to speaker adapted features by a multiplication operation with a transformation matrix. In some literature, fMLLR is also known as the Constrained Maximum Likelihood Linear Regression (cMLLR).

Overview

fMLLR transformations are trained in a maximum likelihood sense on adaptation data. These transformations may be estimated in many ways, but only maximum likelihood (ML) estimation is considered in fMLLR. The fMLLR transformation is trained on a particular set of adaptation data, such that it maximizes the likelihood of that adaptation data given a current model-set.

This technique is a widely used approach for speaker adaptation in HMM-based speech recognition.[1] [2] Later research[3] also shows that fMLLR is an excellent acoustic feature for DNN/HMM[4] hybrid speech recognition models.

The advantage of fMLLR includes the following:

Major problem and disadvantage of fMLLR:

Computing fMLLR transform

Feature transform of fMLLR can be easily computed with the open source speech tool Kaldi, the Kaldi script uses the standard estimation scheme described in Appendix B of the original paper, in particular the section Appendix B.1 "Direct method over rows".

In the Kaldi formulation, fMLLR is an affine feature transform of the form

x

A

x

+b

, which can be written in the form

x

→W

\hat{x}

, where

\hat{x}

=

\begin{bmatrix}x\ 1\end{bmatrix}

is the acoustic feature

x

with a 1 appended. Note that this differs from some of the literature where the 1 comes first as

\hat{x}

=

\begin{bmatrix}1\x\end{bmatrix}

.

The sufficient statistics stored are:

K=\sumt,j,m\gammaj,m

-1
(t)style\Sigma
jm

\mujmx(t)+\displaystyle

where

-1
style\Sigma
jm

\displaystyle

is the inverse co-variance matrix.

And for

0\leqi\leqD

where

D

is the feature dimension:

G(i)=\sumt,j,m\gammaj,m(t)\left(

1
2
\sigma(i)
j,m

\right)x(t)+x(t)+T\displaystyle

For a thorough review that explains fMLLR and the commonly used estimation techniques, see the original paper "Maximum likelihood linear transformations for HMM-based speech recognition ".

Note that the Kaldi script that performs the feature transforms of fMLLR differs with by using a column of the inverse in place of the cofactor row. In other words, the factor of the determinant is ignored, as it does not affect the transform result and can causes potential danger of numerical underflow or overflow.

Comparing with other features or transforms

Experiment result shows that by using the fMLLR feature in speech recognition, constant improvement is gained over other acoustic features on various commonly used benchmark datasets (TIMIT, LibriSpeech, etc).

In particular, fMLLR features outperform MFCCs and FBANKs coefficients, which is mainly due to the speaker adaptation process that fMLLR performs.

In, phoneme error rate (PER, %) is reported for the test set of TIMIT with various neural architectures:

PER results obtained from Pytorch-Kaldi!Models/Features!MFCC!FBANK!fMLLR
MLP18.218.716.7
RNN17.717.215.9
LSTM15.114.314.5
GRU16.015.214.9
Li-GRU15.314.914.2
As expected, fMLLR features outperform MFCCs and FBANKs coefficients despite the use of different model architecture.

Where MLP (multi-layer perceptron) serves as a simple baseline, on the other hand RNN, LSTM, and GRU are all well known recurrent models.

The Li-GRU[5] architecture is based on a single gate and thus saves 33% of the computations over a standard GRU model, Li-GRU thus effectively address the gradient vanishing problem of recurrent models.

As a result, the best performance is obtained with the Li-GRU model on fMLLR features.

Extract fMLLR features with Kaldi

fMLLR can be extracted as reported in the s5 recipe of Kaldi.

Kaldi scripts can certainly extract fMLLR features on different dataset, below are the basic example steps to extract fMLLR features from the open source speech corpora Librispeech.

Note that the instructions below are for the subsets train-clean-100,train-clean-360,dev-clean, and test-clean,

but they can be easily extended to support the other sets dev-other, test-other, and train-other-500.

  1. These instruction are based on the codes provided in this GitHub repository, which contains Kaldi recipes on the LibriSpeech corpora to execute the fMLLR feature extraction process, replace the files under $KALDI_ROOT/egs/librispeech/s5/ with the files in the repository.
  2. Install Kaldi.
  3. Install Kaldiio.
  4. If running on a single machine, change the following lines in $KALDI_ROOT/egs/librispeech/s5/cmd.sh to replace queue.pl to run.pl:

export train_cmd="run.pl --mem 2G"export decode_cmd="run.pl --mem 4G"export mkgraph_cmd="run.pl --mem 8G"

  1. Change the data path in run.sh to your LibriSpeech data path, the directory LibriSpeech/ should be under that path. For example:

data=/media/user/SSD # example path

  1. Install flac with: sudo apt-get install flac
  2. Run the Kaldi recipe run.sh for LibriSpeech at least until Stage 13 (included), for simplicity you can use the modified run.sh.
  3. Copy exp/tri4b/trans.* files into exp/tri4b/decode_tgsmall_train_clean_*/ with the following command:

mkdir exp/tri4b/decode_tgsmall_train_clean_100 && cp exp/tri4b/trans.* exp/tri4b/decode_tgsmall_train_clean_100/

  1. Compute the fMLLR features by running the following script, the script can also be downloaded here:
  2. !/bin/bash

. ./cmd.sh ## You'll want to change cmd.sh to something that will work on your system.. ./path.sh ## Source the tools/utils (import the queue.pl)

gmmdir=exp/tri4b

for chunk in dev_clean test_clean train_clean_100 train_clean_360 ; do dir=fmllr/$chunk steps/nnet/make_fmllr_feats.sh --nj 10 --cmd "$train_cmd" \ --transform-dir $gmmdir/decode_tgsmall_$chunk \ $dir data/$chunk $gmmdir $dir/log $dir/data || exit 1

compute-cmvn-stats --spk2utt=ark:data/$chunk/spk2utt scp:fmllr/$chunk/feats.scp ark:$dir/data/cmvn_speaker.arkdone

  1. Compute alignments using:
  2. alignments on dev_clean and test_clean

steps/align_fmllr.sh --nj 10 data/dev_clean data/lang exp/tri4b exp/tri4b_ali_dev_cleansteps/align_fmllr.sh --nj 10 data/test_clean data/lang exp/tri4b exp/tri4b_ali_test_cleansteps/align_fmllr.sh --nj 30 data/train_clean_100 data/lang exp/tri4b exp/tri4b_ali_clean_100steps/align_fmllr.sh --nj 30 data/train_clean_360 data/lang exp/tri4b exp/tri4b_ali_clean_360

  1. Apply CMVN and dump the fMLLR features to new .ark files, the script can also be downloaded here:
  2. !/bin/bash

data=/user/kaldi/egs/librispeech/s5 ## You'll want to change this path to something that will work on your system.

rm -rf $data/fmllr_cmvn/mkdir $data/fmllr_cmvn/

for part in dev_clean test_clean train_clean_100 train_clean_360; do mkdir $data/fmllr_cmvn/$part/ apply-cmvn --utt2spk=ark:$data/fmllr/$part/utt2spk ark:$data/fmllr/$part/data/cmvn_speaker.ark scp:$data/fmllr/$part/feats.scp ark:- | add-deltas --delta-order=0 ark:- ark:$data/fmllr_cmvn/$part/fmllr_cmvn.arkdone

du -sh $data/fmllr_cmvn/*echo "Done!"

  1. Use the Python script to convert Kaldi generated .ark features to .npy for your own dataloader, an example Python script is provided:

python ark2libri.py

See also

Notes and References

  1. M.J.F. Gales . 1998 . Maximum likelihood linear transformations for HMM-based speech recognition . Computer Speech & Language . 12 . 2 . 75–98 . 10.1006/csla.1998.0043 . 10.1.1.37.8252 .
  2. Rapid Feature Space Speaker Adaptation for Multi-Stream HMM-Based Audio-Visual Speech Recognition . Jing Huang . E Marcheret . K Visweswariah . 2005 . IEEE . 338–341 . IEEE International Conference on Multimedia and Expo (ICME) . 10.1109/ICME.2005.1521429 .
  3. Ravanelli. Mirco. Parcollet. Titouan. Bengio. Yoshua. 2018-11-18. The PyTorch-Kaldi Speech Recognition Toolkit. 1811.07453. eess.AS.
  4. Book: Li. Longfei. Zhao. Yong. Jiang. Dongmei. Zhang. Yanning. Wang. Fengna. Gonzalez. Isabel. Valentin. Enescu. Sahli. Hichem. 2013 Humaine Association Conference on Affective Computing and Intelligent Interaction . Hybrid Deep Neural Network--Hidden Markov Model (DNN-HMM) Based Speech Emotion Recognition . September 2013. 312–317. IEEE. 10.1109/acii.2013.58. 978-0-7695-5048-0. 9665019.
  5. Ravanelli. Mirco. Brakel. Philemon. Omologo. Maurizio. Bengio. Yoshua. 2017-08-20. Improving Speech Recognition by Revising Gated Recurrent Units. Interspeech 2017. 1308–1312. ISCA. ISCA. 10.21437/interspeech.2017-775. 2017arXiv171000641R. 1710.00641. 1696099.