← 返回资源分享
Linear Transformers Are Secretly Fast Weight Programmers
论文
论文
发布时间2021-02-22
发表arXiv:2102.11174
作者:Kazuki Irie,Jürgen Schmidhuber,Imanol Schlag
详细介绍
We show the formal equivalence of linearised self-attention mechanisms and fast weight controllers from the early '90s, where a ``slow" neural net learns by gradient descent to program the ``fast weights" of another net through sequences of elementary programming instructions which are additive outer products of self-invented activation patterns (today called keys and values). Such Fast Weight Programmers (FWPs) learn to manipulate the contents of a finite memory and dynamically interact with it. We infer a memory capacity limitation of recent linearised softmax attention variants, and replace the purely additive outer products by a delta rule-like programming instruction, such that the FWP can more easily learn to correct the current mapping from keys to values. The FWP also learns to compute dynamically changing learning rates. We also propose a new kernel function to linearise attention which balances simplicity and effectiveness. We conduct experiments on synthetic retrieval problems as well as standard machine translation and language modelling tasks which demonstrate the benefits of our methods.
代码仓库 (10)
ischlag/fast-weight-transformers官方PyTorch
IDSIA/recurrent-fwpPyTorch
i404788/DPFP-pytorchPyTorch
IDSIA/lmtool-fwpPyTorch
labmlai/annotated_deep_learning_paper_implementationsPyTorch
IDSIA/lmtool-fwmsPyTorch
idsia/modern-srwmPyTorch
idsia/neuraldiffeq-fwpPyTorch
idsia/fpainterPyTorch
sustcsonglin/flash-linear-attentionPyTorch
