Transformer with Layer Fusion and Interaction
Yan Mingming, You Shuai, Sun Jianing, Xiang Gao, Biao Lu, Wenxian Huang · 2022 19th International Computer Conference on Wavelet Active Media Technology and Information Processing (ICCWAMTIP) · 2022
Neural machine translation models, such as Recurrent Neural Networks, Long Short-Term Memory networks and the Transformer, are widely used in many translation tasks. Neural machine translation models often use the deep encoder and decoder as multiple layers to capture more semantic information. However, these models leverage the encoder top layer and decoder only when they process a sequence, which omits other layers’ useful information. This paper, we propose an approach that includes two strategies. First, we propose a strategy called layer fusion to make full use of all layers. Then, we make the hierarchical interaction between the encoder and the decoder by using the output of each encoder layer as the part input of the corresponding decoder layer. We take the Transformer as an example in this work since it has achieved the best performance. The experimental results on the English and German translation data of WMT14 demonstrate the effectiveness of the proposed approach, and it is also applicable to other models.