HyperAIHyperAI

Command Palette

Search for a command to run...

Transformers are RNNs: Fast Autoregressive Transformers with Linear Attention

Angelos Katharopoulos Apoorv Vyas Nikolaos Pappas François Fleuret

Abstract

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 O(N2)\mathcal{O}\left(N^2\right)O(N2) to O(N)\mathcal{O}\left(N\right)O(N), where NNN 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.


Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing

HyperAI Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Transformers are RNNs: Fast Autoregressive Transformers with Linear Attention | Papers | HyperAI