We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.CL

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computation and Language

Title: Generating Diverse Translation by Manipulating Multi-Head Attention

Abstract: Transformer model has been widely used on machine translation tasks and obtained state-of-the-art results. In this paper, we report an interesting phenomenon in its encoder-decoder multi-head attention: different attention heads of the final decoder layer align to different word translation candidates. We empirically verify this discovery and propose a method to generate diverse translations by manipulating heads. Furthermore, we make use of these diverse translations with the back-translation technique for better data augmentation. Experiment results show that our method generates diverse translations without severe drop in translation quality. Experiments also show that back-translation with these diverse translations could bring significant improvement on performance on translation tasks. An auxiliary experiment of conversation response generation task proves the effect of diversity as well.
Comments: Accepted by AAAI 2020
Subjects: Computation and Language (cs.CL)
Cite as: arXiv:1911.09333 [cs.CL]
  (or arXiv:1911.09333v1 [cs.CL] for this version)

Submission history

From: Zewei Sun [view email]
[v1] Thu, 21 Nov 2019 08:22:07 GMT (2066kb,D)

Link back to: arXiv, form interface, contact.