Redesigning the Transformer Architecture with Insights from
  Multi-particle Dynamical Systems
  The Transformer and its variants have been proven to be efficient sequence
learners in many different domains. Despite their staggering success, a
critical issue has been the enormous number of parameters that must be trained
(ranging from $10^7$ to $10^{11}$) along with the quadratic complexity of
dot-product attention. In this work, we investigate the problem of
approximating the two central components of the Transformer -- multi-head
self-attention and point-wise feed-forward transformation, with reduced
parameter space and computational complexity. We build upon recent developments
in analyzing deep neural networks as numerical solvers of ordinary differential
equations. Taking advantage of an analogy between Transformer stages and the
evolution of a dynamical system of multiple interacting particles, we formulate
a temporal evolution scheme, TransEvolve, to bypass costly dot-product
attention over multiple stacked layers. We perform exhaustive experiments with
TransEvolve on well-known encoder-decoder as well as encoder-only tasks. We
observe that the degree of approximation (or inversely, the degree of parameter
reduction) has different effects on the performance, depending on the task.
While in the encoder-decoder regime, TransEvolve delivers performances
comparable to the original Transformer, in encoder-only tasks it consistently
outperforms Transformer along with several subsequent variants.
