Transformer-based Online Speech Recognition with Decoder-end Adaptive Computation Steps
arXiv:2011.13834
Abstract
Transformer-based end-to-end (E2E) automatic speech recognition (ASR) systems have recently gained wide popularity, and are shown to outperform E2E models based on recurrent structures on a number of ASR tasks. However, like other E2E models, Transformer ASR also requires the full input sequence for calculating the attentions on both encoder and decoder, leading to increased latency and posing a challenge for online ASR. The paper proposes Decoder-end Adaptive Computation Steps (DACS) algorithm to address the issue of latency and facilitate online ASR. The proposed algorithm streams the decoding of Transformer ASR by triggering an output after the confidence acquired from the encoder states reaches a certain threshold. Unlike other monotonic attention mechanisms that risk visiting the entire encoder states for each output step, the paper introduces a maximum look-ahead step into the DACS algorithm to prevent from reaching the end of speech too fast. A Chunkwise encoder is adopted in our system to handle real-time speech inputs. The proposed online Transformer ASR system has been evaluated on Wall Street Journal (WSJ) and AIShell-1 datasets, yielding 5.5% word error rate (WER) and 7.1% character error rate (CER) respectively, with only a minor decay in performance when compared to the offline systems.
arXiv comment7 pages, 1 figure, accepted at SLT 2021
References in corpus (11)
- Adam: A Method for Stochastic Optimization
- SpecAugment: A Simple Data Augmentation Method for Automatic Speech Recognition
- Attention-Based Models for Speech Recognition
- Sequence Transduction with Recurrent Neural Networks
- A Comparative Study on Transformer vs RNN in Speech Applications
- Effective Approaches to Attention-based Neural Machine Translation
- Adaptive Computation Time for Recurrent Neural Networks
- Self-Attention Transducers for End-to-End Speech Recognition
- Online Hybrid CTC/Attention End-to-End Automatic Speech Recognition Architecture
- Towards Online End-to-end Transformer Automatic Speech Recognition
- Multi-head Monotonic Chunkwise Attention For Online Speech Recognition