FMLLR
In signal processing, Feature space Maximum Likelihood Linear Regression 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.
fMLLR transformations are trained in a maximum likelihood sense on adaptation data. These transformations may be estimated in many ways, but only maximum likelihood 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,
later research also shows that fMLLR is an excellent acoustic feature for DNN/HMM hybrid speech recognition models.
The advantage of fMLLR includes the following:
- the adaptation process can be performed within a pre-precessing phase, and is independent of the ASR training and decoding process.
- this type of adapted feature can be applied to deep neural networks to replace traditionally used mel-spectrogram in end-to-end speech recognition models.
- fMLLR's speaker adaptation process leads to a significant performance boost for ASR models, hence outperforming other transform or features like MFCCs and FBANKs coefficients.
- fMLLR features can be efficiently realized with speech toolkits like Kaldi.
- when the amount of adaptation data is limited, the transformation matrices tends to easily overfit the given data.
Computing fMLLR transform
In the Kaldi formulation, fMLLR is an affine feature transform of the form →, which can be written in the form →W, where = is the acoustic feature with a 1 appended. Note that this differs from some of the literature where the 1 comes first as =.
The sufficient statistics stored are:
where is the inverse co-variance matrix.
And for where is the feature dimension:
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.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 is reported for the test set of TIMIT with various neural architectures:
Models/Features | MFCC | FBANK | fMLLR |
MLP | 18.2 | 18.7 | 16.7 |
RNN | 17.7 | 17.2 | 15.9 |
LSTM | 15.1 | 14.3 | 14.5 |
GRU | 16.0 | 15.2 | 14.9 |
Li-GRU | 15.3 | 14.9 | 14.2 |
As expected, fMLLR features outperform MFCCs and FBANKs coefficients despite the use of different model architecture.
Where MLP serves as a simple baseline, on the other hand RNN, LSTM, and GRU are all well known recurrent models.
The Li-GRU 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 .
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.
- These instruction are based on the codes provided in this , 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. - Install Kaldi.
- Install .
- If running on a single machine, change the following lines in
$KALDI_ROOT/egs/librispeech/s5/cmd.sh
to replacequeue.pl
torun.pl
:
export decode_cmd="run.pl --mem 4G"
export mkgraph_cmd="run.pl --mem 8G"
- Change the
data
path inrun.sh
to your LibriSpeech data path, the directoryLibriSpeech/
should be under that path. For example:
- Install
flac
with:sudo apt-get install flac
- Run the Kaldi recipe
run.sh
for LibriSpeech at least until Stage 13, for simplicity you can used the modified . - Copy
exp/tri4b/trans.*
files intoexp/tri4b/decode_tgsmall_train_clean_*/
with the following command:
- Compute the fMLLR features by running the following script, the script can also be downloaded :
- !/bin/bash
../path.sh ## Source the tools/utils
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.ark
done
- Compute alignments using:
- alignments on dev_clean and test_clean
steps/align_fmllr.sh --nj 10 data/test_clean data/lang exp/tri4b exp/tri4b_ali_test_clean
steps/align_fmllr.sh --nj 30 data/train_clean_100 data/lang exp/tri4b exp/tri4b_ali_clean_100
steps/align_fmllr.sh --nj 30 data/train_clean_360 data/lang exp/tri4b exp/tri4b_ali_clean_360
- Apply CMVN and dump the fMLLR features to new.ark files, the script can also be downloaded :
- !/bin/bash
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.ark
done
du -sh $data/fmllr_cmvn/*
echo "Done!"
- Use the python script to convert Kaldi generated.ark features to.npy for your own dataloader, an example is provided: