X-ALMA: Plug & Play Modules and Adaptive Rejection for Quality Translation at Scale

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Xu, Haoran, Murray, Kenton, Koehn, Philipp, Hoang, Hieu, Eriguchi, Akiko, Khayrallah, Huda
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909519033925632
author Xu, Haoran
Murray, Kenton
Koehn, Philipp
Hoang, Hieu
Eriguchi, Akiko
Khayrallah, Huda
author_facet Xu, Haoran
Murray, Kenton
Koehn, Philipp
Hoang, Hieu
Eriguchi, Akiko
Khayrallah, Huda
contents Large language models (LLMs) have achieved remarkable success across various NLP tasks with a focus on English due to English-centric pre-training and limited multilingual data. In this work, we focus on the problem of translation, and while some multilingual LLMs claim to support for hundreds of languages, models often fail to provide high-quality responses for mid- and low-resource languages, leading to imbalanced performance heavily skewed in favor of high-resource languages. We introduce **X-ALMA**, a model designed to ensure top-tier performance across 50 diverse languages, regardless of their resource levels. X-ALMA surpasses state-of-the-art open-source multilingual LLMs, such as Aya-101 and Aya-23, in every single translation direction on the FLORES-200 and WMT'23 test datasets according to COMET-22. This is achieved by plug-and-play language-specific module architecture to prevent language conflicts during training and a carefully designed training regimen with novel optimization methods to maximize the translation performance. After the final stage of training regimen, our proposed **A**daptive **R**ejection **P**reference **O**ptimization (**ARPO**) surpasses existing preference optimization methods in translation tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2410_03115
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle X-ALMA: Plug & Play Modules and Adaptive Rejection for Quality Translation at Scale
Xu, Haoran
Murray, Kenton
Koehn, Philipp
Hoang, Hieu
Eriguchi, Akiko
Khayrallah, Huda
Computation and Language
Large language models (LLMs) have achieved remarkable success across various NLP tasks with a focus on English due to English-centric pre-training and limited multilingual data. In this work, we focus on the problem of translation, and while some multilingual LLMs claim to support for hundreds of languages, models often fail to provide high-quality responses for mid- and low-resource languages, leading to imbalanced performance heavily skewed in favor of high-resource languages. We introduce **X-ALMA**, a model designed to ensure top-tier performance across 50 diverse languages, regardless of their resource levels. X-ALMA surpasses state-of-the-art open-source multilingual LLMs, such as Aya-101 and Aya-23, in every single translation direction on the FLORES-200 and WMT'23 test datasets according to COMET-22. This is achieved by plug-and-play language-specific module architecture to prevent language conflicts during training and a carefully designed training regimen with novel optimization methods to maximize the translation performance. After the final stage of training regimen, our proposed **A**daptive **R**ejection **P**reference **O**ptimization (**ARPO**) surpasses existing preference optimization methods in translation tasks.
title X-ALMA: Plug & Play Modules and Adaptive Rejection for Quality Translation at Scale
topic Computation and Language
url https://arxiv.org/abs/2410.03115