Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Contrastive Preference Optimization: Pushing the Boundaries of LLM Performance in Machine Translation

About

Moderate-sized large language models (LLMs) -- those with 7B or 13B parameters -- exhibit promising machine translation (MT) performance. However, even the top-performing 13B LLM-based translation models, like ALMA, does not match the performance of state-of-the-art conventional encoder-decoder translation models or larger-scale LLMs such as GPT-4. In this study, we bridge this performance gap. We first assess the shortcomings of supervised fine-tuning for LLMs in the MT task, emphasizing the quality issues present in the reference data, despite being human-generated. Then, in contrast to SFT which mimics reference translations, we introduce Contrastive Preference Optimization (CPO), a novel approach that trains models to avoid generating adequate but not perfect translations. Applying CPO to ALMA models with only 22K parallel sentences and 12M parameters yields significant improvements. The resulting model, called ALMA-R, can match or exceed the performance of the WMT competition winners and GPT-4 on WMT'21, WMT'22 and WMT'23 test datasets.

Haoran Xu, Amr Sharaf, Yunmo Chen, Weiting Tan, Lingfeng Shen, Benjamin Van Durme, Kenton Murray, Young Jin Kim• 2024

Related benchmarks

TaskDatasetResultRank
Instruction FollowingAlpacaEval 2.0
Win Rate53.4
722
Multi-turn Dialogue EvaluationMT-Bench--
532
Instruction FollowingAlpacaEval
Win Rate45.13
420
Instruction FollowingArena Hard
Win Rate55.2
263
Bias EvaluationBBQ
Accuracy95.2
171
Instruction FollowingAlpacaEval 2
LC (%)56.4
137
Multi-turn Instruction FollowingMT-Bench
MT-Bench Score (GPT-4)8
129
Multi-turn dialogueMT-Bench
MT-Bench Score8.2
126
LLM Alignment EvaluationAlpacaEval 2
LC Win Rate38.1
89
Machine TranslationEn-Es document-level
d-COMET85.79
66
Showing 10 of 47 rows

Other info

Follow for update