about
Generating Long Sequences with Sparse Transformers (arxiv.org)
3 points by TaurenHunter on Jan 16, 2024 | hide | past | pdf | 1 comment on HN

In plain words: Instead of letting every position look at every other one, each position checks only a small fixed set of others, cutting the computing cost. This let the model handle sequences tens of thousands of steps long and beat the best text and image compression scores.

Abstract

Transformers are powerful sequence models, but require time and memory that grows quadratically with the sequence length. In this paper we introduce sparse factorizations of the attention matrix which reduce this to $O(n \sqrt{n})$. We also introduce a) a variation on architecture and initialization to train deeper networks, b) the recomputation of attention matrices to save memory, and c) fast attention kernels for training. We call networks with these changes Sparse Transformers, and show they can model sequences tens of thousands of timesteps long using hundreds of layers. We use the same architecture to model images, audio, and text from raw bytes, setting a new state of the art for density modeling of Enwik8, CIFAR-10, and ImageNet-64. We generate unconditional samples that demonstrate global coherence and great diversity, and show it is possible in principle to use self-attention to model sequences of length one million or more.

Rewon Child, Scott Gray, Alec Radford, Ilya Sutskever
arXiv:1904.10509 · cs.LG, stat.ML · submitted Apr 23, 2019
abstract · pdf · html

add comment on HN

"... we introduce sparse factorizations of the attention matrix which reduce this to O(nn‾√). We also introduce a) a variation on architecture and initialization to train deeper networks, b) the recomputation of attention matrices to save memory, and c) fast attention kernels for training. We call networks with these changes Sparse Transformers, and show they can model sequences tens of thousands of timesteps long using hundreds of layers. ..."