atlas

Research map / Large language models / Instruction and fine-tuning

Teaching Large Language Models to Translate with Comparison

AAAI 2024 · Large language models › Instruction and fine-tuning · approach: Calibration and knowledge in LLM tuning · 13 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes TIM, which instruction-tunes open LLMs (BLOOMZ-7b-mt, LLaMA-2) for translation using output comparison (order-, dictionary-, error-guided examples) and a token-level preference loss over correct vs. incorrect translations.

Problem

Open-sourced LLMs fine-tuned with standard instruction tuning (next-token loss) underperform on translation, especially smaller models trained on limited high-quality data, and tend to hallucinate.

Approach

Teach translation by comparison: include contrastive examples (reordered, dictionary-aligned, and error-annotated translations) plus a token-level preference loss that pushes the hidden-state reward head to prefer correct translations over noisy or LM-generated bad outputs, added to the LM loss without a separate reward model or RLHF.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add reverse-order output comparison data (ablation)TIM-FixEmb · WMT22 Zh⇒EnBLEU: 23.08 -> 23.41Helped
Add dictionary-guided output comparison data (ablation)TIM-FixEmb · WMT22 Zh⇒EnBLEU: 23.08 -> 23.73Helped
Add error-guided output comparison data (ablation)TIM-FixEmb · WMT22 Zh⇒EnBLEU: 23.08 -> 23.94Helped
Use LM-based rather than noisy-based bad outputs (ablation)TIM-FixEmb · WMT22 Zh⇒EnBLEU: 24.11 -> 24.51Helped
Add the preference loss (ablation)TIM-FixEmb · WMT22 Zh⇒EnBLEU: 23.76 -> 24.51Helped
Add output comparison (ablation)TIM-FixEmb · WMT22 Zh⇒EnBLEU: 23.21 -> 24.51Helped
Add error-free instruction notesTIM with beam-4 decoding · WMT22 De⇒EnBLEU: 26.12 -> 26.24No clear effect
Add dictionary alignment notesTIM with beam-4 decoding · WMT22 En⇒ZhBLEU: 37.83 -> 36.32Hurt
Use TIM rather than MT instruction tuningBLOOMZ-7b-mt · FLORES-200 En⇒DeBLEU: 32.23 -> 33.32Helped

Surprising findings

Limitations

Experimental setup

Datasets: WMT22 test sets FLORES-200 dev-test WMT newstest2017-2021 validation data Alpaca MQM newstest2020 (en-de, zh-en) MUSE bilingual dictionaries

Benchmarks: WMT22 test sets (De-En, En-De, Zh-En, En-Zh) FLORES-200 dev-test WMT22 multilingual-to-English zero-shot (cs, ja, ru, uk to en) WMT22 Metrics Shared Task MQM (De-En, En-De)

Metrics: BLEU COMET hallucination rate (sentence and token level) system-level accuracy Pearson correlation

Baselines: Alpaca-LoRA Alpaca-Full MT-LoRA MT-FixEmb MT-Full WMT22 winners NLLB-3.3B Alpaca-7b Vicuna-13b BayLing-7b BayLing-13b ChatGPT

Builds on: instruction tuning LoRA RLHF reward modeling Alpaca

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy