atlas

Research map / Large language models / Alignment and preferences

GRATH: Gradual Self-Truthifying for Large Language Models

ICML 2024 · Large language models › Alignment and preferences · approach: Robust and calibrated DPO · 2 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 GRATH, a self-supervised post-processing method that generates pairwise correct/incorrect answers to out-of-domain ARC-Challenge questions via few-shot prompting and iteratively refines them with DPO (LoRA) to boost TruthfulQA truthfulness of 7B LLMs.

Problem

LLMs often generate untruthful content, with modest TruthfulQA MC1/MC2 accuracy; existing truthfulness fine-tuning needs in-domain, human-annotated answers, and DPO can be vulnerable to domain shift.

Approach

Use the model itself to generate correct and incorrect answer pairs for out-of-domain questions via few-shot prompting with TruthfulQA demonstrations, fine-tune with DPO, then iteratively regenerate the correct answers using the improved model and repeat (gradual self-truthifying), without any annotated answers.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply GRATH to Llama2-Chat-7B using ARC-Challenge-sourced self-generated pairwise truthfulness data and one gradual refinement iterationLlama2-Chat-7B · TruthfulQA MC1accuracy: 30.23 -> 54.71Helped
Apply GRATH to Llama2-Chat-7B using ARC-Challenge-sourced self-generated pairwise truthfulness data and one gradual refinement iterationLlama2-Chat-7B · TruthfulQA MC2accuracy: 45.32 -> 69.10Helped
Apply GRATHZephyr-7B · TruthfulQA MC1accuracy: 42.23 -> 53.86Helped
Apply GRATHZephyr-7B · TruthfulQA MC2accuracy: 57.83 -> 66.73Helped
Apply GRATHLlama2-Chat-7B · MMLUaccuracy: 48.14 -> 46.88Mixed
Apply DPO using OOD ARC-Challenge annotated pairwise answers instead of GRATH's self-generated pairwise dataLlama2-Chat-7B · TruthfulQA MC1accuracy: 30.23 -> 36.72Helped
Use OOD DPO instead of the pretrained modelZephyr-7B · TruthfulQA MC1accuracy: -4.2%Hurt
Use fewer pairwise truthfulness training examples at fixed DPO executions (ablation)GRATH · TruthfulQA MC1accuracyHurt
Increase the number of DPO executions beyond the effective early iterations (ablation)GRATH · TruthfulQA MC1accuracyMixed
Use the pretrained model rather than the current base model as DPO's reference across iterations (ablation)GRATH · TruthfulQA MC1accuracyHurt
Increase the domain gap in training answers using stronger Shakespearean-style transformations (ablation)DPO self-truthifying · TruthfulQA MC1accuracyHurt
Use in-domain TruthfulQA questions rather than OOD questions (ablation)DPO self-truthifying · TruthfulQAaccuracyHelped

Surprising findings

Limitations

Experimental setup

Datasets: ARC-Challenge TruthfulQA HellaSwag MMLU HaluEval

Benchmarks: TruthfulQA ARC-Challenge HellaSwag MMLU

Metrics: MC1 accuracy MC2 accuracy normalized accuracy

Baselines: SFT DPO RLHF RL RepE ITI Xwin-LM v0.1 Mistral-Instruct v0.1 Zephyr Llama2-Chat

Builds on: DPO RepE ITI TRL

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