Relaxed Attention: A Simple Method to Boost Performance of End-to-End Automatic Speech Recognition

2 Jul 2021  ·  Timo Lohrenz, Patrick Schwarz, Zhengyang Li, Tim Fingscheidt ·

Recently, attention-based encoder-decoder (AED) models have shown high performance for end-to-end automatic speech recognition (ASR) across several tasks. Addressing overconfidence in such models, in this paper we introduce the concept of relaxed attention, which is a simple gradual injection of a uniform distribution to the encoder-decoder attention weights during training that is easily implemented with two lines of code. We investigate the effect of relaxed attention across different AED model architectures and two prominent ASR tasks, Wall Street Journal (WSJ) and Librispeech. We found that transformers trained with relaxed attention outperform the standard baseline models consistently during decoding with external language models. On WSJ, we set a new benchmark for transformer-based end-to-end speech recognition with a word error rate of 3.65%, outperforming state of the art (4.20%) by 13.1% relative, while introducing only a single hyperparameter.

PDF Abstract

Datasets


Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Benchmark
Speech Recognition LibriSpeech test-other Conformer with Relaxed Attention Word Error Rate (WER) 6.85 # 34
Speech Recognition WSJ eval92 Transformer with Relaxed Attention Word Error Rate (WER) 3.19 # 7

Methods