← 返回资源分享
Reformer: The Efficient Transformer
论文
论文
发布时间2020-01-13
发表ICLR 2020 1 · arXiv:2001.04451
作者:Łukasz Kaiser,Nikita Kitaev,Anselm Levskaya
详细介绍
Large Transformer models routinely achieve state-of-the-art results on a number of tasks but training these models can be prohibitively costly, especially on long sequences. We introduce two techniques to improve the efficiency of Transformers. For one, we replace dot-product attention by one that uses locality-sensitive hashing, changing its complexity from O($L^2$) to O($L\log L$), where $L$ is the length of the sequence. Furthermore, we use reversible residual layers instead of the standard residuals, which allows storing activations only once in the training process instead of $N$ times, where $N$ is the number of layers. The resulting model, the Reformer, performs on par with Transformer models while being much more memory-efficient and much faster on long sequences.
代码仓库 (16)
google/trax/tree/master/trax/models/reformer官方JAX
huggingface/transformersPyTorch
lucidrains/reformer-pytorchPyTorch
kiss2smiles/nlp_reading
lucashueda/long_sentence_transformerPyTorch
junnyu/paddle_reformerPyTorch
t-gappy/polygen_pytorchPyTorch
sliao-mi-luku/NLP-Chatbot-Reformer-TraxPyTorch
Rick-McCoy/Reformer-pytorchPyTorch
lucidrains/DALLE-pytorchPyTorch
