Do Transformers Need Deep Long-Range Memory?

Jack Rae, Ali Razavi


Abstract
Deep attention models have advanced the modelling of sequential data across many domains. For language modelling in particular, the Transformer-XL — a Transformer augmented with a long-range memory of past activations — has been shown to be state-of-the-art across a variety of well-studied benchmarks. The Transformer-XL incorporates a long-range memory at every layer of the network, which renders its state to be thousands of times larger than RNN predecessors. However it is unclear whether this is necessary. We perform a set of interventions to show that comparable performance can be obtained with 6X fewer long range memories and better performance can be obtained by limiting the range of attention in lower layers of the network.
Anthology ID:
2020.acl-main.672
Volume:
Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics
Month:
July
Year:
2020
Address:
Online
Editors:
Dan Jurafsky, Joyce Chai, Natalie Schluter, Joel Tetreault
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
7524–7529
Language:
URL:
https://aclanthology.org/2020.acl-main.672
DOI:
10.18653/v1/2020.acl-main.672
Bibkey:
Cite (ACL):
Jack Rae and Ali Razavi. 2020. Do Transformers Need Deep Long-Range Memory?. In Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics, pages 7524–7529, Online. Association for Computational Linguistics.
Cite (Informal):
Do Transformers Need Deep Long-Range Memory? (Rae & Razavi, ACL 2020)
Copy Citation:
PDF:
https://preview.aclanthology.org/naacl24-info/2020.acl-main.672.pdf
Video:
 http://slideslive.com/38928897