arXiv · 1912.11959
Is Attention All What You Need? -- An Empirical Investigation on Convolution-Based Active Memory and Self-Attention
Abstract
The key to a Transformer model is the self-attention mechanism, which allows the model to analyze an entire sequence in a computationally efficient manner. Recent work has suggested the possibility that general attention mechanisms used by RNNs could be replaced by active-memory mechanisms. In this work, we evaluate whether various active-memory mechanisms could replace self-attention in a Transformer. Our experiments suggest that active-memory alone achieves comparable results to the self-attention mechanism for language modelling, but optimal results are mostly achieved by using both active-memory and self-attention mechanisms together. We also note that, for some specific algorithmic tasks, active-memory mechanisms alone outperform both self-attention and a combination of the two.
Explore related subjects
Keep this discovery
Thomas Dowdell, Hongyu Zhang. 2019-12-27. Is Attention All What You Need? -- An Empirical Investigation on Convolution-Based Active Memory and Self-Attention. https://arxiv.org/abs/1912.11959
Cite the original work for its findings. Save a collection to share your selection of sources.