atlas

Research map / Large language models / Alignment and preferences

Beyond Imitation: Leveraging Fine-grained Quality Signals for Alignment

ICLR 2024 · Large language models › Alignment and preferences · approach: Steering and correcting LLM behavior · 1 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 FIGA, an RL-free alignment method that builds a revised-response dataset (SPA) from Alpaca-7b outputs refined by ChatGPT and trains with token-level weights from Levenshtein edit operations to reward added tokens and penalize deleted ones.

Problem

SFT only imitates demonstrations and lacks fine-grained signals about what makes a response good or bad, while RLHF is complex and unstable; the paper seeks an RL-free alignment method with fine-grained quality signals.

Approach

Contrast a model's low-quality initial response with a minimally revised high-quality version (revised by ChatGPT using a reference demonstration), derive token-level encouraged/discouraged sets via Levenshtein distance, and apply weighted likelihood loss that increases probability of added/substituted tokens and decreases probability of deleted/substituted tokens.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use FIGA instead of SFTAlpaca-7b · SPA test setReward: 4.56 -> 4.62Helped
Use FIGA instead of PPO (85K)Alpaca-7b · OverallAverage: 33.1 -> 34.9Helped
Add reward-based filtering to select low-performing cases (ablation)FIGA · OverallAverage: 32.1 -> 34.9Helped
Revise initial responses rather than use reference responses (ablation)FIGA · OverallAverage: 31.1 -> 34.9Helped
Use edit-based token identification rather than bag-of-words (ablation)FIGA · OverallAverage: 32.7 -> 34.9Helped
Use edit-based token identification rather than ChatGPT weighted annotations (ablation)FIGA · OverallAverage: 31.4 -> 34.9Helped
Use edit-based token identification rather than ChatGPT binary annotations (ablation)FIGA · OverallAverage: 31.6 -> 34.9Helped
Select bad tokens with negative log-likelihood < 0.6 rather than ≥ 0.6 (ablation)FIGA · OverallAverage: 29.3 -> 34.9Helped
Select bad tokens by the further-selection rule rather than penalize all edit-identified bad tokens (ablation)FIGA · OverallAverage: 28.1 -> 34.9Helped
Penalize bad tokens by setting β=0.5 rather than β=0 (ablation)FIGA · OverallAverage: 34.2 -> 34.9Helped
Set γ=0 rather than γ=0.3 (ablation)FIGA · OverallAverage: 33.0 -> 34.9Helped
Use α=1 rather than standardized reward scores for token weights (ablation)FIGA · OverallAverage: 33.4 -> 34.9Helped

Surprising findings

Limitations

Experimental setup

Datasets: HH-RLHF ShareGPT Instruct GPT-J Pairwise SHP OpenOrca

Benchmarks: MMLU TruthfulQA CrowS-Pairs WinoGender Vicuna WizardLM

Metrics: reward score accuracy bias rate win rate average score

Baselines: SFT PPO CoH RRHF DPO Alpaca-7b ChatGPT

Builds on: RLHF PPO DPO RRHF CoH

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