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

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

公众号新闻

©作者 | 缪忠剑,李响等
单位 | 小米AI实验室


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



论文标题:
Towards Robust Neural Machine with Iterative Scheduled Data-Switch Training
论文链接:
https://aclanthology.org/2022.coling-1.468

论文作者:

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




研究背景


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


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


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


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


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


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


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


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


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


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


本文方法具有如下优势:


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


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


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





方法介绍


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


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



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


其中 分别表示原始训练样本和对抗训练样本,P 表示模型的输出分布,KL 表示 KL-Divergence 损失函数, 为超参数用于调节 KL-Divergence 损失的重要程度。以上两个训练阶段不断迭代进行,以更好学习原始训练样本和对抗训练样本。
针对使用固定噪音比例构建对抗训练样本的不足,我们提出了在训练过程中基于课程学习策略调节噪音比例来动态构建对抗训练样本。具体来说,每个训练阶段从一个动态变化的数值区间 [0,R(t)] 中均匀采样噪音比例,随着训练进行逐渐增大数值区间的上界 R(t)。
我们设计函数 R(t) 的出发点为:直觉上 R(t) 应该缓慢增大,过快地增大 R(t) 使得抗样本难度变化大,损害模型的优化。因此我们使 R(t) 满足以下约束:


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

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



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


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

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




实验结果

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


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


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


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



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

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


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


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


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


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


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


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


主要实验结论如下:


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


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




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


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


4.1 小爱实时字幕


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





4.2 小爱同声传译


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




4.3 小爱通话传译


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




4.4 MIUI笔记会议秘书


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





4.5 米家眼镜相机


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


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


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


参考文献

[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.


更多阅读



#投 稿 通 道#

 让你的文字被更多人看到 



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


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


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


📝 稿件基本要求:

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

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

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


📬 投稿通道:

• 投稿邮箱:[email protected] 

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

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


△长按添加PaperWeekly小编



🔍


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

进入知乎首页搜索「PaperWeekly」

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


·

微信扫码关注该文公众号作者

戳这里提交新闻线索和高质量文章给我们。
相关阅读
大规模预训练、可控文本生成方向,清华大学计算机系NLP实验室招聘博士后(长期有效)ECCV 2022 Oral | ​Language Matters:面向场景文字检测和端到端识别的弱监督的视觉-语言预训练方法博士后申请 | 西湖大学张岳课题组招收基础自然语言处理、机器翻译、机器学习等方向的博士后黄毅翀@哈尔滨工业大学 - 多语言机器翻译中的收敛一致性研究在美国245.怪电话、年轻褐肤女NeurIPS 2022 | 视觉感知质量攻击:NR-IQA模型鲁棒性的试金石AI对抗训练红队:使 ChatGPT 和 LLM 获得对抗鲁棒性浣溪沙:为佛州,南卡,北卡深深的祈福ECCV 2022 | 多教师对抗鲁棒性蒸馏方法[悲剧]加拿大华人女子异常出血!妇检几个月拿不到结果!实验室:“没办法”Rosalía 登意大利版《VOGUE》封面!七律 重阳兼祝老人节清华大学生命学院、清华大学-IDG/麦戈文脑科学研究院张颖实验室招聘博士后和实验室技术员多伦多45岁女子异常出血!做妇检几个月后拿不到结果!实验室:“积压缺人没办法”广东省重点实验室考核评估出炉,多家材料实验室不合格!清华、上交等联合发表Nature子刊:「分片线性神经网络」最新综述!Nature重磅突破!首次揭示纹状体多巴胺腺苷影响PKA活性调控神经机制,助力帕金森药物研究新策略不想在实验室过年!实验老出错,眼看要延毕,这个protocol拯救了我!!腾讯发布万亿大模型训练方法:最快256卡1天训完万亿NLP大模型防治抑郁症新曙光:胡海岚团队发表Cell论文,揭示社会地位下降导致抑郁的神经机制小透明变身实验大神!吹爆这个保姆级实验protocol!(附13个细胞实验教程)Unsupervised NMT、PBSMT、coarse-to-fine...你都掌握了吗?一文总结机器翻译必备经典模型(二)RNNsearch、Multi-task、attention-model...你都掌握了吗?一文总结机器翻译必备经典模型(一)肿瘤自噬、外泌体实验研究,从细胞实验到动物实验,超靠谱的protocol活着(七)之江实验室:情感计算白皮书(106页)FlowSeq、mBART、BERT-fused、mRASP、mRASP2...你都掌握了吗?一文总结机器翻译必备经典模型(三)苏州实验室及新型显示与视觉感知石城实验室简介选择孤独COLING 2022 | 融合自适应机制与自训练框架的无监督文本顺滑方法清华&上交等发表Nature子刊!分片线性神经网络最新综述!DeepMind关闭首个海外实验室:设立仅五年,曾由Richard Sutton领导大模型如何可靠?IBM等学者最新《基础模型的基础鲁棒性》教程|NeurIPS 2022WMT 2022 国际机器翻译大赛发榜,微信翻译获对话翻译和生物医学领域翻译共三项冠军
logo
联系我们隐私协议©2024 redian.news
Redian新闻
Redian.news刊载任何文章,不代表同意其说法或描述,仅为提供更多信息,也不构成任何建议。文章信息的合法性及真实性由其作者负责,与Redian.news及其运营公司无关。欢迎投稿,如发现稿件侵权,或作者不愿在本网发表文章,请版权拥有者通知本网处理。