UM
Residential Collegefalse
Status已發表Published
Information aggregation for multi-head attention with routing-by-agreement
Li, Jian1; Yang, Baosong2; Dou, Zi Yi3; Wang, Xing4; Lyu, Michael R.1; Tu, Zhaopeng4
2019
Conference NameConference of the North-American-Chapter of the Association-for-Computational-Linguistics - Human Language Technologies (NAACL-HLT)
Source PublicationNAACL HLT 2019 - 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies - Proceedings of the Conference
Volume1
Pages3566-3575
Conference DateJUN 02-07, 2019
Conference PlaceMinneapolis, MN
Abstract

Multi-head attention is appealing for its ability to jointly extract different types of information from multiple representation subspaces. Concerning the information aggregation, a common practice is to use a concatenation followed by a linear transformation, which may not fully exploit the expressiveness of multi-head attention. In this work, we propose to improve the information aggregation for multi-head attention with a more powerful routing-by-agreement algorithm. Specifically, the routing algorithm iteratively updates the proportion of how much a part (i.e. the distinct information learned from a specific subspace) should be assigned to a whole (i.e. the final output representation), based on the agreement between parts and wholes. Experimental results on linguistic probing tasks and machine translation tasks prove the superiority of the advanced information aggregation over the standard linear transformation.

URLView the original
Indexed ByCPCI-S ; CPCI-SSH
Language英語English
WOS Research AreaComputer Science ; Linguistics
WOS SubjectComputer Science, Artificial Intelligence ; Computer Science, Interdisciplinary Applications ; Linguistics
WOS IDWOS:000900116903063
Scopus ID2-s2.0-85075572190
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionUniversity of Macau
Corresponding AuthorTu, Zhaopeng
Affiliation1.Department of Computer Science and Engineering, Chinese University of Hong Kong, Hong Kong
2.University of Macau, Macao
3.Carnegie Mellon University, United States
4.Tencent AI Lab,
Recommended Citation
GB/T 7714
Li, Jian,Yang, Baosong,Dou, Zi Yi,et al. Information aggregation for multi-head attention with routing-by-agreement[C], 2019, 3566-3575.
APA Li, Jian., Yang, Baosong., Dou, Zi Yi., Wang, Xing., Lyu, Michael R.., & Tu, Zhaopeng (2019). Information aggregation for multi-head attention with routing-by-agreement. NAACL HLT 2019 - 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies - Proceedings of the Conference, 1, 3566-3575.
Files in This Item:
There are no files associated with this item.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Li, Jian]'s Articles
[Yang, Baosong]'s Articles
[Dou, Zi Yi]'s Articles
Baidu academic
Similar articles in Baidu academic
[Li, Jian]'s Articles
[Yang, Baosong]'s Articles
[Dou, Zi Yi]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Li, Jian]'s Articles
[Yang, Baosong]'s Articles
[Dou, Zi Yi]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.