English

FNetAR: Mixing Tokens with Autoregressive Fourier Transforms

Computation and Language 2021-07-26 v1 Machine Learning

Abstract

In this note we examine the autoregressive generalization of the FNet algorithm, in which self-attention layers from the standard Transformer architecture are substituted with a trivial sparse-uniformsampling procedure based on Fourier transforms. Using the Wikitext-103 benchmark, we demonstratethat FNetAR retains state-of-the-art performance (25.8 ppl) on the task of causal language modelingcompared to a Transformer-XL baseline (24.2 ppl) with only half the number self-attention layers,thus providing further evidence for the superfluity of deep neural networks with heavily compoundedattention mechanisms. The autoregressive Fourier transform could likely be used for parameterreduction on most Transformer-based time-series prediction models.

Keywords

Cite

@article{arxiv.2107.10932,
  title  = {FNetAR: Mixing Tokens with Autoregressive Fourier Transforms},
  author = {Tim Lou and Michael Park and Mohammad Ramezanali and Vincent Tang},
  journal= {arXiv preprint arXiv:2107.10932},
  year   = {2021}
}

Comments

final experimental results forthcoming