atlas

Research map / Large language models / Alignment and preferences

FLAME : Factuality-Aware Alignment for Large Language Models

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Active and token-level preference learning · 3 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.

Shows that standard SFT and DPO alignment can increase hallucination and proposes FLAME, which uses self-generated responses for fact-based instructions in SFT and a factuality reward model for DPO preference pairs, improving FActScore while keeping instruction following.

Problem

Conventional LLM alignment (SFT followed by RL/DPO) fails to improve factual accuracy and often increases hallucination, because human-written responses introduce unfamiliar knowledge and helpfulness rewards favor long, detailed responses.

Approach

Classify instructions as fact-based; for fact-based ones, train SFT on the model's own few-shot generations (not human responses) and build DPO pairs with a retrieval-augmented atomic-fact factuality reward, while keeping a self-rewarding instruction-following reward for other pairs.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use factuality-aware SFT and combine instruction-following with factuality preference optimizationLlama-2 70B · Alpaca FactFActScore: 41.4 -> 48.7Helped
use factuality-aware SFT and combine instruction-following with factuality preference optimizationLlama-2 70B · Alpaca Evalwin rate: 50.4 -> 51.2Helped
use human responses for non-fact-based instructions and PT generations for fact-based instructionsLlama-2 70B SFT · BioFActScore: 44.7 -> 49.5Helped
fine-tune using human responses rather than PT-generated responsesLlama-2 70B SFT · Alpaca Evalwin rate: 48.1 -> 50.0Helped
fine-tune using human responses rather than PT-generated responsesLlama-2 70B SFT · Alpaca FactFActScore: 42.0 -> 38.6Hurt
add instruction-following DPOLlama-2 70B SFT · BioFActScore: 44.7 -> 42.3Hurt
add factuality-only DPO rather than instruction-following DPO (ablation)Llama-2 70B SFT · BioFActScore: 42.3 -> 47.1Helped
combine instruction-following and factuality preference pairsLlama-2 70B SFT · Alpaca Evalwin rate: 50.0 -> 51.7Helped
remove fact-based instruction classification (ablation)factuality-aware SFT · BioFActScore: 48.4 -> 49.5Hurt
remove fact-based instruction classification (ablation)SFT + factuality-aware DPO · Alpaca Evalwin rate: 46.8 -> 51.7Hurt
classify fact-based sentences and exclude other sentences from fact checking (ablation)SFT + factuality-aware DPO · BioFActScore: 45.0 -> 42.9Hurt
enumerate all possible response pairs instead of selecting maximum- and minimum-reward responses (ablation)SFT + factuality-aware DPO · Alpaca Evalwin rate: 51.7 -> 50.7Hurt

Surprising findings

Limitations

Experimental setup

Datasets: OpenAssistant oasst1 Alpaca Eval FActScore biography entities FAVA annotations TruthfulQA Wikipedia Dec. 2021 dump

Benchmarks: Alpaca Eval Biography (FActScore) Alpaca Fact FAVA TruthfulQA

Metrics: win rate FActScore number of correct/erroneous facts BLEU ROUGE

Baselines: SFT SFT + DPO Llama-2-Chat 70B FActScore-based DPO (Tian et al.) SFT + factuality-only DPO

Builds on: self-rewarding language models direct preference optimization FActScore DRAGON+

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