Wed-1-5-6 Streaming Transformer-based Acoustic Models Using Self-attention with Augmented Memory

Chunyang Wu(Facebook), Yongqiang Wang(Facebook), Yangyang Shi(Facebook), Ching-Feng Yeh(Facebook Inc.) and Frank Zhang(Facebook)
Abstract: Transformer-based acoustic modeling has achieved great success for both hybrid and sequence-to-sequence speech recognition. However, it requires access to the full sequence, and the computational cost grows quadratically with respect to the input sequence length. These factors limit its adoption for streaming applications. In this work, we proposed a novel augmented memory self-attention, which attends on a short segment of the input sequence and a bank of memories. The memory bank stores the embedding information for all the processed segments. On the librispeech benchmark, our proposed method outperforms all the existing streamable transformer methods by a large margin and achieved over 15% relative error reduction, compared with the widely used LC-BLSTM baseline. Our findings are also confirmed on some large internal datasets.
Student Information

Student Events

Travel Grants