EIT: Enhanced Interactive Transformer

Tong Zheng,Bei Li,Huiwen Bao,Tong Xiao,Jingbo Zhu
2024-06-05
Abstract:Two principles: the complementary principle and the consensus principle are widely acknowledged in the literature of multi-view learning. However, the current design of multi-head self-attention, an instance of multi-view learning, prioritizes the complementarity while ignoring the consensus. To address this problem, we propose an enhanced multi-head self-attention (EMHA). First, to satisfy the complementary principle, EMHA removes the one-to-one mapping constraint among queries and keys in multiple subspaces and allows each query to attend to multiple keys. On top of that, we develop a method to fully encourage consensus among heads by introducing two interaction models, namely inner-subspace interaction and cross-subspace interaction. Extensive experiments on a wide range of language tasks (e.g., machine translation, abstractive summarization and grammar correction, language modeling), show its superiority, with a very modest increase in model size. Our code would be available at: <a class="link-external link-https" href="https://github.com/zhengkid/EIT-Enhanced-Interactive-Transformer" rel="external noopener nofollow">this https URL</a>.
Computation and Language
What problem does this paper attempt to address?