arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.03045cs.CL
收藏

为纯文本翻译模型添加多模态能力

Adding Multimodal Capabilities to a Text-only Translation Model

PAR Government Systems Corporation(PAR政府系统公司);Air Force Research Laboratory(空军研究实验室)

Vipin Vijayan, Braeden Bowen, Scott Grigsby, Timothy Anderson, Jeremy Gwinnup

AI总结:

针对现有多模态机器翻译模型过拟合Multi30k、在纯文本测试集表现差的问题,研究以高性能纯文本MT模型为基础,加入门控连接的视觉-文本适配器层,经视觉掩码预训练和Multi30k微调,实现两类数据集上的性能兼顾。

摘要:

尽管当前大多数多模态机器翻译(MMT)研究使用Multi30k数据集进行训练和评估,但我们发现所得模型会极度过拟合Multi30k数据集。因此,这些模型在WMT newstest数据集等典型纯文本测试集上的表现非常差。为了在Multi30k和典型纯文本数据集上都取得良好表现,我们以一个性能优异的纯文本机器翻译(MT)模型作为MMT模型的构建起点。我们向该MT模型添加通过门控机制连接的视觉-文本适配器层,并通过两个步骤将MT模型逐步转化为MMT模型:1)使用基于视觉的源文本掩码进行预训练;2)在Multi30k上进行微调。

英文摘要:

While most current work in multimodal machine translation (MMT) uses the Multi30k dataset for training and evaluation, we find that the resulting models overfit to the Multi30k dataset to an extreme degree. Consequently, these models perform very badly when evaluated against typical text-only testing sets such as the WMT newstest datasets. In order to perform well on both Multi30k and typical text-only datasets, we use a performant text-only machine translation (MT) model as the starting point of our MMT model. We add vision-text adapter layers connected via gating mechanisms to the MT model, and incrementally transform the MT model into an MMT model by 1) pre-training using vision-based masking of the source text and 2) fine-tuning on Multi30k.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑