Multi-mode Transformer Transducer with Stochastic Future Context
arXiv:2106.09760
Abstract
Automatic speech recognition (ASR) models make fewer errors when more surrounding speech information is presented as context. Unfortunately, acquiring a larger future context leads to higher latency. There exists an inevitable trade-off between speed and accuracy. Naively, to fit different latency requirements, people have to store multiple models and pick the best one under the constraints. Instead, a more desirable approach is to have a single model that can dynamically adjust its latency based on different constraints, which we refer to as Multi-mode ASR. A Multi-mode ASR model can fulfill various latency requirements during inference -- when a larger latency becomes acceptable, the model can process longer future context to achieve higher accuracy and when a latency budget is not flexible, the model can be less dependent on future context but still achieve reliable accuracy. In pursuit of Multi-mode ASR, we propose Stochastic Future Context, a simple training procedure that samples one streaming configuration in each iteration. Through extensive experiments on AISHELL-1 and LibriSpeech datasets, we show that a Multi-mode ASR model rivals, if not surpasses, a set of competitive streaming baselines trained with different latency budgets.
arXiv commentAccepted to Interspeech 2021
References in corpus (12)
- SpecAugment: A Simple Data Augmentation Method for Automatic Speech Recognition
- Gaussian Error Linear Units (GELUs)
- Sequence Transduction with Recurrent Neural Networks
- Conformer: Convolution-augmented Transformer for Speech Recognition
- Listen, Attend and Spell
- Online and Linear-Time Attention by Enforcing Monotonic Alignments
- Pushing the Limits of Semi-Supervised Learning for Automatic Speech Recognition
- Transformer-Transducer: End-to-End Speech Recognition with Self-Attention
- A Neural Transducer
- Monotonic Chunkwise Attention
- Transformer Transducer: One Model Unifying Streaming and Non-streaming Speech Recognition
- Universal ASR: Unifying Streaming and Non-Streaming ASR Using a Single Encoder-Decoder Model