3 个月前

将预训练语言模型集成到神经机器翻译中

将预训练语言模型集成到神经机器翻译中

摘要

神经机器翻译(Neural Machine Translation, NMT)在自然语言处理领域已发展成为一项关键技术,得益于持续深入的研究与开发。然而,高质量双语语料数据的匮乏仍然是制约NMT性能提升的主要瓶颈。近年来,研究者们开始探索利用预训练语言模型(Pre-trained Language Model, PLM)中的上下文信息来缓解这一问题。然而,PLM与NMT模型之间的不兼容性问题尚未得到有效解决。为此,本研究提出了一种融合PLM的NMT模型——PiNMT(PLM-integrated NMT),以应对上述挑战。PiNMT模型包含三个核心组件:PLM多层转换器(PLM Multi Layer Converter)、嵌入融合模块(Embedding Fusion)以及余弦对齐机制(Cosine Alignment),各组件在有效传递PLM信息至NMT模型方面发挥关键作用。此外,本文还引入两种新的训练策略:独立学习率策略(Separate Learning Rates)与双阶段训练策略(Dual Step Training)。通过采用所提出的PiNMT模型及其训练方法,我们在IWSLT'14 En↔De数据集上取得了当前最优的翻译性能。本研究的成果具有重要意义,首次展示了高效融合PLM与NMT的创新路径,有效解决了两者之间的不兼容性问题,显著提升了翻译性能。

基准测试

基准方法指标
machine-translation-on-iwslt2014-englishPiNMT
BLEU score: 32.2
machine-translation-on-iwslt2014-germanPiNMT
BLEU score: 40.43

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
将预训练语言模型集成到神经机器翻译中 | 论文 | HyperAI超神经