通过分治专家架构提升大模型多语言翻译能力,缓解参数干扰问题。
Mix-MoE: Improving Multilingual Machine Translation of Large Language Models through Mixed MoEs

- 将专家层分为语言建模与翻译专用两类,分工存储单语与双语知识。
- 在多语言翻译任务上显著优于基线模型,有效缓解微调时的参数冲突。
- 引入傅里叶特征路由机制,挖掘文本潜在结构模式,增强专家协作。
大型语言模型(LLMs)在多语言机器翻译(MT)中展现出巨大潜力,即使在有限双语监督下亦然。然而,使用平行语料微调LLM面临主要挑战:参数干扰。为此,我们提出Mix-MoE,一种用于多语言MT的混合专家框架。该框架分两阶段进行:(1) 在单语语料上进行MoE后预训练;(2) 在平行语料上进行MoE后预训练。关键在于将MoE层分为两类:语言模型专家(LM Experts)和机器翻译专家(MT Experts)。LM Experts 用于保留预训练模型学到的单语知识,而 MT Experts 则专门学习并存储双语翻译知识。此外,为促进两类专家间的有效交互并利用文本中潜在的结构模式,我们引入基于模型表示傅里叶变换特征的路由机制。实验结果表明,Mix-MoE在多语言翻译任务中表现优异,显著超越现有基线,并在缓解参数干扰方面取得明显进展。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have shown great promise in multilingual machine translation (MT), even with limited bilingual supervision. However, fine-tuning LLMs with parallel corpora presents major challenges, namely parameter interference. To address these issues, we propose Mix-MoE, a mixed Mixture-of-Experts framework designed to train LLMs for multilingual MT. Our framework operates in two distinct stages: (1) post-pretraining with MoE on monolingual corpora, and (2) post-pretraining with MoE on parallel corpora. Crucially, we divide the MoE layers into two specialized groups: Language Model Experts (LM Experts) and Machine Translation Experts (MT Experts). LM Experts are designed to capture and retain the monolingual knowledge learned by the pre-trained LLM. MT Experts, on the other hand, are specifically trained to acquire and store bilingual translation knowledge. Furthermore, to facilitate effective interaction between these specialized experts and leverage potential underlying structural patterns in text, we introduce a routing mechanism enhanced by Fourier Transform features derived from model representations. The experimental results demonstrate that Mix-MoE excels in multilingual MT, significantly outperforming existing baselines and showing notable progress in mitigating parameter interference.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。