English

Highway Transformer: Self-Gating Enhanced Self-Attentive Networks

Computation and Language 2020-11-25 v5 Machine Learning

Abstract

Self-attention mechanisms have made striking state-of-the-art (SOTA) progress in various sequence learning tasks, standing on the multi-headed dot product attention by attending to all the global contexts at different locations. Through a pseudo information highway, we introduce a gated component self-dependency units (SDU) that incorporates LSTM-styled gating units to replenish internal semantic importance within the multi-dimensional latent space of individual representations. The subsidiary content-based SDU gates allow for the information flow of modulated latent embeddings through skipped connections, leading to a clear margin of convergence speed with gradient descent algorithms. We may unveil the role of gating mechanism to aid in the context-based Transformer modules, with hypothesizing that SDU gates, especially on shallow layers, could push it faster to step towards suboptimal points during the optimization process.

Keywords

Cite

@article{arxiv.2004.08178,
  title  = {Highway Transformer: Self-Gating Enhanced Self-Attentive Networks},
  author = {Yekun Chai and Shuo Jin and Xinwen Hou},
  journal= {arXiv preprint arXiv:2004.08178},
  year   = {2020}
}

Comments

Accepted at ACL 2020

R2 v1 2026-06-23T14:55:06.478Z