about
Faster Autoregressive Transformers with Linear Attention (arxiv.org)
6 points by fofoz on Jul 3, 2020 | hide | past | pdf | discuss on HN

In plain words: Attention compares every word to all others, so cost grows as length squared; this rewrite reorders the math so it grows linearly. It works like a recurrent network keeping a running summary, with accuracy similar to transformers' while up to 4000x faster on long sequences.

Abstract · Transformers are RNNs: Fast Autoregressive Transformers with Linear Attention

Transformers achieve remarkable performance in several tasks but due to their quadratic complexity, with respect to the input's length, they are prohibitively slow for very long sequences. To address this limitation, we express the self-attention as a linear dot-product of kernel feature maps and make use of the associativity property of matrix products to reduce the complexity from $\mathcal{O}\left(N^2\right)$ to $\mathcal{O}\left(N\right)$, where $N$ is the sequence length. We show that this formulation permits an iterative implementation that dramatically accelerates autoregressive transformers and reveals their relationship to recurrent neural networks. Our linear transformers achieve similar performance to vanilla transformers and they are up to 4000x faster on autoregressive prediction of very long sequences.

Angelos Katharopoulos, Apoorv Vyas, Nikolaos Pappas, François Fleuret
arXiv:2006.16236 · cs.LG, stat.ML · submitted Jun 29, 2020 · updated Aug 31, 2020
abstract · pdf · html · ICML 2020, project at https://linear-transformers.com/

add comment on HN