Folding Attention: Memory and Power Optimization for On-Device Transformer-based Streaming Speech Recognition
Abstract
Transformer-based models excel in speech recognition. Existing efforts to optimize Transformer inference, typically for long-context applications, center on simplifying attention score calculations. However, streaming speech recognition models usually process a limited number of tokens each time, making attention score calculation less of a bottleneck. Instead, the bottleneck lies in the linear projection layers of multi-head attention and feedforward networks, constituting a substantial portion of the model size and contributing significantly to computation, memory, and power usage. To address this bottleneck, we propose folding attention, a technique targeting these linear layers, significantly reducing model size and improving memory and power efficiency. Experiments on on-device Transformer-based streaming speech recognition models show that folding attention reduces model size (and corresponding memory consumption) by up to 24% and power consumption by up to 23%, all without compromising model accuracy or computation overhead.
- Publication:
-
arXiv e-prints
- Pub Date:
- September 2023
- DOI:
- 10.48550/arXiv.2309.07988
- arXiv:
- arXiv:2309.07988
- Bibcode:
- 2023arXiv230907988L
- Keywords:
-
- Computer Science - Machine Learning;
- Computer Science - Hardware Architecture;
- Computer Science - Sound;
- Electrical Engineering and Systems Science - Audio and Speech Processing