References & Citations
Computer Science > Computation and Language
Title: Multi-Task Learning with Shared Encoder for Non-Autoregressive Machine Translation
(Submitted on 24 Oct 2020 (v1), last revised 17 May 2021 (this version, v2))
Abstract: Non-Autoregressive machine Translation (NAT) models have demonstrated significant inference speedup but suffer from inferior translation accuracy. The common practice to tackle the problem is transferring the Autoregressive machine Translation (AT) knowledge to NAT models, e.g., with knowledge distillation. In this work, we hypothesize and empirically verify that AT and NAT encoders capture different linguistic properties of source sentences. Therefore, we propose to adopt Multi-Task learning to transfer the AT knowledge to NAT models through encoder sharing. Specifically, we take the AT model as an auxiliary task to enhance NAT model performance. Experimental results on WMT14 English-German and WMT16 English-Romanian datasets show that the proposed Multi-Task NAT achieves significant improvements over the baseline NAT models. Furthermore, the performance on large-scale WMT19 and WMT20 English-German datasets confirm the consistency of our proposed method. In addition, experimental results demonstrate that our Multi-Task NAT is complementary to knowledge distillation, the standard knowledge transfer method for NAT.
Submission history
From: Yongchang Hao [view email][v1] Sat, 24 Oct 2020 11:00:58 GMT (7169kb,D)
[v2] Mon, 17 May 2021 07:24:55 GMT (98kb,D)
Link back to: arXiv, form interface, contact.