Title
Dynamic Layer Aggregation for Neural Machine Translation with Routing-by-Agreement
Abstract
With the promising progress of deep neural networks, layer aggregation has been used to fuse information across layers in various fields, such as computer vision and machine translation. However, most of the previous methods combine layers in a static fashion in that their aggregation strategy is independent of specific hidden states. Inspired by recent progress on capsule networks, in this paper we propose to use routing-by-agreement strategies to aggregate layers dynamically. Specifically, the algorithm learns the probability of a part (individual layer representations) assigned to a whole (aggregated representations) in an iterative way and combines parts accordingly. We implement our algorithm on top of the state-of-the-art neural machine translation model TRANSFORMER and conduct experiments on the widely-used WMT14 English double right arrow German and WMT17 Chinese double right arrow English translation datasets. Experimental results across language pairs show that the proposed approach consistently outperforms the strong baseline model and a representative static aggregation model.
Year
DOI
Venue
2019
10.1609/aaai.v33i01.330186
national conference on artificial intelligence
Field
DocType
Volume
Computer science,Machine translation,Transformer,Artificial intelligence,Fuse (electrical),Machine learning,Deep neural networks
Journal
33
Citations 
PageRank 
References 
1
0.35
21
Authors
6
Name
Order
Citations
PageRank
Zi-Yi Dou1207.01
Zhaopeng Tu251839.95
Xing Wang35810.07
Longyue Wang47218.24
Shuming Shi562058.27
Zhang, Tong67126611.43