AdvAug: Robust Adversarial Augmentation for Neural Machine Translation
About
In this paper, we propose a new adversarial augmentation method for Neural Machine Translation (NMT). The main idea is to minimize the vicinal risk over virtual sentences sampled from two vicinity distributions, of which the crucial one is a novel vicinity distribution for adversarial sentences that describes a smooth interpolated embedding space centered around observed training sentence pairs. We then discuss our approach, AdvAug, to train NMT models using the embeddings of virtual sentences in sequence-to-sequence learning. Experiments on Chinese-English, English-French, and English-German translation benchmarks show that AdvAug achieves significant improvements over the Transformer (up to 4.9 BLEU points), and substantially outperforms other data augmentation techniques (e.g. back-translation) without using extra corpora.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Machine Translation (Chinese-to-English) | NIST 2003 (MT-03) | BLEU49.81 | 52 | |
| Machine Translation (Chinese-to-English) | NIST MT-05 2005 | BLEU50.72 | 42 | |
| Machine Translation | NIST MT 04 2004 (test) | BLEU0.5061 | 27 | |
| Machine Translation | NIST MT 06 2006 (test) | BLEU49.98 | 27 | |
| English-German Machine Translation | WMT (newstest2014) | BLEU29.57 | 19 | |
| Machine Translation | NIST Chinese-English MT02 (test) | BLEU50.34 | 14 | |
| Machine Translation | NIST Chinese-English MT08 (test) | BLEU40.45 | 11 | |
| Machine Translation | NIST MT04 | BLEU48.9 | 10 | |
| Machine Translation | IWSLT En-Fr 2016 (test) | BLEU40.91 | 9 | |
| Machine Translation | IWSLT English-French 2016 (test2013) | BLEU Score43.03 | 6 |