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

By Haoran Xu, Amr Sharaf, Yunmo Chen, Weiting Tan, Lingfeng Shen, Benjamin Van Durme, Kenton Murray, Young Jin Kim. International Conference on Machine Learning, 2024

477 citations recorded by the source corpus.

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.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

View Paper: https://www.semanticscholar.org/paper/ebd1c04c61f73f46def3305ca11d038c46665b65

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Dan Jurafsky
  2. Douwe Kiela
  3. Niklas Muennighoff
  4. Kawin Ethayarajh
  5. Mohit Bansal
  6. Jason Aaron Edward Weston
  7. Deep Karkhanis
  8. Siddartha Naidu

In Computer Vision and Pattern Recognition

  1. Dan Jurafsky
  2. Kawin Ethayarajh
  3. Douwe Kiela
  4. Deep Karkhanis
  5. Niklas Muennighoff
  6. Siddartha Naidu
  7. Samuel Dooley
  8. Manley Roberts

Citations in the corpus