Analysis on the Number of Layers in the Transformer-Based Model for Neural Machine Translation

Dongxing Li,Zuying Luo
DOI: https://doi.org/10.1117/12.2661982
2022-01-01
Abstract:Recently, the transformer-based models have been widely used in sequence-to-sequence (seq2seq) tasks, especially neural machine translation (NMT). In the original transformer, the layer number in encoder is equal to the layer number in decoder. However, the structure is more complex and task is more difficult in decoder than those in encoder, so the layer number should not be same. In order to verify how many layer number in encoder and decoder is properly valued, we improve transformer as our model and conduct four experiments on four translation tasks of IWSLT2017. The experimental results show that the layer number in decoder should be larger than that in encoder, which can bring better translation performance.
What problem does this paper attempt to address?