高级检索

基于知识蒸馏的低频词翻译优化策略

Knowledge Distillation-based Optimization Strategy for Low-frequency Word Translation in Neural Machine Translation

  • 摘要: 该文针对神经机器翻译中低频词翻译不足的问题,研究了基于知识蒸馏的低频词增强方法。首先训练低频词翻译能力更强的教师模型,并通过单教师蒸馏模型引导学生模型学习低频词的翻译知识;在此基础上,进一步提出双教师蒸馏模型,以增强训练稳定性并兼顾高频词翻译性能。实验结果表明,单教师蒸馏模型在英语到德语任务上较现有最优方法提升0.64个BLEU值;双教师蒸馏模型在保证高频词翻译效果基本不变的前提下,使英语到德语、捷克语和法语任务的低频词翻译分别提升1.24、0.47和0.87个BLEU值。结果说明,该方法能够有效提升低频词的翻译质量,并在多个翻译任务上取得稳定改进。

     

    Abstract: Low-frequency words remain difficult for neural machine translation This paper proposes a knowledge distillation method to improve translation quality for informative but relatively infrequent words. A teacher model with stronger low-frequency-word translation ability is first trained, and a single-teacher distillation framework is then used to guide student learning. A dual-teacher distillation model is further designed to improve training stability and preserve high-frequency-word performance. Experimental results showed that the single-teacher model increases BLEU by 0.64 in the English-German task over the state-of-the-art system, and the dual-teacher model improve BLEU by 1.24 in English-German, 0.47 in English-Czech, and 0.87 in English-French.

     

/

返回文章
返回