References & Citations
Computer Science > Computation and Language
Title: SimpleTRON: Simple Transformer with O(N) Complexity
(Submitted on 23 Nov 2021 (v1), last revised 28 Jun 2022 (this version, v4))
Abstract: In this paper, we propose that the dot product pairwise matching attention layer, which is widely used in Transformer-based models, is redundant for the model performance. Attention, in its original formulation, has to be seen rather as a human-level tool to explore and/or visualize relevancy scores in sequential data. However, the way how it is constructed leads to significant computational complexity. Instead, we present SimpleTRON: Simple Transformer with O(N) Complexity, a simple and fast alternative without any approximation that, unlike other approximation models, does not have any architecture-related overhead and therefore can be seen as a purely linear Transformer-like model. This architecture, to the best of our knowledge, outperforms existing sub-quadratic attention approximation models on several tasks from the Long-Range Arena benchmark. Moreover, we show, that SimpleTRON can benefit from weight transfer from pretrained large language models, as its parameters can be fully transferable.
Submission history
From: Alexander Kovalenko [view email][v1] Tue, 23 Nov 2021 17:06:01 GMT (47kb,D)
[v2] Wed, 1 Dec 2021 09:45:25 GMT (106kb,D)
[v3] Thu, 2 Dec 2021 08:16:33 GMT (106kb,D)
[v4] Tue, 28 Jun 2022 13:18:03 GMT (666kb,D)
Link back to: arXiv, form interface, contact.