Evolving Attention with Residual Convolutions

  • Yujing Wang ,
  • Yaming Yang ,
  • Jiangang Bai ,
  • Mingliang Zhang ,
  • Jing Bai ,
  • Jing Yu ,
  • Ce Zhang ,
  • Gao Huang ,
  • Yunhai Tong

2021 International Conference on Machine Learning |

Publication

Transformer is a ubiquitous model for natural language processing and has attracted wide attentions in computer vision. The attention maps are indispensable for a transformer model to encode the dependencies among input tokens. However, they are learned independently in each layer and sometimes fail to capture precise patterns. In this paper, we propose a novel and generic mechanism based on evolving attention to improve the performance of transformers. On one hand, the attention maps in different layers share common knowledge, thus the ones in preceding layers can instruct the attention in succeeding layers through residual connections. On the other hand, low-level and high-level attentions vary in the level of abstraction, so we adopt convolutional layers to model the evolutionary process of attention maps. The proposed evolving attention mechanism achieves significant performance improvement over various state-of-the-art models for multiple tasks, including image classification, natural language understanding and machine translation.