English

Do Multilingual Neural Machine Translation Models Contain Language Pair Specific Attention Heads?

Computation and Language 2021-06-01 v1 Artificial Intelligence Machine Learning

Abstract

Recent studies on the analysis of the multilingual representations focus on identifying whether there is an emergence of language-independent representations, or whether a multilingual model partitions its weights among different languages. While most of such work has been conducted in a "black-box" manner, this paper aims to analyze individual components of a multilingual neural translation (NMT) model. In particular, we look at the encoder self-attention and encoder-decoder attention heads (in a many-to-one NMT model) that are more specific to the translation of a certain language pair than others by (1) employing metrics that quantify some aspects of the attention weights such as "variance" or "confidence", and (2) systematically ranking the importance of attention heads with respect to translation quality. Experimental results show that surprisingly, the set of most important attention heads are very similar across the language pairs and that it is possible to remove nearly one-third of the less important heads without hurting the translation quality greatly.

Keywords

Cite

@article{arxiv.2105.14940,
  title  = {Do Multilingual Neural Machine Translation Models Contain Language Pair Specific Attention Heads?},
  author = {Zae Myung Kim and Laurent Besacier and Vassilina Nikoulina and Didier Schwab},
  journal= {arXiv preprint arXiv:2105.14940},
  year   = {2021}
}

Comments

10 pages, accepted at Findings of ACL 2021 (short)

R2 v1 2026-06-24T02:39:34.724Z