arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.08046cs.LGcs.CL

Tree-structured Attention with Hierarchical Accumulation

Salesforce Research(Salesforce研究院);Nanyang Technological University(南洋理工大学)

Xuan-Phi Nguyen, Shafiq Joty, Steven C. H. Hoi, Richard Socher

英文摘要:

Incorporating hierarchical structures like constituency trees has been shown to be effective for various natural language processing (NLP) tasks. However, it is evident that state-of-the-art (SOTA) sequence-based models like the Transformer struggle to encode such structures inherently. On the other hand, dedicated models like the Tree-LSTM, while explicitly modeling hierarchical structures, do not perform as efficiently as the Transformer. In this paper, we attempt to bridge this gap with "Hierarchical Accumulation" to encode parse tree structures into self-attention at constant time complexity. Our approach outperforms SOTA methods in four IWSLT translation tasks and the WMT'14 English-German translation task. It also yields improvements over Transformer and Tree-LSTM on three text classification tasks. We further demonstrate that using hierarchical priors can compensate for data shortage, and that our model prefers phrase-level attentions over token-level attentions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑