A Comparison of Transformer and LSTM Encoder Decoder Models for ASR
Authors:
Top Cited Papers
- 1 December 2019
- conference paper
- Published by Institute of Electrical and Electronics Engineers (IEEE)
- p. 8-15
Abstract
We present competitive results using a Transformer encoder-decoder-attention model for end-to-end speech recognition needing less training time compared to a similarly performing LSTM model. We observe that the Transformer training is in general more stable compared to the LSTM, although it also seems to overfit more, and thus shows more problems with generalization. We also find that two initial LSTM layers in the Transformer encoder provide a much better positional encoding. Data-augmentation, a variant of SpecAugment, helps to improve both the Transformer by 33% and the LSTM by 15% relative. We analyze several pretraining and scheduling schemes, which is crucial for both the Transformer and the LSTM models. We improve our LSTM model by additional convolutional layers. We perform our experiments on Lib-riSpeech 1000h, Switchboard 300h and TED-LIUM-v2 200h, and we show state-of-the-art performance on TED-LIUM-v2 for attention based end-to-end models. We deliberately limit the training on LibriSpeech to 12.5 epochs of the training data for comparisons, to keep the results of practical interest, although we show that longer training time still improves more. We publish all the code and setups to run our experiments.Keywords
This publication has 31 references indexed in Scilit:
- On integrating a language model into neural machine translationComputer Speech & Language, 2017
- In-Datacenter Performance Analysis of a Tensor Processing Unit2017
- End-to-end attention-based large vocabulary speech recognition2016
- A Decomposable Attention Model for Natural Language Inference2016
- Long Short-Term Memory-Networks for Machine Reading2016
- Modeling Coverage for Neural Machine Translation2016
- Learning Phrase Representations using RNN Encoder–Decoder for Statistical Machine Translation2014
- Curriculum learning2009
- Long Short-Term MemoryNeural Computation, 1997
- An application of recurrent nets to phone probability estimationIEEE Transactions on Neural Networks, 1994