We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computation and Language

Title: SimpleTRON: Simple Transformer with O(N) Complexity

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.
Subjects: Computation and Language (cs.CL)
Cite as: arXiv:2111.15588 [cs.CL]
  (or arXiv:2111.15588v4 [cs.CL] for this version)

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.