COLING 2022 | 小米AI实验室:更有效的鲁棒性神经机器翻译训练方法

b20fe65a2f232efc3b345bbbd479f191.gif

©作者 | 缪忠剑,李响等

单位 | 小米AI实验室

COLING全称International Conference on Computational Linguistics,是自然语言处理领域重要的国际会议,由国际计算语言学委员会(International Committee on Computational Linguistics,ICCL)创办,每两年举办一次。小米AI实验室联合厦门大学提出的一种更有效的鲁棒神经机器翻译训练方法被COLING 2022 接收为大会口头报告(Oral)。

5d049c4c913e1cceb73fe7b0d5005a11.png

论文标题:

Towards Robust Neural Machine with Iterative Scheduled Data-Switch Training

论文链接:

https://aclanthology.org/2022.coling-1.468

论文作者:

缪忠剑,李响,康立言,张文,周楚伦,陈毅东,王斌,张民,苏劲松

0c440b34593456e657e6a8c60eb050e8.png

研究背景

神经机器翻译(Neural Machine Translation,NMT)一般基于高质量双语平行语料训练模型,在干净无噪音的测试集上获得了显著的翻译质量,并成为主流范式。然而,当待翻译的句子含有少量错误时,例如语音翻译中语音识别产生的错误,NMT 通常会被噪音干扰产生错误译文。

在真实应用场景中含有少量错误的句子十分常见,因此,提升 NMT 的鲁棒性,使其具备一定的抗干扰能力,不仅具有重要的学术研究价值,而且也是实际机器翻译产品应有的能力。 

此前的研究者们主要从两个方面提升 NMT 的鲁棒性: 

1. 面向数据的方法:可以视为一种数据增强的方法,针对实际错误类型构造含有噪音的对抗训练样本(Adversarial Training Example),并将其添加到原始训练数据中形成混合类型训练数据用于训练 NMT 模型; 

2. 面向模型的方法:从优化模型结构和训练方法等维度入手,研究者们探索过多任务学习, 对比学习,对抗训练,引入额外模块等方法提升 NMT 鲁棒性。 

虽然上述方法有效提升了 NMT 的鲁棒性,但仍存在两点不足: 

1. 在利用对抗训练样本时,往往不加区分地使用对抗训练数据和原始训练数据混合而成的训练数据,由于两种数据存在较大差异,因此直接粗暴使用混合数据不是最佳选择; 

2. 在构建对抗训练样本时,往往按照固定比例对干净样本添加错误(噪声),比例过大降低模型在干净无噪音数据上的翻译质量,反之则无法有效提升模型对噪音数据的抗干扰能力,因此使用固定噪音比例不是最佳选择。 

基于课程学习思想,我们提出了一种更有效的构建和使用对抗训练样本方法来提升 NMT 鲁棒性,引入了两个训练阶段,每个阶段使模型专注于基于原始训练数据或对抗训练样本进行优化,克服了同时使用混合数据的缺陷。为了更好地学习这两类数据,防止灾难遗忘问题,我们迭代式的开展两个训练阶段。

此外,为了防止固定噪音比例构建对抗训练样本的不足,我们设计了一种基于课程学习的噪音比例采样方法,在每个训练阶段渐进式增加噪音比例使模型能够从易到难学习训练数据中的噪音。 

本文方法具有如下优势:

1. 我们的训练框架使得 NMT 能够更加专注优化原始干净样本和对抗样本,实验证实这种训练策略能够获得更好的性能;

2. 设计了基于采样的课程学习策略来控制用于生成噪声数据的噪音比例,使得模型能以从易到难的顺序学习对抗样本的信息,实验证实了该方案的优越性;

3. 和模型结构无关,适用于不同的 NMT 模型,并且可以应用于其他 NLP 任务。

f5c6a2ffe9b799cb1d09fa962dab500b.png

方法介绍

252943bc9fc645dab4ceb62b72b0dc06.png

▲ 图1. 迭代式数据交替使用的鲁棒机器翻译训练框架

针对不加区分地使用对抗训练数据和原始训练数据的不足,我们提出了两个训练阶段迭代式的训练方法,每个阶段使模型更专注优化在一类训练数据上的性能。具体来说,图 1 所示粉色框训练阶段主要是基于对抗训练样本优化模型性能,侧重提升模型在噪音样本上的鲁棒性,其损失函数为:

0293708304d8e0d27d10fecafd58e48e.png

而图 1 所示的蓝色框训练阶段主要是基于原始训练数据优化模型性能,侧重提升模型在干净样本上的泛化性,其损失函数为:

13adf1404fd6786be998a44f802b2b15.png

其中 和 分别表示原始训练样本和对抗训练样本,P 表示模型的输出分布,KL 表示 KL-Divergence 损失函数, 为超参数用于调节 KL-Divergence 损失的重要程度。以上两个训练阶段不断迭代进行,以更好学习原始训练样本和对抗训练样本。

针对使用固定噪音比例构建对抗训练样本的不足,我们提出了在训练过程中基于课程学习策略调节噪音比例来动态构建对抗训练样本。具体来说,每个训练阶段从一个动态变化的数值区间 [0,R(t)] 中均匀采样噪音比例,随着训练进行逐渐增大数值区间的上界 R(t)。

我们设计函数 R(t) 的出发点为:直觉上 R(t) 应该缓慢增大,过快地增大 R(t) 使得抗样本难度变化大,损害模型的优化。因此我们使 R(t) 满足以下约束:

741aa637a478442aada6890d1804f8e4.png

其中 为大于 0 的常数。该式表明,随着训练步数 t 的增加,R(t) 在增加的同时,其对于 t 的导数(即:R(t) 的增加速率)在减小,满足缓慢增加 R(t) 的目的。结合训练步数为零时噪音比例为 0 以及最大训练步数时达到最大噪音比例的约束:

65a0882e04738182817ec4e02890f010.png

R(t) 的最终表达式和曲线变化如下:

8cbfa45a9c2c9deddb99db302ec57c16.png

2d8fee8c0a151bbce8bca341f85ef4a4.jpeg

▲ 图2. 噪音比例R(t)的变化曲线

其中 表示最大的噪音比例,T 表示每个训练阶段的最大训练步数,t 表示当前的训练步数,R(t) 的含义为动态变化的数值区间的上界。引入噪音比例控制后,每个训练阶段的具体流程如下:

339c4aa6f8b312f4dd976084deb5fcaf.png

▲ 图3. 每个阶段的训练流程 

6a0fa8457d79af47989a71ec79d12501.png

实验结果

我们在如下 4 个不同规模的机器翻译数据集上分别对比了本文方法和现有其他方法的性能。

a1a6c2410d658f475c069629a49fe6e6.png

3.1 在干净测试集和真实噪音测试集上的翻译质量

我们评估了本文方法在 5 个公开的干净测试集和 2 个真实噪音测试集上的有效性,其中 MTNT Fr→En 数据集包含 2 个来源于社交媒体的噪声测试集(mtnt18 和 mtnt19)和 2 个常规干净测试集(NT14 和 NT15),具体实验结果如表 1 和表 2 所示。

945847e1815c58bb81ded299931805b9.png

▲ 表1. IWSLT'14 De→En,WMT'14 En→De和WMT'20 Zh→En干净测试集上的翻译质量

f5ad0f704e4c26216879298b7243de90.png

▲ 表2. MTNT Fr→En测试集上的翻译质量

3.2 在人工模拟噪音测试集上的翻译质量

我们构造了含有不同噪音比例的噪音测试集来全面评估本文方法的有效性,具体实验结果如图 4 所示。

340336fec4d65d75a140d815c4654cc3.png

▲ 图4. 在不同噪声比例的人工模拟噪音测试集上的模型性能

3.3 对抗样本和干净样本的降维表示

为了更加直观地展示本文方法的优势,我们绘制了不同模型在干净和对应的对抗训练数据上的源句子降维表示。具体地,我们利用词平均编码器输出代表每个源句子的表示,并利用 PCA 降维算法来获取其 2 维表示,具体结果如图 5 所示。

如图 5(e),基于我们所提框架训练得到的模型相对于其他模型(图 5(a)-(d))能够显著降低两类样本的源句差异,为得到更优的鲁棒机器翻译模型奠定了基础。

1917b0ceff4a73231fbb5f759252aa30.png

▲ 图5. 干净数据和对应噪声数据的源句子降维表示(其中蓝色和粉色圆点分别对应噪声数据和干净数据)

主要实验结论如下:

1. 本文方法在真实和人工模拟的噪声测试集上取得了最佳翻译质量,有效提升了 NMT 的鲁棒性;

2. 本文方法在常规的干净测试集上也获得了最佳翻译质量,可作为提升 NMT 泛化性的一种有效方法。

5833b43cb2775dc46f6a8d0b0e7219cd.png

鲁棒性机器翻译的用武之地

目前,小米自研同声传译技术已在多款手机和小爱翻译等产品中落地应用。今年,小米 AI 实验室机器翻译团队不仅获得第十九届国际口语机器翻译大会(IWSLT 2022)“英中文本同传”赛道冠军(夺冠!小米在 IWSLT 2022 国际机器同传比赛中获佳绩),而且还获得了第三届国际机器同传挑战赛“中英文本同传”赛道冠军。

fb0a3e9f14be935b4cf3ae97f8bd856d.png

对于同声传译来说,语音识别受说话人口音和环境噪音等影响难免会出现错误,这些错误一般来说会干扰机器翻译正确性,严重影响用户体验。因此,本文提出的工作对于改善用户体验具有重要价值。

4.1 小爱实时字幕

高通骁龙 8+芯片内置的第七代 AI 芯片提供了强大算力,小米 MIX Fold 2 折叠屏手机行业首发了全离线语音识别和机器翻译(详见又快又准又安全?实时字幕有“芯”秘密!),用户在观看外语视频、开外语在线会议、甚至无网需要外语交流时都可以接近实时的看到双语字幕,同时,音频数据默认保存在手机本地,无需上传云端处理,隐私更安全。

1df8e1d9a334838c49ab7cf21e62b36c.jpeg

4.2 小爱同声传译

双人对话,边说边译,面对面跨语言沟通无障碍,能轻松应对全球多个国家的不同英文口音。

6e5d407fdb57831f6a3f127b2dc8ace1.png

4.3 小爱通话传译

解决跨语言通话中的语言障碍问题,说话人直接说母语,接收方听到的也是用母语发音的译文,实现通话双方自由交流。

37c5504e5fff320d9f96731d26b64e31.png

4.4 MIUI笔记会议秘书

涉外会议中,使用小米 MIX Fold 2 折叠屏手机可以直接翻译并记录会议内容,还有精细的声纹识别可以将会中角色也区别开来,全面解放你的双手和头脑,把时间留给思考,实现高效的 AI 会议纪要。

540f2898007f8dfadfee1c2a553abfb4.png

4.5 米家眼镜相机

最新发布的米家眼镜相机也内置了能通提供实时中英互译的“小爱翻译”功能,用户在参加外语会议和出国旅游时,佩戴眼镜就可以轻松理解外语内容和应对外语交流。

近年来,小米在机器翻译方向的技术积累日渐丰厚,并已在小米手机和 AIoT 智能硬件产品中广泛落地,陆续自研了“同声传译”、“会议秘书”、“实时字幕”、“通话翻译”,“图片翻译”,“网页翻译”等功能。 

通过“全能翻,极速译”的小爱翻译 App,小米手机用户无论是与外国友人面对面还是通话沟通、观看没有字幕外语视频,浏览外文网页、甚至遇到含有外文的图片时,都能随时随地便捷的获得机器翻译译文,实现了边说边译、边听边译、边看边译,轻松应对各种语言障碍,让全球每个人享受没有语言障碍的美好生活。

outside_default.png

参考文献

outside_default.png

[1] Belinkov et al. 2018. Synthetic and Natural Noise Both Break Neural Machine Translation. In Proc. of ICLR.

[2] Cheng et al. 2018. Towards Robust Neural Machine Translation. In Proc. of ACL.

[3] Jiao et al. 2021. Alternated Training with Synthetic and Authentic Data for Neural Machine Translation. In Proc. of ACL Findings.

[4] Karpukhin et al. 2019. Training on Synthetic Noise Improves Robustness to Natural Noise in Machine Translation. In Proc. of W-NUT@EMNLP.

[5] Wang et al. 2021. Secoco: Self-Correcting Encoding for Neural Machine Translation. In Proc. of EMNLP Findings.

[6] Wang et al. 2021. A Comprehensive Survey on Curriculum Learning. IEEE T-PAMI.

[7] Xu et al. 2021. Addressing the Vulnerability of NMT in Input Perturbations. In Proc. of NAACL.

[8] Zhou et al. 2019. Improving Robustness of Neural Machine Translation with Multi-task Learning. In Proc. of WMT.

[9] Qin et al. 2021. Modeling Homophone Noise for Robust Neural Machine Translation. In Proc. of ICASSP.

[10] Michel et al. 2018. MTNT: A Testbed for Machine Translation of Noisy Text. In Proc. of EMNLP.

[11] Cheng et al. 2020. Advaug: Robust Adversarial Augmentation for Neural Machine Translation. In Proc. of ACL.

[12] Lee et al. 2021. Contrastive Learning with Adversarial Perturbations for Conditional Text Generation. In Proc. of ICLR.

更多阅读

6e5e157cb28767a4383bcf0fbd872bcc.png

0c9b21f90d24c8e2514cc3378da3a8d6.png

cc14cce42662b424546df0d234b873e1.png

4750f20348d4e943aa991f1f3b6322e7.gif

#投 稿 通 道#

 让你的文字被更多人看到 

如何才能让更多的优质内容以更短路径到达读者群体,缩短读者寻找优质内容的成本呢?答案就是:你不认识的人。

总有一些你不认识的人,知道你想知道的东西。PaperWeekly 或许可以成为一座桥梁,促使不同背景、不同方向的学者和学术灵感相互碰撞,迸发出更多的可能性。 

PaperWeekly 鼓励高校实验室或个人,在我们的平台上分享各类优质内容,可以是最新论文解读,也可以是学术热点剖析科研心得竞赛经验讲解等。我们的目的只有一个,让知识真正流动起来。

📝 稿件基本要求:

• 文章确系个人原创作品,未曾在公开渠道发表,如为其他平台已发表或待发表的文章,请明确标注 

• 稿件建议以 markdown 格式撰写,文中配图以附件形式发送,要求图片清晰,无版权问题

• PaperWeekly 尊重原作者署名权,并将为每篇被采纳的原创首发稿件,提供业内具有竞争力稿酬,具体依据文章阅读量和文章质量阶梯制结算

📬 投稿通道:

• 投稿邮箱:hr@paperweekly.site 

• 来稿请备注即时联系方式(微信),以便我们在稿件选用的第一时间联系作者

• 您也可以直接添加小编微信(pwbot02)快速投稿,备注:姓名-投稿

57b6b0bb39faf0cbea71f8d396135819.png

△长按添加PaperWeekly小编

🔍

现在,在「知乎」也能找到我们了

进入知乎首页搜索「PaperWeekly」

点击「关注」订阅我们的专栏吧

·

94504d202543298ff9f3ef12e28ee50b.jpeg

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值